Merge pull request #170 from nkubala/environment_fix

return string instead of bool in integration test sample app env handler
GoogleCloudPlatform · Nov 15, 2017 · b759359 · b759359
2 parents 4a50266 + 714dcb7
commit b759359
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/integration/server.py b/tests/integration/server.py
@@ -245,8 +245,11 @@ def _check_environment():
     # for GAE, we'll check the existence env vars set on
     # vm:true or env:flex
     # if neither exist, assume we're in GKE
-    return (_APPENGINE_FLEXIBLE_ENV_VM in os.environ or
-            _APPENGINE_FLEXIBLE_ENV_FLEX in os.environ), 200
+    environment = "GKE"
+    if (_APPENGINE_FLEXIBLE_ENV_VM in os.environ or
+        _APPENGINE_FLEXIBLE_ENV_FLEX in os.environ):
+        environment = "GAE"
+    return environment, 200
 
 
 class ErrorResponse(Exception):