Added tests

engelke · engelke · commit 16f190087cae · 2020-02-18T13:59:05.000-08:00
diff --git a/appengine/standard/migration/urlfetch/async/main_test.py b/appengine/standard/migration/urlfetch/async/main_test.py
@@ -1 +1,32 @@
+# Copyright 2020 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
+import os
+
+
+def test_index():
+    main.app.testing = True
+    client = main.app.test_client()
+
+    r = client.get('/')
+    assert r.status_code == 200
+    assert 'Google is built by a large team ' in r.data.decode('utf-8')
+
+    r = client.get('/callback')
+    assert r.status_code == 200
+    assert 'Response number 1 is ' in r.data.decode('utf-8')
+    assert 'Response number 2 is ' in r.data.decode('utf-8')
+    assert 'Response number 3 is ' in r.data.decode('utf-8')
+    assert 'Response number 4 is ' in r.data.decode('utf-8')
diff --git a/appengine/standard/migration/urlfetch/requests/main.py b/appengine/standard/migration/urlfetch/requests/main.py
@@ -41,4 +41,9 @@ def server_error(e):
     An internal error occurred: <pre>{}</pre>
     See logs for full stacktrace.
     """.format(e), 500
-# [END app]
+# [END app]
+
+
+if __name__ == '__main__':
+    # This is used when running locally.
+    app.run(host='127.0.0.1', port=8080, debug=True)
diff --git a/appengine/standard/migration/urlfetch/requests/main_test.py b/appengine/standard/migration/urlfetch/requests/main_test.py
@@ -0,0 +1,25 @@
+# Copyright 2020 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
+import os
+
+
+def test_index():
+    main.app.testing = True
+    client = main.app.test_client()
+
+    r = client.get('/')
+    assert r.status_code == 200
+    assert 'Google is built by a large team ' in r.data.decode('utf-8')
