test_common_db.py

import unittest

from csst_dfs_api_local.common.db import DBClient

class DBClientTestCase(unittest.TestCase):
    def setUp(self):
        self.db_path = "/opt/temp/csst"

    def test_db_init(self):
        db = DBClient()
        
        r = db.select_one("select count(*) as c from ifs_rawfits")
        if r is not None:
            print("ifs_rawfits count:", r['c'])

        r = db.exists("select * from ifs_rawfits where id=2323")
        if r:
            print("existed")
        else:
            print("not existed")