Update oauth test

segmentio · Sep 25, 2024 · c10e43d · c10e43d
1 parent e045297
commit c10e43d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/segment/analytics/test/test_oauth.py b/segment/analytics/test/test_oauth.py
@@ -55,7 +55,7 @@ def json(self):
     elif kwargs['url'] == 'http://127.0.0.1:400/token':
         return MockResponse({"reason": "test_reason", "json_data" : {"error":"unrecoverable", "error_description":"nah"}}, 400)
     elif kwargs['url'] == 'http://127.0.0.1:429/token':
-        return MockResponse({"reason": "test_reason", "headers" : {"X-RateLimit-Reset": time.time()*1000 + 2000}}, 429)
+        return MockResponse({"reason": "test_reason", "headers" : {"X-RateLimit-Reset": 2000}}, 429)
     elif kwargs['url'] == 'http://127.0.0.1:500/token':
         return MockResponse({"reason": "test_reason", "json_data" : {"error":"recoverable", "error_description":"nah"}}, 500)
     elif kwargs['url'] == 'http://127.0.0.1:501/token':
@@ -152,4 +152,4 @@ def test_oauth_integration_fail_bad_key(self, mock_post):
         self.assertTrue(self.failed)
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()