Fixing broken system test helpers.

They were using the dataset ID environment variable key rather than the value.
dhermes · Jan 30, 2016 · 2b0d245 · 2b0d245
1 parent ddc8f85
commit 2b0d245
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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']: