Mock requests in test setup to speed up agent tests

canonical · Oct 31, 2023 · dd9938e · dd9938e
1 parent 7f853a1
commit dd9938e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/agent/testflinger_agent/tests/test_agent.py b/agent/testflinger_agent/tests/test_agent.py
@@ -16,7 +16,7 @@
 
 class TestClient:
     @pytest.fixture
-    def agent(self):
+    def agent(self, requests_mock):
         self.tmpdir = tempfile.mkdtemp()
         self.config = {
             "agent_id": "test01",
@@ -31,6 +31,8 @@ def agent(self):
         }
         testflinger_agent.configure_logging(self.config)
         client = _TestflingerClient(self.config)
+        requests_mock.get(rmock.ANY)
+        requests_mock.post(rmock.ANY)
         yield _TestflingerAgent(client)
         # Inside tests, we patch rmtree so that we can check files after the
         # run, so we need to clean up the tmpdirs here