[pre-commit.ci] Apply automatic pre-commit fixes

conda-store-server/tests/_internal/worker/test_app.py

@@ -2,26 +2,27 @@
 # Use of this source code is governed by a BSD-style
 # license that can be found in the LICENSE file.
 
-import pytest
 import logging
 import sys
 
-from traitlets.config import Config
-
 from unittest.mock import patch
 
+import pytest
+
+from traitlets.config import Config
+
 from conda_store_server._internal.worker.app import CondaStoreWorker
 
 
-class MockCeleryApp():
+class MockCeleryApp:
     def worker_main(argv):
         pass
 
 
 @pytest.mark.skipif(
     sys.platform == "win32", reason="celery beat is not supported on windows"
 )
-@patch('conda_store_server.app.CondaStore.celery_app')
+@patch("conda_store_server.app.CondaStore.celery_app")
 def test_start_worker(mock_celery_app, conda_store_config):
     """Test that the celery worker is started with the right arguments"""
     mock_celery_app.return_value = MockCeleryApp()
@@ -35,13 +36,15 @@ def test_start_worker(mock_celery_app, conda_store_config):
     worker = CondaStoreWorker(config=conda_store_config)
     worker.initialize()
     worker.start()
-    mock_celery_app.worker_main.assert_called_with([
-        "worker",
-        "--loglevel=WARNING",
-        "--max-tasks-per-child=10",
-        "--beat",
-        "--concurrency=4"
-    ])
+    mock_celery_app.worker_main.assert_called_with(
+        [
+            "worker",
+            "--loglevel=WARNING",
+            "--max-tasks-per-child=10",
+            "--beat",
+            "--concurrency=4",
+        ]
+    )
 
 
 def test_initialize_worker_with_valid_config_file(conda_store_config, tmp_path):
@@ -64,21 +67,21 @@ def test_initialize_worker_with_valid_config_file(conda_store_config, tmp_path):
 def test_initialize_worker_with_invalid_config(conda_store_config):
     """Test that a worker is not able to init if the provided config file does not exist"""
     worker = CondaStoreWorker(config=conda_store_config)
-    with pytest.raises(SystemExit) as exc_info:   
+    with pytest.raises(SystemExit) as exc_info:
         worker.config_file = "/i/dont/exist"
         worker.initialize()
-    
+
     assert exc_info.value.code == 1
 
 
 @pytest.mark.parametrize(
-        "logging_level,output",
-        [
-            (logging.INFO, "INFO"),
-            (logging.CRITICAL, "CRITICAL"),
-            (logging.FATAL, "CRITICAL"),
-            (logging.WARN, "WARNING"),
-        ]
+    "logging_level,output",
+    [
+        (logging.INFO, "INFO"),
+        (logging.CRITICAL, "CRITICAL"),
+        (logging.FATAL, "CRITICAL"),
+        (logging.WARN, "WARNING"),
+    ],
 )
 def test_logger_to_celery_logging_level(logging_level, output):
     worker = CondaStoreWorker()

conda-store-server/tests/test_storage.py

@@ -2,9 +2,9 @@
 
 import pytest
 
-from conda_store_server import storage
+from conda_store_server import api, storage
 from conda_store_server._internal import schema
-from conda_store_server import api
+
 
 @pytest.fixture()
 def local_file_store(tmp_path):
@@ -16,7 +16,7 @@ def local_file_store(tmp_path):
     tf_two = str(tmp_path / "testfile2")
     with open(tf_two, "w") as file:
         file.write("testfile2")
-    
+
     logs_path = tmp_path / "logs"
     os.mkdir(logs_path)
 
@@ -29,65 +29,80 @@ def test_fset_new_package(self, db):
         store.fset(db, 123, "new_build_key", "", schema.BuildArtifactType.YAML)
         assert len(api.list_build_artifacts(db).all()) == 1
 
-
     def test_fset_existing_package(self, seed_conda_store):
         store = storage.Storage()
         db = seed_conda_store
-        
+
         inital_artifacts = api.list_build_artifacts(db).all()
         artifact = inital_artifacts[0]
-        
+
         store.fset(db, artifact.build_id, artifact.key, "", artifact.artifact_type)
         assert len(api.list_build_artifacts(db).all()) == len(inital_artifacts)
 
     def test_delete_existing_artifact(self, seed_conda_store):
         store = storage.Storage()
         db = seed_conda_store
-        
+
         inital_artifacts = api.list_build_artifacts(db).all()
         artifact = inital_artifacts[0]
-        
+
         store.delete(db, artifact.build_id, artifact.key)
-        assert len(api.list_build_artifacts(db).all()) == len(inital_artifacts)-1
+        assert len(api.list_build_artifacts(db).all()) == len(inital_artifacts) - 1
 
 
 class TestLocalStorage:
     def test_fset_new_package(self, db, local_file_store):
         store = storage.LocalStorage()
         store.storage_path = local_file_store
-        
+
         filename = str(local_file_store / "testfile1")
-        store.fset(db, 123, "new_build_key", filename, "contenttype", schema.BuildArtifactType.YAML)
+        store.fset(
+            db,
+            123,
+            "new_build_key",
+            filename,
+            "contenttype",
+            schema.BuildArtifactType.YAML,
+        )
         assert len(api.list_build_artifacts(db).all()) == 1
-        
+
         target_file = str(local_file_store / "new_build_key")
         assert os.path.exists(target_file)
-        target_content = open(target_file, 'r').read()
+        target_content = open(target_file, "r").read()
         assert target_content == "testfile1"
 
     def test_fset_dont_overwrite_file(self, db, local_file_store):
         store = storage.LocalStorage()
         store.storage_path = local_file_store
-        
+
         filename = str(local_file_store / "testfile1")
-        store.fset(db, 123, "testfile2", filename, "contenttype", schema.BuildArtifactType.YAML)
+        store.fset(
+            db, 123, "testfile2", filename, "contenttype", schema.BuildArtifactType.YAML
+        )
         assert len(api.list_build_artifacts(db).all()) == 1
-        
+
         target_file = str(local_file_store / "testfile2")
         assert os.path.exists(target_file)
-        target_content = open(target_file, 'r').read()
+        target_content = open(target_file, "r").read()
         assert target_content == "testfile1"
 
     def test_set_new_package(self, db, local_file_store):
         store = storage.LocalStorage()
         store.storage_path = local_file_store
-
-        store.set(db, 123, "new_build_key", b"somestuff", "contenttype", schema.BuildArtifactType.YAML)
+
+        store.set(
+            db,
+            123,
+            "new_build_key",
+            b"somestuff",
+            "contenttype",
+            schema.BuildArtifactType.YAML,
+        )
         assert len(api.list_build_artifacts(db).all()) == 1
-        
+
         target_file = str(local_file_store / "new_build_key")
         assert os.path.exists(target_file)
-        target_content = open(target_file, 'r').read()
+        target_content = open(target_file, "r").read()
         assert target_content == "somestuff"
 
     def test_get(self, local_file_store):
@@ -110,6 +125,6 @@ def test_delete(self, seed_conda_store, local_file_store):
             file.write("testfile2")
 
         store.delete(db, artifact.build_id, artifact.key)
-        assert len(api.list_build_artifacts(db).all()) == len(inital_artifacts)-1
-        
+        assert len(api.list_build_artifacts(db).all()) == len(inital_artifacts) - 1
+
         assert not os.path.exists(target_file)