import unittest

from gfwlist_to_dns import get_dnsmasq_text, get_smartdns_domain_set


class MyTestCase(unittest.TestCase):
    def test_get_dnsmasq_text(self):
        dnsmasq_text = get_dnsmasq_text()
        # print(dnsmasq_text)
        self.assertIsInstance(dnsmasq_text, str)

    def test_get_smartdns_domain_set(self):
        domain_set = get_smartdns_domain_set()
        # print(domain_set)
        self.assertIsInstance(domain_set, str)


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