Fix go integration tests on flink

damccorm · damccorm · commit 1769be95fd7e · 2022-03-14T13:54:22.000-04:00
diff --git a/sdks/go/test/resources/flink-conf.yaml b/sdks/go/test/resources/flink-conf.yaml
@@ -0,0 +1,20 @@
+# 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.
+#
+
+taskmanager.memory.network.fraction: 0.2
+taskmanager.memory.network.max: 2gb
diff --git a/sdks/go/test/run_validatesrunner_tests.sh b/sdks/go/test/run_validatesrunner_tests.sh
@@ -73,6 +73,9 @@ set -e
 trap '! [[ "$BASH_COMMAND" =~ ^(echo|read|if|ARGS|shift|SOCKET_SCRIPT|\[\[) ]] && \
 cmd=`eval echo "$BASH_COMMAND" 2>/dev/null` && echo "\$ $cmd"' DEBUG
 
+# Resolve current directory
+CURRENT_DIRECTORY=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
+
 # Default test targets.
 TESTS="./test/integration/... ./test/regression"
 
@@ -271,6 +274,7 @@ elif [[ "$RUNNER" == "flink" || "$RUNNER" == "spark" || "$RUNNER" == "samza" ||
       java \
           -jar $FLINK_JOB_SERVER_JAR \
           --flink-master [local] \
+          --flink-conf-dir $CURRENT_DIRECTORY/resources \
           --job-port $JOB_PORT \
           --expansion-port 0 \
           --artifact-port 0 &
