nucliadb-utils 6.2.1.post2706__py3-none-any.whl → 6.2.1.post2716__py3-none-any.whl

nucliadb_utils/nats.py

@@ -130,6 +130,7 @@ class NatsConnectionManager:
         self._needs_reconnection = False
         self._reconnect_task: Optional[asyncio.Task] = None
         self._expected_subscriptions: set[str] = set()
+        self._initialized = False
 
     def healthy(self) -> bool:
         if not self._healthy:
@@ -147,6 +148,9 @@ class NatsConnectionManager:
         return True
 
     async def initialize(self) -> None:
+        if self._initialized:
+            return
+
         options: dict[str, Any] = {
             "error_cb": self.error_cb,
             "closed_cb": self.closed_cb,
@@ -165,7 +169,12 @@ class NatsConnectionManager:
 
         self._expected_subscription_task = asyncio.create_task(self._verify_expected_subscriptions())
 
+        self._initialized = True
+
     async def finalize(self):
+        if not self._initialized:
+            return
+
         async with self._lock:
             if self._reconnect_task:
                 self._reconnect_task.cancel()
@@ -209,6 +218,8 @@ class NatsConnectionManager:
                 pass
             await self._nc.close()
 
+        self._initialized = False
+
     async def disconnected_cb(self) -> None:
         logger.info("Disconnected from NATS!")
         self._last_unhealthy = time.monotonic()

nucliadb_utils/tests/nats.py

@@ -29,6 +29,7 @@ from io import BytesIO
 from urllib.request import urlopen
 from zipfile import ZipFile
 
+import nats
 import pytest
 from pytest_docker_fixtures import images  # type: ignore
 from pytest_docker_fixtures.containers._base import BaseImage  # type: ignore
@@ -256,3 +257,13 @@ def natsdocker():  # pragma: no cover
     print("Started natsd docker")
     yield f"nats://{nats_host}:{nats_port}"
     nats_image.stop()
+
+
+@pytest.fixture(scope="function")
+async def nats_server(natsd: str):
+    yield natsd
+
+    # cleanup nats
+    nc = await nats.connect(servers=[natsd])
+    await nc.drain()
+    await nc.close()

{nucliadb_utils-6.2.1.post2706.dist-info → nucliadb_utils-6.2.1.post2716.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb_utils
-Version: 6.2.1.post2706
+Version: 6.2.1.post2716
 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>=6.2.1.post2706
-Requires-Dist: nucliadb-telemetry>=6.2.1.post2706
+Requires-Dist: nucliadb-protos>=6.2.1.post2716
+Requires-Dist: nucliadb-telemetry>=6.2.1.post2716
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-6.2.1.post2706.dist-info → nucliadb_utils-6.2.1.post2716.dist-info}/RECORD

@@ -8,7 +8,7 @@ nucliadb_utils/featureflagging.py,sha256=dKp8u4g52KTcupKFJkdGYk3Z2pGQdi_q7IaoY_G
 nucliadb_utils/grpc.py,sha256=apu0uePnkGHCAT7GRQ9YZfRYyFj26kJ440i8jitbM3U,3314
 nucliadb_utils/helpers.py,sha256=nPw8yod3hP-pxq80VF8QC36s7ygSg0dBUdfI-LatvCs,1600
 nucliadb_utils/indexing.py,sha256=Luaqcar3CySpdYOFp6Q9Fyr8ZYwhYhaKRHQ_VGL78f8,2318
-nucliadb_utils/nats.py,sha256=YsSuhGKpZjyZXjVT7uAF6KjRMWvp1k7-SF9ddFXdVC0,15059
+nucliadb_utils/nats.py,sha256=99YJkK1QMfWDxKJK8u0acK0lLYb8yThalzulGETiT80,15266
 nucliadb_utils/partition.py,sha256=jBgy4Hu5Iwn4gjbPPcthSykwf-qNx-GcLAIwbzPd1d0,1157
 nucliadb_utils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nucliadb_utils/run.py,sha256=Es0_Bu5Yc-LWczvwL6gzWqSwC85RjDCk-0oFQAJi9g4,1827
@@ -57,10 +57,10 @@ nucliadb_utils/tests/fixtures.py,sha256=i0sqPqe5a5JlKGFdaIvOlHYkZ3pHZ2hTIgTsaIB3
 nucliadb_utils/tests/gcs.py,sha256=KW_DLet1WRlssSW55eI-IQ-0d94Jo2Oh7Di4xGv4JCc,4685
 nucliadb_utils/tests/indexing.py,sha256=YW2QhkhO9Q_8A4kKWJaWSvXvyQ_AiAwY1VylcfVQFxk,1513
 nucliadb_utils/tests/local.py,sha256=fXIBasrvdaFJM-sw2wk1_oiFzBcm9O10iCyC-OiXwY8,1914
-nucliadb_utils/tests/nats.py,sha256=xqpww4jZjTKY9oPGlJdDJG67L3FIBQsa9qDHxILR8r8,7687
+nucliadb_utils/tests/nats.py,sha256=nykEfwxPc3F3aO8n7g_V7IY3RQBXU3_R_ukwBKxYXa8,7893
 nucliadb_utils/tests/s3.py,sha256=pl-RJFjA4MH6iXkqhsh5g8gDuEhrYu1nPZ-laxlrMlE,3704
-nucliadb_utils-6.2.1.post2706.dist-info/METADATA,sha256=PnZXNsfOR3kXYxgyG-2egWAbLMrgZHd-8AtWI9i5NxE,2055
-nucliadb_utils-6.2.1.post2706.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-nucliadb_utils-6.2.1.post2706.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-6.2.1.post2706.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-6.2.1.post2706.dist-info/RECORD,,
+nucliadb_utils-6.2.1.post2716.dist-info/METADATA,sha256=pPpkXMVwOGk63XzFzShhQfQvSpMibwDXZ0eRQH3y5Dk,2055
+nucliadb_utils-6.2.1.post2716.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+nucliadb_utils-6.2.1.post2716.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-6.2.1.post2716.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-6.2.1.post2716.dist-info/RECORD,,