import unittest
import bvid
import os
import sys
# 获取当前文件所在目录的绝对路径
current_dir = os.path.dirname(os.path.abspath(__file__))

# 获取 developed_code 目录的绝对路径，并添加到 sys.path 中
developed_code_path = os.path.abspath(os.path.join(current_dir, '..'))
sys.path.append(developed_code_path)


class TestBvid(unittest.TestCase):

    def test_get_source(self):
        source = bvid.get_source(1)
        self.assertTrue('bvid' in source)

    def test_extract_bv(self):
        html = '{"bvid":"test_bvid","title":"test"}'
        bvs = bvid.extract_bv(html)
        self.assertIn("test_bvid", bvs)

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