fast-test-database 0.1.2__tar.gz → 0.2.0__tar.gz

{fast-test-database-0.1.2 → fast_test_database-0.2.0}/PKG-INFO

@@ -1,17 +1,16 @@
 Metadata-Version: 2.1
 Name: fast-test-database
-Version: 0.1.2
+Version: 0.2.0
 Summary: Configure an in-memory database for running Django tests
 License: GPL-3.0-or-later
 Author: Alexey Kotlyarov
 Author-email: a@koterpillar.com
-Requires-Python: >=3.7,<4.0
+Requires-Python: >=3.11,<4.0
 Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.7
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Requires-Dist: Django (>=3.0,<4.0)
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: Django (>=5.0.0,<6.0.0)
 Description-Content-Type: text/x-rst
 
 Fast test database

{fast-test-database-0.1.2 → fast_test_database-0.2.0}/fast_test_database/__init__.py

@@ -13,7 +13,7 @@ from time import sleep
 def docker(*args):
     """Run Docker with the specified arguments and return the output."""
 
-    return subprocess.check_output(('docker',) + args).decode().strip()
+    return subprocess.check_output(("docker",) + args).decode().strip()
 
 
 class DatabaseProvider(object):
@@ -35,18 +35,12 @@ class DatabaseProvider(object):
     @property
     def container_name(self):
         """The Docker container name."""
-        return 'fast_database_{}_{}'.format(
-            os.path.basename(os.getcwd()),
-            self.image.replace('/', '.').replace(':', '-'),
-        )
+        return f"fast_database_{os.path.basename(os.getcwd())}_{self.image.replace('/', '.').replace(':', '-')}"
 
     @property
     def image(self):
         """The Docker image."""
-        return '{}:{}'.format(
-            self.IMAGE,
-            self.version or 'latest'
-        )
+        return f"{self.IMAGE}:{self.version or 'latest'}"
 
     def provide(self, engine):
         """
@@ -56,38 +50,38 @@ class DatabaseProvider(object):
         """
 
         # TODO: Randomize password
-        password = 'fast_database'
+        password = "fast_database"
 
         try:
-            container = json.loads(docker('inspect', self.container_name))
-            available = container[0]['State']['Running']
+            container = json.loads(docker("inspect", self.container_name))
+            available = container[0]["State"]["Running"]
         except (LookupError, ValueError, subprocess.CalledProcessError):
             available = False
 
         if not available:
             # Delete the old container in case it's stopped
             try:
-                docker('rm', '-f', self.container_name)
+                docker("rm", "-f", self.container_name)
             except subprocess.CalledProcessError:
                 pass
 
-            args = [
-                'run',
-                '--detach',
-                '--env', '{}={}'.format(self.PASSWORD_ENV_VAR, password),
-            ]
+            args = ["run", "--detach", "--env", f"{self.PASSWORD_ENV_VAR}={password}"]
 
             # Add custom environment variables
             if self.CUSTOM_ENV:
                 for k, v in self.CUSTOM_ENV.items():
-                    args.extend(['--env', '{}={}'.format(k, v)])
-
-            args.extend([
-                '--tmpfs={}'.format(self.DATA_DIR),
-                '--publish', str(self.PORT),
-                '--name', self.container_name,
-                self.image,
-            ])
+                    args.extend(["--env", f"{k}={v}"])
+
+            args.extend(
+                [
+                    f"--tmpfs={self.DATA_DIR}",
+                    "--publish",
+                    str(self.PORT),
+                    "--name",
+                    self.container_name,
+                    self.image,
+                ]
+            )
 
             docker(*args)
             # Give it time to start
@@ -97,44 +91,46 @@ class DatabaseProvider(object):
         0.0.0.0:49153
         0.0.0.0:49153\n:::49153
         """
-        host, port = docker(
-            'port', self.container_name, str(self.PORT)
-        ).splitlines()[0].split(':')
+        host, port = (
+            docker("port", self.container_name, str(self.PORT))
+            .splitlines()[0]
+            .split(":")
+        )
         port = int(port)
 
         return {
-            'ENGINE': engine,
-            'NAME': self.DATABASE,
-            'USER': self.USER,
-            'PASSWORD': password,
-            'HOST': host,
-            'PORT': port,
+            "ENGINE": engine,
+            "NAME": self.DATABASE,
+            "USER": self.USER,
+            "PASSWORD": password,
+            "HOST": host,
+            "PORT": port,
         }
 
 
 class PostgreSQL(DatabaseProvider):
     """Provide a PostgreSQL database via Docker."""
 
-    IMAGE = 'postgres'
+    IMAGE = "postgres"
     PORT = 5432
-    PASSWORD_ENV_VAR = 'POSTGRES_PASSWORD'
-    DATA_DIR = '/var/lib/postgresql/data'
-    ENGINE_MATCH = 'postgresql'
-    USER = 'postgres'
-    DATABASE = 'postgres'
+    PASSWORD_ENV_VAR = "POSTGRES_PASSWORD"
+    DATA_DIR = "/var/lib/postgresql/data"
+    ENGINE_MATCH = "postgresql"
+    USER = "postgres"
+    DATABASE = "postgres"
 
 
 class MySQL(DatabaseProvider):
     """Provide a MySQL database via Docker."""
 
-    IMAGE = 'mysql/mysql-server'
+    IMAGE = "mysql"
     PORT = 3306
-    PASSWORD_ENV_VAR = 'MYSQL_ROOT_PASSWORD'
+    PASSWORD_ENV_VAR = "MYSQL_ROOT_PASSWORD"
     CUSTOM_ENV = {"MYSQL_ROOT_HOST": "%"}
-    DATA_DIR = '/var/lib/mysql'
-    ENGINE_MATCH = 'mysql'
-    USER = 'root'
-    DATABASE = 'mysql'  # TODO: is this correct?
+    DATA_DIR = "/var/lib/mysql"
+    ENGINE_MATCH = "mysql"
+    USER = "root"
+    DATABASE = "mysql"  # TODO: is this correct?
 
 
 PROVIDERS = [
@@ -146,7 +142,7 @@ PROVIDERS = [
 ]
 
 
-def fast_test_database(databases, test_commands=('test',), version=None):
+def fast_test_database(databases, test_commands=("test",), version=None):
     """
     If running tests, start a database on a tmpfs and set it as the default
     connection.
