Override docker host with env variable (#192)

* Add ability to redefine docker url with env variable * Inline * Add changelog + bump version --------- Co-authored-by: Iurii Pliner <[email protected]>
anna-money · Apr 3, 2024 · 365db9f · 365db9f
1 parent fc02877
commit 365db9f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,8 @@
+## v0.0.17 (2024-04-03)
+
+* Override docker host with env variable
+
+
 ## v0.0.15 (2023-05-04)
 
 * Pin version of `urllib3` ([see for more details](https://github.com/docker/docker-py/issues/3113))

diff --git a/pytest_pg/__init__.py b/pytest_pg/__init__.py
@@ -7,7 +7,7 @@
 
 __all__: Tuple[str, ...] = ("PG", "run_pg", "pg", "pg_11", "pg_12", "pg_13", "pg_14", "pg_15", "pg_16")
 
-__version__ = "0.0.16"
+__version__ = "0.0.17"
 
 version = f"{__version__}, Python {sys.version}"
 

diff --git a/pytest_pg/fixtures.py b/pytest_pg/fixtures.py
@@ -1,5 +1,6 @@
 import contextlib
 import dataclasses
+import os
 import time
 import uuid
 from typing import Generator
@@ -10,7 +11,6 @@
 from .utils import find_unused_local_port, is_pg_ready
 
 LOCALHOST = "127.0.0.1"
-
 DEFAULT_PG_USER = "postgres"
 DEFAULT_PG_PASSWORD = "mysecretpassword"
 DEFAULT_PG_DATABASE = "postgres"
@@ -27,7 +27,7 @@ class PG:
 
 @contextlib.contextmanager
 def run_pg(image: str, ready_timeout: float = 30.0) -> Generator[PG, None, None]:
-    docker_client = docker.APIClient(version="auto")
+    docker_client = docker.APIClient(base_url=os.getenv("DOCKER_HOST"), version="auto")
 
     docker_client.pull(image)