Use standard environment variable name for bucket

engelke · engelke · commit 4df74166e0ae · 2020-02-18T12:46:35.000-08:00
diff --git a/appengine/standard/migration/storage/app.yaml b/appengine/standard/migration/storage/app.yaml
@@ -3,7 +3,7 @@ api_version: 1
 threadsafe: yes
 
 env_variables:
-    BUCKET_NAME: "REPLACE THIS WITH YOUR EXISTING BUCKET NAME"
+    CLOUD_STORAGE_BUCKET: "REPLACE THIS WITH YOUR EXISTING BUCKET NAME"
     BLOB_NAME: "my-demo-blob"
 
 libraries:
diff --git a/appengine/standard/migration/storage/app3.yaml b/appengine/standard/migration/storage/app3.yaml
@@ -1,5 +1,5 @@
 runtime: python37
 
 env_variables:
-    BUCKET_NAME: "REPLACE THIS WITH YOUR EXISTING BUCKET NAME"
+    CLOUD_STORAGE_BUCKET: "REPLACE THIS WITH YOUR EXISTING BUCKET NAME"
     BLOB_NAME: "my-demo-blob"
diff --git a/appengine/standard/migration/storage/main.py b/appengine/standard/migration/storage/main.py
@@ -23,7 +23,7 @@
 
 @app.route('/', methods=['GET'])
 def get():
-    bucket_name = os.environ['BUCKET_NAME']
+    bucket_name = os.environ['CLOUD_STORAGE_BUCKET']
     blob_name = os.environ['BLOB_NAME']
 
     client = storage.Client()
diff --git a/appengine/standard/migration/storage/main_test.py b/appengine/standard/migration/storage/main_test.py
@@ -24,8 +24,7 @@ def test_index():
     assert r.status_code == 200
     assert 'Downloaded text matches uploaded text' in r.data.decode('utf-8')
 
-    bucket_name = os.environ['BUCKET_NAME']
+    bucket_name = os.environ['CLOUD_STORAGE_BUCKET']
     blob_name = os.environ['BLOB_NAME']
     assert '    {}/{}'.format(bucket_name, blob_name) in r.data.decode('utf-8')
     assert 'Blob {} deleted.'.format(blob_name) in r.data.decode('utf-8')
-    
