import unittest
import time
import os
from ctypes import *

so_path = os.getenv('HOME') + '/lib/librealbase.so'
realbase = CDLL(so_path) # 我们使用 ctypes 来库加载 C 语言编写的 .so 动态库

class RealBaseTests(unittest.TestCase):

    def test_GetSystemDateTime(self):
        pdate = create_string_buffer(9)
        ptime = create_string_buffer(9)
        realbase.GetSystemDateTime(pdate, ptime) #调用 librealbase.so 中的 GetSystemDateTime 函数
        now = time.localtime(time.time()) #通过 python 库获取当前时间
        self.assertEqual(now[0], int(pdate[0:4])) #测试从C代码里获取的年
        self.assertEqual(now[1], int(pdate[4:6])) #测试月
        self.assertEqual(now[2], int(pdate[6:8])) #测试日

        #测试获取的时间格式是否符合 hh:mm:ss
        self.assertEqual(':', ptime[2])
        self.assertEqual(':', ptime[5])


if __name__ == '__main__':
    unittest.main()