@@ -156,7 +152,7 @@ def fast_test_database(databases, test_commands=('test',), version=None):
         # Not under test, leave connections alone
         return databases
 
-    engine = databases['default']['ENGINE']
+    engine = databases["default"]["ENGINE"]
 
     match = None
     for engine_match, provider in PROVIDERS:
@@ -168,10 +164,8 @@ def fast_test_database(databases, test_commands=('test',), version=None):
         return databases
 
     databases = copy(databases)
-    databases['default'] = match(version).provide(engine)
+    databases["default"] = match(version).provide(engine)
     return databases
 
 
-__all__ = (
-    'fast_test_database',
-)
+__all__ = ("fast_test_database",)

{fast-test-database-0.1.2 → fast_test_database-0.2.0}/pyproject.toml

@@ -1,19 +1,21 @@
 [tool.poetry]
 name = "fast-test-database"
-version = "v0.1.2"
+version = "0.2.0"
 description = "Configure an in-memory database for running Django tests"
 authors = ["Alexey Kotlyarov <a@koterpillar.com>"]
 license = "GPL-3.0-or-later"
 readme = "README.rst"
 
 [tool.poetry.dependencies]
-python = "^3.7"
-Django = "^3.0"
+python = "^3.11"
+Django = "^5.0.0"
 
 [tool.poetry.dev-dependencies]
-mysqlclient = "^2.0.3"
-psycopg2 = "^2.9.1"
-pycodestyle = "^2.7.0"
+black = "^24.2.0"
+isort = "^5.13.2"
+mysqlclient = "^2.2.4"
+psycopg2 = "^2.9.9"
+pycodestyle = "^2.11.1"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

fast-test-database-0.1.2/setup.py

@@ -1,30 +0,0 @@
-# -*- coding: utf-8 -*-
-from setuptools import setup
-
-packages = \
-['fast_test_database']
-
-package_data = \
-{'': ['*']}
-
-install_requires = \
-['Django>=3.0,<4.0']
-
-setup_kwargs = {
-    'name': 'fast-test-database',
-    'version': '0.1.2',
-    'description': 'Configure an in-memory database for running Django tests',
-    'long_description': "Fast test database\n==================\n\nUse a pure in-memory database for running Django tests.\n\nUsage\n-----\n\nIn ``settings.py``:\n\n.. code:: python\n\n    from fast_test_database import fast_test_database\n\n    DATABASES = fast_test_database(DATABASES)\n\n    # Or:\n    DATABASES = fast_test_database(DATABASES,\n                                   test_commands=('test', 'harvest'))\n\n    # Or:\n    DATABASES = fast_test_database(DATABASES,\n                                   version='5.7')\n\nThis will be a no-op except for ``./manage.py test``, when an in-memory\ndatabase will be automatically started and supplied to the application.\n\nDetails\n-------\n\nThe in-memory database is a full PostgreSQL or MySQL instance started\nusing Docker, using tmpfs for storing the data. A single container will\nbe started if not yet running. It will not be shut down automatically,\nand instead reused for subsequent tests.\n\nThe type of the database (PostgreSQL or MySQL) is chosen based on the\nexisting default database engine.\n\nThe default version of the database (PostgreSQL or MySQL) is latest.\nBut it can be specified by version parameter.\n",
-    'author': 'Alexey Kotlyarov',
-    'author_email': 'a@koterpillar.com',
-    'maintainer': None,
-    'maintainer_email': None,
-    'url': None,
-    'packages': packages,
-    'package_data': package_data,
-    'install_requires': install_requires,
-    'python_requires': '>=3.7,<4.0',
-}
-
-
-setup(**setup_kwargs)