modify static initializaer to setup_class

c-3lab · Mar 27, 2023 · 0cb5f27 · 0cb5f27
1 parent 9380b43
commit 0cb5f27
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/ckanext/feedback/tests/controllers/test_download.py b/ckanext/feedback/tests/controllers/test_download.py
@@ -27,11 +27,13 @@ def get_downloads(resource_id):
 
 @pytest.mark.usefixtures('clean_db', 'with_plugins', 'with_request_context')
 class TestDownloadController:
-    model.repo.init_db()
-    engine = get_engine('db', '5432', 'ckan_test', 'ckan', 'ckan')
-    create_utilization_tables(engine)
-    create_resource_tables(engine)
-    create_download_tables(engine)
+    @classmethod
+    def setup_class(cls):
+        model.repo.init_db()
+        engine = get_engine('db', '5432', 'ckan_test', 'ckan', 'ckan')
+        create_utilization_tables(engine)
+        create_resource_tables(engine)
+        create_download_tables(engine)
 
     @patch('ckanext.feedback.controllers.download.download')
     def test_extended_download(self, mocker):

diff --git a/ckanext/feedback/tests/services/download/test_summary.py b/ckanext/feedback/tests/services/download/test_summary.py
@@ -28,11 +28,13 @@ def get_downloads(resource_id):
 
 @pytest.mark.usefixtures('clean_db', 'with_plugins', 'with_request_context')
 class TestDownloadServices:
-    model.repo.init_db()
-    engine = get_engine('db', '5432', 'ckan_test', 'ckan', 'ckan')
-    create_utilization_tables(engine)
-    create_resource_tables(engine)
-    create_download_tables(engine)
+    @classmethod
+    def setup_class(cls):
+        model.repo.init_db()
+        engine = get_engine('db', '5432', 'ckan_test', 'ckan', 'ckan')
+        create_utilization_tables(engine)
+        create_resource_tables(engine)
+        create_download_tables(engine)
 
     def test_increment_resource_downloads(self):
         resource = factories.Resource()