diff --git a/ingestion/tests/integration/postgres/conftest.py b/ingestion/tests/integration/postgres/conftest.py index 566b4c43542..bb7ff898fd2 100644 --- a/ingestion/tests/integration/postgres/conftest.py +++ b/ingestion/tests/integration/postgres/conftest.py @@ -1,4 +1,5 @@ import os +import tarfile import zipfile from subprocess import CalledProcessError @@ -11,9 +12,10 @@ def postgres_container(tmp_path_factory): data_dir = tmp_path_factory.mktemp("data") dvd_rental_zip = os.path.join(os.path.dirname(__file__), "data", "dvdrental.zip") zipfile.ZipFile(dvd_rental_zip, "r").extractall(str(data_dir)) + with tarfile.open(data_dir / "dvdrental_data.tar", "w") as tar: + tar.add(data_dir / "dvdrental.tar", arcname="dvdrental.tar") container = PostgresContainer("postgres:15", dbname="dvdrental") - container.volumes = {str(data_dir): {"bind": "/data"}} container._command = [ "-c", "shared_preload_libraries=pg_stat_statements", @@ -25,6 +27,10 @@ def postgres_container(tmp_path_factory): with container as container: docker_container = container.get_wrapped_container() + docker_container.exec_run(["mkdir", "/data"]) + docker_container.put_archive( + "/data/", open(data_dir / "dvdrental_data.tar", "rb") + ) for query in ( "CREATE USER postgres SUPERUSER;", "CREATE EXTENSION pg_stat_statements;",