Skip to content

Commit 03dfdf7

Browse files
committed
Changing use of PUBSUB_HOST env. var. to PUBSUB_EMULATOR_HOST.
Also changing how the variable is used. Now, pre-pending the protocol in front. (Assumes the user does something like `localhost:8888`.)
1 parent 03d4104 commit 03dfdf7

File tree

3 files changed

+8
-5
lines changed

3 files changed

+8
-5
lines changed

gcloud/environment_vars.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
GCD_HOST = 'DATASTORE_HOST'
3434
"""Environment variable defining host for GCD dataset server."""
3535

36-
PUBSUB_EMULATOR = 'PUBSUB_HOST'
36+
PUBSUB_EMULATOR = 'PUBSUB_EMULATOR_HOST'
3737
"""Environment variable defining host for Pub/Sub emulator."""
3838

3939
TESTS_DATASET = 'GCLOUD_TESTS_DATASET_ID'

gcloud/pubsub/connection.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,11 @@ class Connection(base_connection.JSONConnection):
5151
def __init__(self, credentials=None, http=None, api_base_url=None):
5252
super(Connection, self).__init__(credentials=credentials, http=http)
5353
if api_base_url is None:
54-
api_base_url = os.getenv(PUBSUB_EMULATOR,
55-
self.__class__.API_BASE_URL)
54+
emulator_host = os.getenv(PUBSUB_EMULATOR)
55+
if emulator_host is None:
56+
api_base_url = self.__class__.API_BASE_URL
57+
else:
58+
api_base_url = 'http://' + emulator_host
5659
self.api_base_url = api_base_url
5760

5861
def build_api_url(self, path, query_params=None,

gcloud/pubsub/test_connection.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,15 +34,15 @@ def test_custom_url_from_env(self):
3434
from gcloud._testing import _Monkey
3535
from gcloud.environment_vars import PUBSUB_EMULATOR
3636

37-
HOST = object()
37+
HOST = 'localhost:8187'
3838
fake_environ = {PUBSUB_EMULATOR: HOST}
3939

4040
with _Monkey(os, getenv=fake_environ.get):
4141
conn = self._makeOne()
4242

4343
klass = self._getTargetClass()
4444
self.assertNotEqual(conn.api_base_url, klass.API_BASE_URL)
45-
self.assertEqual(conn.api_base_url, HOST)
45+
self.assertEqual(conn.api_base_url, 'http://' + HOST)
4646

4747
def test_custom_url_from_constructor(self):
4848
HOST = object()

0 commit comments

Comments
 (0)