nucliadb-utils 6.2.1.post2835__py3-none-any.whl → 6.2.1.post2842__py3-none-any.whl

nucliadb_utils/tests/nats.py

@@ -17,220 +17,16 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 
-import http.client
-import os
-import platform
-import signal
 import socket
-import subprocess
-import tempfile
-import time
-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
 
-
-class Gnatsd(object):
-    def __init__(
-        self,
-        port=4222,
-        user="",
-        password="",
-        token="",
-        timeout=0,
-        http_port=8222,
-        debug=False,
-        tls=False,
-        cluster_listen=None,
-        routes=None,
-        config_file=None,
-    ):  # pragma: no cover
-        self.port = port
-        self.user = user
-        self.password = password
-        self.timeout = timeout
-        self.http_port = http_port
-        self.proc = None
-        self.debug = debug
-        self.tls = tls
-        self.token = token
-        self.cluster_listen = cluster_listen
-        self.routes = routes or []
-        self.bin_name = "gnatsd"
-        self.config_file = config_file
-        self.folder = None
-
-        env_debug_flag = os.environ.get("DEBUG_NATS_TEST")
-        if env_debug_flag == "true":
-            self.debug = True
-
-    def start(self):
-        folder = tempfile.TemporaryDirectory()
-        cmd = [
-            f"{self.path}/{self.bin_name}",
-            "-js",
-            "-p",
-            "%d" % self.port,
-            "-m",
-            "%d" % self.http_port,
-            "-a",
-            "0.0.0.0",
-            "-sd",
-            folder.name,
-            "--log",
-            f"{folder.name}/log.log",
-        ]
-        self.folder = folder
-        if self.user != "":
-            cmd.append("--user")
-            cmd.append(self.user)
-        if self.password != "":
-            cmd.append("--pass")
-            cmd.append(self.password)
-
-        if self.token != "":
-            cmd.append("--auth")
-            cmd.append(self.token)
-
-        if self.debug:
-            cmd.append("-DV")
-
-        if self.tls:
-            cmd.append("--tls")
-            cmd.append("--tlscert")
-            cmd.append("tests/certs/server-cert.pem")
-            cmd.append("--tlskey")
-            cmd.append("tests/certs/server-key.pem")
-            cmd.append("--tlsverify")
-            cmd.append("--tlscacert")
-            cmd.append("tests/certs/ca.pem")
-
-        if self.cluster_listen is not None:
-            cmd.append("--cluster_listen")
-            cmd.append(self.cluster_listen)
-
-        if len(self.routes) > 0:
-            cmd.append("--routes")
-            cmd.append(",".join(self.routes))
-
-        if self.config_file is not None:
-            cmd.append("--config")
-            cmd.append(self.config_file)
-
-        if self.debug:
-            self.proc = subprocess.Popen(cmd)
-        else:
-            self.proc = subprocess.Popen(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-
-        if self.debug:
-            if self.proc is None:
-                print(
-                    "[\031[0;33mDEBUG\033[0;0m] Failed to start server listening on port %d started."
-                    % self.port
-                )
-            else:
-                print("[\033[0;33mDEBUG\033[0;0m] Server listening on port %d started." % self.port)
-        return self.proc
-
-    def stop(self):
-        if self.folder is not None:
-            self.folder.cleanup()
-        if self.debug:
-            print("[\033[0;33mDEBUG\033[0;0m] Server listening on %d will stop." % self.port)
-
-        if self.debug:
-            if self.proc is None:
-                print(
-                    "[\033[0;31mDEBUG\033[0;0m] Failed terminating server listening on port %d"
-                    % self.port
-                )
-
-        if self.proc.returncode is not None:
-            if self.debug:
-                print(
-                    "[\033[0;31mDEBUG\033[0;0m] Server listening on port {port} finished running already with exit {ret}".format(  # noqa: E501
-                        port=self.port, ret=self.proc.returncode
-                    )
-                )
-        else:
-            os.kill(self.proc.pid, signal.SIGKILL)
-            self.proc.wait()
-            if self.debug:
-                print("[\033[0;33mDEBUG\033[0;0m] Server listening on %d was stopped." % self.port)
-
-
-class NatsServer(Gnatsd):  # pragma: no cover
-    def __init__(self):
-        super(Gnatsd, self)
-        self.bin_name = "nats-server"
-
-
-def start_gnatsd(gnatsd: Gnatsd):  # pragma: no cover
-    gnatsd.start()
-
-    endpoint = "127.0.0.1:{port}".format(port=gnatsd.http_port)
-    retries = 0
-    while True:
-        if retries > 100:
-            break
-
-        try:
-            httpclient = http.client.HTTPConnection(endpoint, timeout=5)
-            httpclient.request("GET", "/varz")
-            response = httpclient.getresponse()
-            if response.status == 200:
-                break
-        except Exception:
-            retries += 1
-            time.sleep(0.1)
-
-
-@pytest.fixture(scope="session")
-def natsd_server():  # pragma: no cover
-    # Create a persistent temporary directory
-    tmpdir = tempfile.mkdtemp()
-    nats_server_path = os.path.join(tmpdir, "nats-server")
-
-    if not os.path.isfile(nats_server_path):
-        version = "v2.10.12"
-        arch = platform.machine()
-        if arch == "x86_64":
-            arch = "amd64"
-        system = platform.system().lower()
-
-        url = f"https://github.com/nats-io/nats-server/releases/download/{version}/nats-server-{version}-{system}-{arch}.zip"
-
-        resp = urlopen(url)
-        zipfile = ZipFile(BytesIO(resp.read()))
-
-        file = zipfile.open(f"nats-server-{version}-{system}-{arch}/nats-server")
-        content = file.read()
-        with open(nats_server_path, "wb") as f:
-            f.write(content)
-        os.chmod(nats_server_path, 0o755)
-
-    server = Gnatsd(port=4222)
-    server.bin_name = "nats-server"
-    server.path = tmpdir
-    return server
-
-
-@pytest.fixture(scope="session")
-def natsd(natsd_server: Gnatsd):  # pragma: no cover
-    start_gnatsd(natsd_server)
-    print("Started natsd")
-    yield f"nats://localhost:{natsd_server.port}"
-    natsd_server.stop()
-
-
 images.settings["nats"] = {
     "image": "nats",
-    "version": "2.10.12",
+    "version": "2.10.21",
     "options": {"command": ["-js"], "ports": {"4222": None}},
 }
 
@@ -252,7 +48,7 @@ nats_image = NatsImage()
 
 
 @pytest.fixture(scope="session")
-def natsdocker():  # pragma: no cover
+def natsd():  # pragma: no cover
     nats_host, nats_port = nats_image.run()
     print("Started natsd docker")
     yield f"nats://{nats_host}:{nats_port}"

{nucliadb_utils-6.2.1.post2835.dist-info → nucliadb_utils-6.2.1.post2842.dist-info}/METADATA

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

{nucliadb_utils-6.2.1.post2835.dist-info → nucliadb_utils-6.2.1.post2842.dist-info}/RECORD

@@ -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=nykEfwxPc3F3aO8n7g_V7IY3RQBXU3_R_ukwBKxYXa8,7893
+nucliadb_utils/tests/nats.py,sha256=RWHjwqq5esuO7OFbP24yYX1cXnpPLcWJwDUdmwCpH28,1897
 nucliadb_utils/tests/s3.py,sha256=pl-RJFjA4MH6iXkqhsh5g8gDuEhrYu1nPZ-laxlrMlE,3704
-nucliadb_utils-6.2.1.post2835.dist-info/METADATA,sha256=KwZVJk5wY5SxTDZuETS_635KUdS_t26MDDckIUf_YcY,2204
-nucliadb_utils-6.2.1.post2835.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-nucliadb_utils-6.2.1.post2835.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-6.2.1.post2835.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-6.2.1.post2835.dist-info/RECORD,,
+nucliadb_utils-6.2.1.post2842.dist-info/METADATA,sha256=5qR7hlR1UIG34SJP1cSWz7-qRL0q7AEWp1fFs3mOtAU,2204
+nucliadb_utils-6.2.1.post2842.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+nucliadb_utils-6.2.1.post2842.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-6.2.1.post2842.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-6.2.1.post2842.dist-info/RECORD,,