import os
import unittest
from astropy.io import fits

from csst_dfs_api.mbi.level1prc import Level1PrcApi

class MBILevel1PrcTestCase(unittest.TestCase):

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

    def test_find(self):
        recs = self.api.find(level1_id=1)
        print('find:', recs)

    def test_update_proc_status(self):
        rec = self.api.update_proc_status(id = 1, status = 4)
        print('update_proc_status:', rec)

    def test_write(self):
        rec = self.api.write(level1_id=1, 
            pipeline_id = "P1",
            prc_module = "QC0",
            params_file_path = "/opt/dddasd.params",
            prc_status = 3,
            prc_time = '2021-06-04 11:12:13',
            result_file_path = "/opt/dddasd.header")
        print('write:', rec)