nucliadb-utils 5.1.1.post1343__py3-none-any.whl → 5.1.1.post1349__py3-none-any.whl

nucliadb_utils/tests/fixtures.py

@@ -18,10 +18,16 @@
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 #
 import os
+from typing import Any, Type
+from unittest.mock import Mock
 
 import pytest
 from pytest_lazy_fixtures import lazy_fixture
 
+from nucliadb_utils.storages.azure import AzureStorage
+from nucliadb_utils.storages.gcs import GCSStorage
+from nucliadb_utils.storages.local import LocalStorage
+from nucliadb_utils.storages.s3 import S3Storage
 from nucliadb_utils.utilities import Utility, clean_utility, set_utility
 
 
@@ -49,7 +55,7 @@ def get_testing_storage_backend(default="gcs"):
     return os.environ.get("TESTING_STORAGE_BACKEND", default)
 
 
-def lazy_storage_fixture(default="gcs"):
+def lazy_storage_fixture():
     backend = get_testing_storage_backend()
     fixture_name = f"{backend}_storage"
     return [lazy_fixture.lf(fixture_name)]
@@ -66,3 +72,34 @@ async def storage(request):
     yield storage_driver
 
     clean_utility(Utility.STORAGE)
+
+
+@pytest.fixture(scope="function")
+def storage_settings(request, storage) -> dict[str, Any]:
+    """Useful fixture that returns the settings used in the generic `storage`
+    fixture.
+
+    This becomes useful when `storage` is overwritten programatically. Tests or
+    fixtures depending on generic storage settings can use this fixutre to get
+    the appropiate settings, being agnostic to the specific storage used.
+
+    """
+    if isinstance(storage, Mock):
+        yield {}
+    else:
+        storage_backend_map: dict[Type, str] = {
+            AzureStorage: "azure",
+            GCSStorage: "gcs",
+            LocalStorage: "local",
+            S3Storage: "s3",
+        }
+
+        backend = storage_backend_map.get(type(storage), None)
+        if backend is None:
+            raise Exception(f"Unknown storage configured ({type(storage)}), can't get settings")
+
+        fixture_name = f"{backend}_storage_settings"
+        # print("Using storage settings:", fixture_name)
+        assert fixture_name in request.fixturenames
+        settings: dict[str, Any] = request.getfixturevalue(fixture_name)
+        yield settings

{nucliadb_utils-5.1.1.post1343.dist-info → nucliadb_utils-5.1.1.post1349.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb_utils
-Version: 5.1.1.post1343
+Version: 5.1.1.post1349
 Home-page: https://nuclia.com
 License: BSD
 Classifier: Development Status :: 4 - Beta
@@ -24,8 +24,8 @@ Requires-Dist: PyNaCl
 Requires-Dist: pyjwt>=2.4.0
 Requires-Dist: memorylru>=1.1.2
 Requires-Dist: mrflagly>=0.2.9
-Requires-Dist: nucliadb-protos>=5.1.1.post1343
-Requires-Dist: nucliadb-telemetry>=5.1.1.post1343
+Requires-Dist: nucliadb-protos>=5.1.1.post1349
+Requires-Dist: nucliadb-telemetry>=5.1.1.post1349
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-5.1.1.post1343.dist-info → nucliadb_utils-5.1.1.post1349.dist-info}/RECORD

@@ -58,14 +58,14 @@ nucliadb_utils/storages/utils.py,sha256=8g2rIwJeYIumQLOB47Yw1rx3twlhRB_cJxer65Qf
 nucliadb_utils/tests/__init__.py,sha256=Oo9CAE7B0eW5VHn8sHd6o30SQzOWUhktLPRXdlDOleA,1456
 nucliadb_utils/tests/asyncbenchmark.py,sha256=x4be2IwCawle9zWgMOJkmwoUwk5p1tv7cLQGmybkEOg,10587
 nucliadb_utils/tests/azure.py,sha256=Dg-Eb4KVScG-O6P9y-bVQZTAKTNUMQ0i-CKEd9IdrWw,4474
-nucliadb_utils/tests/fixtures.py,sha256=b-e6Ciewh7GSh9rriG01bN9GiMW6xbGgrp3KiX7HNiY,2089
+nucliadb_utils/tests/fixtures.py,sha256=i0sqPqe5a5JlKGFdaIvOlHYkZ3pHZ2hTIgTsaIB3vSM,3472
 nucliadb_utils/tests/gcs.py,sha256=Bx-kxf2NoYYeyNFPpBFxbRN8kIiYFpELEvqG0V1_9Ks,4183
 nucliadb_utils/tests/indexing.py,sha256=YW2QhkhO9Q_8A4kKWJaWSvXvyQ_AiAwY1VylcfVQFxk,1513
 nucliadb_utils/tests/local.py,sha256=7nuP8EFUAiA8ZH50R1iPV9EUXBySQxOanVm3Zht_e0g,1835
 nucliadb_utils/tests/nats.py,sha256=xqpww4jZjTKY9oPGlJdDJG67L3FIBQsa9qDHxILR8r8,7687
 nucliadb_utils/tests/s3.py,sha256=IdMxK_cNdSHLvO1u8BwsKFzD87Hk1MVPDZ57zx6h-rA,3656
-nucliadb_utils-5.1.1.post1343.dist-info/METADATA,sha256=f7IKZ1pSojpApZqrBFPEXIxgOmxrANVUxI2x5lyn9uw,2071
-nucliadb_utils-5.1.1.post1343.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-nucliadb_utils-5.1.1.post1343.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-5.1.1.post1343.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-5.1.1.post1343.dist-info/RECORD,,
+nucliadb_utils-5.1.1.post1349.dist-info/METADATA,sha256=TuyUdvQXJwsaPpjCRbEDs9GKx63vIyzgplZMztcTAZk,2071
+nucliadb_utils-5.1.1.post1349.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+nucliadb_utils-5.1.1.post1349.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-5.1.1.post1349.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-5.1.1.post1349.dist-info/RECORD,,