Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly
Appearance settings

Commit c24be57

Browse filesBrowse files
cwxie-googlegguuss
cwxie-google
authored and
gguuss
committed
Asset: Update Real Time Feed API unit tests (GoogleCloudPlatform#2467)
* Add Real Time Feed API Sample Code * Update Real Time Feed API unit tests
1 parent 40a1010 commit c24be57
Copy full SHA for c24be57

File tree

Expand file treeCollapse file tree

5 files changed

+25
-1
lines changed
Filter options
Expand file treeCollapse file tree

5 files changed

+25
-1
lines changed

‎asset/cloud-client/quickstart_createfeed_test.py

Copy file name to clipboardExpand all lines: asset/cloud-client/quickstart_createfeed_test.py
+5Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import quickstart_createfeed
2222
import quickstart_deletefeed
2323
from google.cloud import resource_manager
24+
from google.cloud import pubsub_v1
2425

2526
json_data = open(os.environ["GOOGLE_APPLICATION_CREDENTIALS"]).read()
2627
data = json.loads(json_data)
@@ -34,6 +35,9 @@ def test_create_feed(capsys):
3435
client = resource_manager.Client()
3536
project_number = client.fetch_project(PROJECT).number
3637
full_topic_name = "projects/{}/topics/{}".format(PROJECT, TOPIC)
38+
publisher = pubsub_v1.PublisherClient()
39+
topic_path = publisher.topic_path(PROJECT, TOPIC)
40+
publisher.create_topic(topic_path)
3741
quickstart_createfeed.create_feed(
3842
PROJECT, FEED_ID, [ASSET_NAME, ], full_topic_name)
3943
out, _ = capsys.readouterr()
@@ -42,3 +46,4 @@ def test_create_feed(capsys):
4246
# Clean up, delete the feed
4347
feed_name = "projects/{}/feeds/{}".format(project_number, FEED_ID)
4448
quickstart_deletefeed.delete_feed(feed_name)
49+
publisher.delete_topic(topic_path)

‎asset/cloud-client/quickstart_deletefeed_test.py

Copy file name to clipboardExpand all lines: asset/cloud-client/quickstart_deletefeed_test.py
+5Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import quickstart_createfeed
2121
import quickstart_deletefeed
2222
from google.cloud import resource_manager
23+
from google.cloud import pubsub_v1
2324

2425
PROJECT = os.environ['GCLOUD_PROJECT']
2526
ASSET_NAME = 'assets-{}'.format(int(time.time()))
@@ -32,6 +33,9 @@ def test_delete_feed(capsys):
3233
project_number = client.fetch_project(PROJECT).number
3334
# First create the feed, which will be deleted later
3435
full_topic_name = "projects/{}/topics/{}".format(PROJECT, TOPIC)
36+
publisher = pubsub_v1.PublisherClient()
37+
topic_path = publisher.topic_path(PROJECT, TOPIC)
38+
publisher.create_topic(topic_path)
3539
quickstart_createfeed.create_feed(
3640
PROJECT, FEED_ID, [ASSET_NAME, ], full_topic_name)
3741

@@ -40,3 +44,4 @@ def test_delete_feed(capsys):
4044

4145
out, _ = capsys.readouterr()
4246
assert "deleted_feed" in out
47+
publisher.delete_topic(topic_path)

‎asset/cloud-client/quickstart_getfeed_test.py

Copy file name to clipboardExpand all lines: asset/cloud-client/quickstart_getfeed_test.py
+5Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import quickstart_deletefeed
2222
import quickstart_getfeed
2323
from google.cloud import resource_manager
24+
from google.cloud import pubsub_v1
2425

2526
PROJECT = os.environ['GCLOUD_PROJECT']
2627
ASSET_NAME = 'assets-{}'.format(int(time.time()))
@@ -33,6 +34,9 @@ def test_get_feed(capsys):
3334
project_number = client.fetch_project(PROJECT).number
3435
# First create the feed, which will be gotten later
3536
full_topic_name = "projects/{}/topics/{}".format(PROJECT, TOPIC)
37+
publisher = pubsub_v1.PublisherClient()
38+
topic_path = publisher.topic_path(PROJECT, TOPIC)
39+
publisher.create_topic(topic_path)
3640
quickstart_createfeed.create_feed(
3741
PROJECT, FEED_ID, [ASSET_NAME, ], full_topic_name)
3842

@@ -43,3 +47,4 @@ def test_get_feed(capsys):
4347
assert "gotten_feed" in out
4448
# Clean up and delete the feed
4549
quickstart_deletefeed.delete_feed(feed_name)
50+
publisher.delete_topic(topic_path)

‎asset/cloud-client/quickstart_updatefeed_test.py

Copy file name to clipboardExpand all lines: asset/cloud-client/quickstart_updatefeed_test.py
+8Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import quickstart_deletefeed
2222
import quickstart_updatefeed
2323
from google.cloud import resource_manager
24+
from google.cloud import pubsub_v1
2425

2526
PROJECT = os.environ['GCLOUD_PROJECT']
2627
ASSET_NAME = 'assets-{}'.format(int(time.time()))
@@ -34,14 +35,21 @@ def test_update_feed(capsys):
3435
project_number = client.fetch_project(PROJECT).number
3536
# First create the feed, which will be updated later
3637
full_topic_name = "projects/{}/topics/{}".format(PROJECT, TOPIC)
38+
publisher = pubsub_v1.PublisherClient()
39+
topic_path = publisher.topic_path(PROJECT, TOPIC)
40+
publisher.create_topic(topic_path)
3741
quickstart_createfeed.create_feed(
3842
PROJECT, FEED_ID, [ASSET_NAME, ], full_topic_name)
3943

4044
feed_name = "projects/{}/feeds/{}".format(project_number, FEED_ID)
4145
new_full_topic_name = "projects/" + PROJECT + "/topics/" + NEW_TOPIC
46+
new_topic_path = publisher.topic_path(PROJECT, NEW_TOPIC)
47+
publisher.create_topic(new_topic_path)
4248
quickstart_updatefeed.update_feed(feed_name, new_full_topic_name)
4349
out, _ = capsys.readouterr()
4450

4551
assert "updated_feed" in out
4652
# Clean up and delete the feed
4753
quickstart_deletefeed.delete_feed(feed_name)
54+
publisher.delete_topic(topic_path)
55+
publisher.delete_topic(new_topic_path)

‎asset/cloud-client/requirements.txt

Copy file name to clipboard
+2-1Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
11
google-cloud-storage==1.18.0
22
google-cloud-asset==0.4.1
3-
google-cloud-resource-manager==0.29.2
3+
google-cloud-resource-manager==0.29.2
4+
google-cloud-pubsub=1.0.2

0 commit comments

Comments
0 (0)
Morty Proxy This is a proxified and sanitized view of the page, visit original site.