import unittest

from csst_dfs_api.facility.level0 import Level0DataApi

class Level0DataTestCase(unittest.TestCase):

    def setUp(self):
        self.api = Level0DataApi()

    def test_find(self):
        recs = self.api.find(file_type = 'FLAT', create_time = ('2024-01-01 01:01:01','2024-11-01 01:01:01'),  limit = 0)
        print('find:', recs)

    def test_get(self):
        rec = self.api.get(id = 14)
        print('get:', rec)

        rec = self.api.get(level0_id = '000001101',file_type='sci')
        print('get:', rec)        

    def test_update_proc_status(self):
        rec = self.api.update_proc_status(level0_id = '000001101', status = 6)
        print('update_proc_status:', rec)

    def test_update_qc0_status(self):
        rec = self.api.update_qc0_status(level0_id = '000001101', status = 7)
        print('update_qc0_status:', rec)

    def test_write(self):
        rec = self.api.write(
            level0_id = '10000010101',
            obs_id = '100000101',
            detector_no = "01",
            filter = "u",
            file_type = "SCI",            
            obs_time = "2021-06-06 11:12:13",
            exp_time = 150,
            detector_status_id = 3,
            filename = "CSST_MSC_MS_SCI_20270810142128_20270810142358_100000101_01_L0_1.fits",
            file_path = "/opt/temp/csst/CSST_MSC_MS_SCI_20270810142128_20270810142358_100000101_01_L0_1.fits")
        print('write:', rec)