Migrate api_clients tests to pytest

python_apps/api_clients/tests/requestprovider_test.py

@@ -1,39 +1,33 @@
-import json
-import unittest
-
+import pytest
 from api_clients.utils import RequestProvider
 from api_clients.version1 import api_config
-from configobj import ConfigObj
-from mock import MagicMock, patch
 
 
-class TestRequestProvider(unittest.TestCase):
-    def setUp(self):
-        self.cfg = api_config
-        self.cfg["general"] = {}
-        self.cfg["general"]["base_dir"] = "/test"
-        self.cfg["general"]["base_port"] = 80
-        self.cfg["general"]["base_url"] = "localhost"
-        self.cfg["general"]["api_key"] = "TEST_KEY"
-        self.cfg["api_base"] = "api"
-
-    def test_test(self):
-        self.assertTrue("general" in self.cfg)
-
-    def test_init(self):
-        rp = RequestProvider(self.cfg, {})
-        self.assertEqual(len(rp.available_requests()), 0)
-
-    def test_contains(self):
-        methods = {
-            "upload_recorded": "/1/",
-            "update_media_url": "/2/",
-            "list_all_db_files": "/3/",
-        }
-        rp = RequestProvider(self.cfg, methods)
-        for meth in methods:
-            self.assertTrue(meth in rp.requests)
+@pytest.fixture()
+def config():
+    return {
+        **api_config,
+        "general": {
+            "base_dir": "/test",
+            "base_port": 80,
+            "base_url": "localhost",
+            "api_key": "TEST_KEY",
+        },
+        "api_base": "api",
+    }
 
 
-if __name__ == "__main__":
-    unittest.main()
+def test_request_provider_init(config):
+    request_provider = RequestProvider(config, {})
+    assert len(request_provider.available_requests()) == 0
+
+
+def test_request_provider_contains(config):
+    endpoints = {
+        "upload_recorded": "/1/",
+        "update_media_url": "/2/",
+        "list_all_db_files": "/3/",
+    }
+    request_provider = RequestProvider(config, endpoints)
+    for endpoint in endpoints:
+        assert endpoint in request_provider.requests