test: add tests for authenticating-users and gce samples (GoogleCloudPlatform#492)

engelke · web-flow · commit 5eba607563ab · 2022-08-08T21:12:03.000-04:00
* Feature: add test

* Fix: more tests
diff --git a/authenticating-users/main_test.py b/authenticating-users/main_test.py
@@ -0,0 +1,41 @@
+# Copyright 2022 Google LLC
+#
+# Licensed 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.
+
+import main
+
+
+def fake_validate(assertion):
+    if assertion == "Valid":
+        return "nobody@example.com", "user0001"
+    else:
+        return None, None
+
+
+main.validate_assertion = fake_validate
+
+
+def test_home_page():
+    client = main.app.test_client()
+
+    # Good request check
+    r = client.get("/", headers={"X-Goog-IAP-JWT-Assertion": "Valid"})
+    assert "nobody@example.com" in r.text
+
+    # Missing header check
+    r = client.get("/")
+    assert "None" in r.text
+
+    # Bad header check
+    r = client.get("/", headers={"X-Goog-IAP-JWT-Assertion": "Not Valid"})
+    assert "None" in r.text
diff --git a/authenticating-users/requirements-test.txt b/authenticating-users/requirements-test.txt
@@ -0,0 +1 @@
+pytest==7.1.2
diff --git a/gce/main_test.py b/gce/main_test.py
@@ -0,0 +1,25 @@
+# Copyright 2022 Google LLC
+#
+# Licensed 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.
+
+import main
+
+
+def test_hello():
+    main.app.testing = True
+    client = main.app.test_client()
+
+    r = client.get("/")
+    assert r.status_code == 200
+    response_text = r.data.decode("utf-8")
+    assert "Hello, world!" in response_text
