test: support running in both enviro

Signed-off-by: Vladislav Polyakov <[email protected]>

Author: polRk

.devcontainer/compose.yml

@@ -28,6 +28,7 @@ services:
       - YDB_CONNECTION_STRING=grpc://ydb:2136/local
       - YDB_CONNECTION_STRING_SECURE=grpcs://ydb:2135/local
       - YDB_SSL_ROOT_CERTIFICATES_FILE=/ydb_certs/ca.pem
+      - TEST_COMPOSE_FILE=compose.remote.yml
 
     # Overrides default command so things don't shut down after the process ends.
     command: sleep infinity

compose.remote.yml

@@ -0,0 +1,22 @@
+networks:
+  default:
+    name: ydb-python-sdk_default
+    external: true
+
+services:
+  py-sdk-ydb:
+    image: ydbplatform/local-ydb:trunk
+    restart: always
+    hostname: py-sdk-ydb
+    ports:
+      - 2135
+      - 2136
+      - 8765
+    volumes:
+    - ./ydb_certs:/ydb_certs
+    environment:
+      - YDB_USE_IN_MEMORY_PDISKS=true
+      - YDB_ENABLE_COLUMN_TABLES=true
+      - GRPC_TLS_PORT=2135
+      - GRPC_PORT=2136
+      - MON_PORT=8765

compose.yml

@@ -1,13 +1,8 @@
-networks:
-  default:
-    external: true
-    name: ydb-python-sdk_default
-
 services:
   py-sdk-ydb:
     image: ydbplatform/local-ydb:trunk
     restart: always
-    hostname: py-sdk-ydb
+    hostname: localhost
     ports:
       - 2135
       - 2136

tests/conftest.py

@@ -8,7 +8,11 @@
 
 @pytest.fixture(scope="module")
 def docker_compose_file(pytestconfig):
-    return os.path.join(str(pytestconfig.rootdir), "compose.yml")
+    f = "compose.yml"
+    if os.environ.get('REMOTE_CONTAINERS') is not None:
+        f = "compose.remote.yml"
+
+    return os.path.join(str(pytestconfig.rootdir), f)
 
 
 def wait_container_ready(driver):

tox.ini

@@ -10,9 +10,11 @@ install_command = pip install {opts} {packages}
 setenv =
     PYTHONPATH = {env:PYTHONPATH}{:}{toxinidir}
     REMOTE_CONTAINERS = {env:REMOTE_CONTAINERS}
+    TEST_COMPOSE_FILE = {env:TEST_COMPOSE_FILE:compose.yml}
 deps =
     -r{toxinidir}/test-requirements.txt
 
+
 [testenv:dev-proto5]
 commands =
 deps =
@@ -33,14 +35,14 @@ deps =
 
 [testenv:py]
 commands =
-    pytest -v -m "not tls" --docker-compose-remove-volumes --docker-compose=compose.yml {posargs}
+    pytest -v -m "not tls" --docker-compose-remove-volumes --docker-compose={env:TEST_COMPOSE_FILE} {posargs}
 deps =
     -r{toxinidir}/test-requirements.txt
     protobuf<6.0.0
 
 [testenv:py-proto4]
 commands =
-    pytest -v -m "not tls" --docker-compose-remove-volumes --docker-compose=compose.yml {posargs}
+    pytest -v -m "not tls" --docker-compose-remove-volumes --docker-compose={env:TEST_COMPOSE_FILE} {posargs}
 deps =
     -r{toxinidir}/test-requirements.txt
     protobuf<5.0.0
@@ -49,35 +51,35 @@ deps =
 commands =
     pytest -v -m "not tls" \
         --cov-report html:cov_html --cov=ydb \
-        --docker-compose-remove-volumes --docker-compose=compose.yml {posargs}
+        --docker-compose-remove-volumes --docker-compose={env:TEST_COMPOSE_FILE} {posargs}
 deps =
     -r{toxinidir}/test-requirements.txt
     protobuf<5.0.0
 
 [testenv:py-proto3]
 commands =
-    pytest -v -m "not tls" --docker-compose-remove-volumes --docker-compose=compose.yml {posargs}
+    pytest -v -m "not tls" --docker-compose-remove-volumes --docker-compose={env:TEST_COMPOSE_FILE} {posargs}
 deps =
     -r{toxinidir}/test-requirements.txt
     protobuf<4.0.0
 
 [testenv:py-tls]
 commands =
-    pytest -v -m tls --docker-compose-remove-volumes --docker-compose=compose.yml {posargs}
+    pytest -v -m tls --docker-compose-remove-volumes --docker-compose={env:TEST_COMPOSE_FILE} {posargs}
 deps =
     -r{toxinidir}/test-requirements.txt
     protobuf<6.0.0
 
 [testenv:py-tls-proto4]
 commands =
-    pytest -v -m tls --docker-compose-remove-volumes --docker-compose=compose.yml {posargs}
+    pytest -v -m tls --docker-compose-remove-volumes --docker-compose={env:TEST_COMPOSE_FILE} {posargs}
 deps =
     -r{toxinidir}/test-requirements.txt
     protobuf<5.0.0
 
 [testenv:py-tls-proto3]
 commands =
-    pytest -v -m tls --docker-compose-remove-volumes --docker-compose=compose.yml {posargs}
+    pytest -v -m tls --docker-compose-remove-volumes --docker-compose={env:TEST_COMPOSE_FILE} {posargs}
 deps =
     -r{toxinidir}/test-requirements.txt
     protobuf<4.0.0