AVRO-3832: [Python] Make Python test work with Docker (apache#2444)

* AVRO-3832: [Python] Make Python test work with Docker. * Use contextlib.closing.
RanbirK · May 13, 2024 · 370fcaa · 370fcaa
1 parent 67a5a95
commit 370fcaa
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/build.sh b/build.sh
@@ -341,7 +341,7 @@ do
 
     docker-test)
       tar -cf- share/docker/Dockerfile $DOCKER_EXTRA_CONTEXT |
-        docker build -t avro-test -f share/docker/Dockerfile -
+        DOCKER_BUILDKIT=1 docker build -t avro-test -f share/docker/Dockerfile -
       docker run --rm -v "${PWD}:/avro${DOCKER_MOUNT_FLAG}" --env "JAVA=${JAVA:-8}" avro-test /avro/share/docker/run-tests.sh
       ;;
 

diff --git a/lang/py/avro/test/gen_interop_data.py b/lang/py/avro/test/gen_interop_data.py
@@ -23,6 +23,7 @@
 import io
 import json
 import os
+from contextlib import closing
 from pathlib import Path
 from typing import IO, TextIO
 
@@ -93,7 +94,8 @@ def _parse_args() -> argparse.Namespace:
 
 def main() -> int:
     args = _parse_args()
-    generate(args.schema_path, args.output_path)
+    with closing(args.output_path) as op:
+        generate(args.schema_path, op)
     return 0
 
 

diff --git a/share/docker/Dockerfile b/share/docker/Dockerfile
@@ -67,7 +67,9 @@ RUN apt-get -qqy update \
                                                  python3-wheel \
                                                  python3.10 \
                                                  python3.11 \
+                                                 python3.11-dev \
                                                  python3.7 \
+                                                 python3.7-distutils \
                                                  python3.8 \
                                                  python3.9 \
                                                  source-highlight \