diff --git a/tests/test_login.py b/tests/test_login.py index 924a8bc..a0bf48e 100644 --- a/tests/test_login.py +++ b/tests/test_login.py @@ -15,7 +15,7 @@ class TestLogin(unittest.TestCase): constant.CONFIG['useragent'] = os.getenv('NHENTAI_UA') def test_cookie_set(self): - self.assertTrue(bool(constant.CONFIG['cookie'])) + self.assertTrue(constant.CONFIG['useragent'].startswith('Mozilla')) def test_cookie(self): try: