Add system tests for pubsub and storage GCF (GoogleCloudPlatform#1584)

andrewsg · web-flow · commit 7d9e38020b6e · 2018-07-19T14:04:43.000-07:00
diff --git a/functions/helloworld/sample_pubsub_test_system.py b/functions/helloworld/sample_pubsub_test_system.py
@@ -0,0 +1,64 @@
+# Copyright 2018 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.
+
+# [START functions_pubsub_system_test]
+from datetime import datetime
+from os import getenv
+import subprocess
+import time
+import uuid
+
+from google.cloud import pubsub_v1
+import pytest
+
+PROJECT = getenv('GCLOUD_PROJECT')
+TOPIC = getenv('TOPIC')
+
+assert PROJECT is not None
+assert TOPIC is not None
+
+
+@pytest.fixture(scope='module')
+def publisher_client():
+    yield pubsub_v1.PublisherClient()
+
+
+def test_print_name(publisher_client):
+    start_time = datetime.utcnow().isoformat()
+    topic_path = publisher_client.topic_path(PROJECT, TOPIC)
+
+    # Publish the message
+    name = uuid.uuid4()
+    data = str(name).encode('utf-8')
+    publisher_client.publish(topic_path, data=data).result()
+
+    # Wait for logs to become consistent
+    time.sleep(15)
+
+    # Check logs after a delay
+    log_process = subprocess.Popen([
+        'gcloud',
+        'alpha',
+        'functions',
+        'logs',
+        'read',
+        'hello_pubsub',
+        '--start-time',
+        start_time
+    ], stdout=subprocess.PIPE)
+    logs = str(log_process.communicate()[0])
+    print logs
+    print start_time
+    assert 'Hello, {}!'.format(name) in logs
+# [END functions_pubsub_system_test]
diff --git a/functions/helloworld/sample_storage_test_system.py b/functions/helloworld/sample_storage_test_system.py
@@ -0,0 +1,70 @@
+# Copyright 2018 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.
+
+# [START functions_storage_system_test]
+from datetime import datetime
+from os import getenv, path
+import subprocess
+import time
+import uuid
+
+from google.cloud import storage
+import pytest
+
+PROJECT = getenv('GCLOUD_PROJECT')
+BUCKET = getenv('BUCKET')
+
+assert PROJECT is not None
+assert BUCKET is not None
+
+
+@pytest.fixture(scope='module')
+def storage_client():
+    yield storage.Client()
+
+
+@pytest.fixture(scope='module')
+def bucket_object(storage_client):
+    bucket_object = storage_client.get_bucket(BUCKET)
+    yield bucket_object
+
+
+@pytest.fixture(scope='module')
+def uploaded_file(bucket_object):
+    name = 'test-{}.txt'.format(str(uuid.uuid4()))
+    blob = bucket_object.blob(name)
+
+    test_dir = path.dirname(path.abspath(__file__))
+    blob.upload_from_filename(path.join(test_dir, 'test.txt'))
+    yield name
+    blob.delete()
+
+
+def test_hello_gcs(uploaded_file):
+    start_time = datetime.utcnow().isoformat()
+    time.sleep(10)  # Wait for logs to become consistent
+
+    log_process = subprocess.Popen([
+        'gcloud',
+        'alpha',
+        'functions',
+        'logs',
+        'read',
+        'hello_gcs',
+        '--start-time',
+        start_time
+    ], stdout=subprocess.PIPE)
+    logs = str(log_process.communicate()[0])
+    assert uploaded_file in logs
+# [END functions_storage_system_test]
