test_facility_level0.py 1.35 KB
Newer Older
Wei Shoulin's avatar
c3  
Wei Shoulin committed
1
2
import unittest

Wei Shoulin's avatar
Wei Shoulin committed
3
from csst_dfs_api.facility.level0 import Level0DataApi
Wei Shoulin's avatar
c3  
Wei Shoulin committed
4

Wei Shoulin's avatar
Wei Shoulin committed
5
class Level0DataTestCase(unittest.TestCase):
Wei Shoulin's avatar
c3  
Wei Shoulin committed
6
7
8
9
10

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

    def test_find(self):
Wei Shoulin's avatar
Wei Shoulin committed
11
        recs = self.api.find(obs_id = '0000011', obs_type = 'sci', limit = 0)
Wei Shoulin's avatar
c3  
Wei Shoulin committed
12
13
        print('find:', recs)

Wei Shoulin's avatar
hstdm    
Wei Shoulin committed
14
15
16
17
    def test_find_by_brick_ids(self):
        recs = self.api.find_by_brick_ids(brick_ids = [1,2,3,4])
        print('find:', recs)

Wei Shoulin's avatar
c3  
Wei Shoulin committed
18
    def test_get(self):
Wei Shoulin's avatar
Wei Shoulin committed
19
        rec = self.api.get(id = 3)
Wei Shoulin's avatar
c3  
Wei Shoulin committed
20
21
        print('get:', rec)

Wei Shoulin's avatar
Wei Shoulin committed
22
        rec = self.api.get(level0_id = '000001101',obs_type='sci')
Wei Shoulin's avatar
Wei Shoulin committed
23
24
        print('get:', rec)        

Wei Shoulin's avatar
c3  
Wei Shoulin committed
25
    def test_update_proc_status(self):
Wei Shoulin's avatar
Wei Shoulin committed
26
        rec = self.api.update_proc_status(level0_id = '000001101', status = 6)
Wei Shoulin's avatar
c3  
Wei Shoulin committed
27
28
29
        print('update_proc_status:', rec)

    def test_update_qc0_status(self):
Wei Shoulin's avatar
Wei Shoulin committed
30
        rec = self.api.update_qc0_status(level0_id = '000001101', status = 7)
Wei Shoulin's avatar
c3  
Wei Shoulin committed
31
32
33
34
        print('update_qc0_status:', rec)

    def test_write(self):
        rec = self.api.write(
Wei Shoulin's avatar
Wei Shoulin committed
35
            level0_id = '000001101',
Wei Shoulin's avatar
Wei Shoulin committed
36
            obs_id = '0000011',
Wei Shoulin's avatar
Wei Shoulin committed
37
            detector_no = "01",
Wei Shoulin's avatar
c3  
Wei Shoulin committed
38
39
40
41
42
43
44
            obs_type = "sci",            
            obs_time = "2021-06-06 11:12:13",
            exp_time = 150,
            detector_status_id = 3,
            filename = "MSC_00001234",
            file_path = "/opt/MSC_00001234.fits")
        print('write:', rec)