SNOW-2160718 adjust ec2 IP in WIF tests, limit docker resources (#2503)

Author: sfc-gh-akolodziejczyk

ci/test_wif.sh

@@ -18,6 +18,8 @@ run_tests_and_set_result() {
       set -o pipefail
       docker run \
         --rm \
+        --cpus=1 \
+        -m 1g \
         -e BRANCH \
         -e SNOWFLAKE_TEST_WIF_PROVIDER \
         -e SNOWFLAKE_TEST_WIF_HOST \
@@ -48,11 +50,14 @@ EOF
 
 get_branch() {
   local branch
-  branch=$(git rev-parse --abbrev-ref HEAD)
-  if [[ "$branch" == "HEAD" ]]; then
-    branch=$(git name-rev --name-only HEAD | sed 's#^remotes/origin/##;s#^origin/##')
+  if [[ -n "${GIT_BRANCH}" ]]; then
+    # Jenkins
+    branch="${GIT_BRANCH}"
+  else
+    # Local
+    branch=$(git rev-parse --abbrev-ref HEAD)
   fi
-  echo "$branch"
+  echo "${branch}"
 }
 
 setup_parameters() {