From 2b0d2453c2dc384725dd286763726805a282fa30 Mon Sep 17 00:00:00 2001 From: Danny Hermes Date: Sat, 30 Jan 2016 10:37:29 -0800 Subject: [PATCH] Fixing broken system test helpers. They were using the dataset ID environment variable key rather than the value. --- system_tests/clear_datastore.py | 2 +- system_tests/populate_datastore.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/system_tests/clear_datastore.py b/system_tests/clear_datastore.py index 85a7b08c1914..f544dd484cbb 100644 --- a/system_tests/clear_datastore.py +++ b/system_tests/clear_datastore.py @@ -90,7 +90,7 @@ def remove_kind(kind, client): def remove_all_entities(client=None): if client is None: # Get a client that uses the test dataset. - client = datastore.Client(project=TESTS_DATASET) + client = datastore.Client(project=os.getenv(TESTS_DATASET)) for kind in ALL_KINDS: remove_kind(kind, client) diff --git a/system_tests/populate_datastore.py b/system_tests/populate_datastore.py index 3c349fdff074..e4be6cfbadad 100644 --- a/system_tests/populate_datastore.py +++ b/system_tests/populate_datastore.py @@ -91,7 +91,7 @@ def print_func(message): def add_characters(client=None): if client is None: # Get a client that uses the test dataset. - client = datastore.Client(project=TESTS_DATASET) + client = datastore.Client(project=os.getenv(TESTS_DATASET)) with client.transaction() as xact: for key_path, character in zip(KEY_PATHS, CHARACTERS): if key_path[-1] != character['name']: