Actually fix WebAuth tests

solsticegamestudios · Feb 22, 2025 · 6438c15 · 6438c15
1 parent 2591b04
commit 6438c15
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/tests/test_webauth.py b/tests/test_webauth.py
@@ -10,12 +10,14 @@ class WACase(unittest.TestCase):
     def test_http_error(self, mrs_mock):
         mm = mrs_mock.return_value = mock.MagicMock()
 
+        mm.get.side_effect = requests.exceptions.ConnectTimeout('test')
         mm.post.side_effect = requests.exceptions.ConnectTimeout('test')
 
         with self.assertRaises(wa.HTTPError):
             wa.WebAuth('a', 'b').login()
 
         mm.post.reset_mock()
+        mm.get.side_effect = requests.exceptions.ReadTimeout('test')
         mm.post.side_effect = requests.exceptions.ReadTimeout('test')
 
         with self.assertRaises(wa.HTTPError):
@@ -27,13 +29,10 @@ def test_login_user_and_pass_only_ok(self):
         s = user.login()
 
         self.assertTrue(user.logged_on)
-        self.assertIsInstance(s, requests.Session)
-
-        for domain in s.cookies.list_domains():
-            self.assertEqual(s.cookies.get('steamLogin', domain=domain), '0%7C%7C{}'.format('A'*16))
-            self.assertEqual(s.cookies.get('steamLoginSecure', domain=domain), '0%7C%7C{}'.format('B'*16))
-            self.assertEqual(s.cookies.get('steamMachineAuth0', domain=domain), 'C'*16)
+        self.assertEqual(user.access_token, 'A'*16)
+        self.assertEqual(user.refresh_token, 'B'*16)
 
+        self.assertIsInstance(s, requests.Session)
         self.assertEqual(s, user.login())
 
     @vcr.use_cassette('vcr/webauth_user_pass_only_fail.yaml', mode='none', serializer='yaml')