diff --git a/.github/trigger_files/beam_PostCommit_Python.json b/.github/trigger_files/beam_PostCommit_Python.json index 8b8054dac42b..4db5f0675328 100644 --- a/.github/trigger_files/beam_PostCommit_Python.json +++ b/.github/trigger_files/beam_PostCommit_Python.json @@ -1,5 +1,5 @@ { "comment": "Modify this file in a trivial way to cause this test suite to run.", - "modification": 36 + "modification": 37 } diff --git a/sdks/python/apache_beam/transforms/enrichment_tests_requirements.txt b/sdks/python/apache_beam/transforms/enrichment_tests_requirements.txt new file mode 100644 index 000000000000..eca8bbb58599 --- /dev/null +++ b/sdks/python/apache_beam/transforms/enrichment_tests_requirements.txt @@ -0,0 +1,18 @@ +# +# Licensed to the Apache Software Foundation (ASF) under one or more +# contributor license agreements. See the NOTICE file distributed with +# this work for additional information regarding copyright ownership. +# The ASF licenses this file to You under the Apache License, Version 2.0 +# (the "License"); you may not use this file except in compliance with +# the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +redis>=5.0.0,<6 \ No newline at end of file diff --git a/sdks/python/test-suites/dataflow/common.gradle b/sdks/python/test-suites/dataflow/common.gradle index 6a0777bd667c..7d65da6ee3ad 100644 --- a/sdks/python/test-suites/dataflow/common.gradle +++ b/sdks/python/test-suites/dataflow/common.gradle @@ -575,7 +575,7 @@ task tftTests { task mockAPITests { dependsOn 'initializeForDataflowJob' dependsOn ':sdks:python:sdist' - + def requirementsFile = "${rootDir}/sdks/python/apache_beam/transforms/enrichment_tests_requirements.txt" doLast { def testOpts = basicTestOpts def argMap = [ @@ -584,11 +584,12 @@ task mockAPITests { "runner": "TestDataflowRunner", "project": "apache-beam-testing", "region": "us-west1", + "requirements_file": "$requirementsFile" ] def cmdArgs = mapToArgString(argMap) exec { executable 'sh' - args '-c', ". ${envdir}/bin/activate && ${runScriptsDir}/run_integration_test.sh $cmdArgs" + args '-c', ". ${envdir}/bin/activate && pip install -r $requirementsFile && ${runScriptsDir}/run_integration_test.sh $cmdArgs" } } }