def testGetCloudSDKRoot(self):
class SDKRootValue(object):
def __init__(self, sdk_root=None):
# Accessed as a property.
self.sdk_root = sdk_root
# Limit the patch scope to this method, because TearDown() logic from other
# modules also accesses util.config.Paths().
with mock.patch.object(util.config, 'Paths') as paths_mock:
paths_mock.return_value = SDKRootValue()
with self.assertRaises(util.NoCloudSDKError):
util.GetCloudSDKRoot()
paths_mock.return_value = SDKRootValue('valid_root')
self.assertEqual('valid_root', util.GetCloudSDKRoot())
def _BuildStartArgsForNativeExecutable(args):
spanner_executable = os.path.join(util.GetCloudSDKRoot(), 'bin',
SPANNER_EMULATOR_EXECUTABLE_DIR,
SPANNER_EMULATOR_EXECUTABLE_FILE)
return execution_utils.ArgsForExecutableTool(
spanner_executable, '--hostname', args.host_port.host, '--grpc_port',
args.host_port.port, '--http_port', six.text_type(args.rest_port))
def GetGCDRoot():
"""Gets the directory of the GCD emulator installation in the Cloud SDK.
Raises:
NoCloudSDKError: If there is no SDK root.
NoGCDError: If the GCD installation dir does not exist.
Returns:
str, The path to the root of the GCD emulator installation within Cloud SDK.
"""
sdk_root = util.GetCloudSDKRoot()
gcd_dir = os.path.join(sdk_root, 'platform', 'cloud-datastore-emulator')
if not os.path.isdir(gcd_dir):
raise NoGCDError()
return gcd_dir
def GetFirestoreEmulatorRoot():
"""Gets the directory of the Firestore emulator installation in the Cloud SDK.
Raises:
NoCloudSDKError: If there is no SDK root.
NoFirestoreEmulatorError: If the installation dir does not exist.
Returns:
str, The path to the root of the Firestore emulator installation within
Cloud SDK.
"""
sdk_root = util.GetCloudSDKRoot()
firestore_dir = os.path.join(sdk_root, 'platform', 'cloud-firestore-emulator')
if not os.path.isdir(firestore_dir):
raise NoFirestoreEmulatorError()
return firestore_dir
def GetGCDRoot(args):
"""Gets the directory of the GCD emulator installation in the Cloud SDK.
Args:
args: Arguments passed to the command.
Raises:
NoCloudSDKError: If there is no SDK root.
NoGCDError: If the GCD installation dir does not exist.
Returns:
str, The path to the root of the GCD emulator installation within Cloud SDK.
"""
sdk_root = util.GetCloudSDKRoot()
if args.legacy:
gcd_dir = os.path.join(sdk_root, 'platform', 'gcd')
else:
gcd_dir = os.path.join(sdk_root, 'platform', 'cloud-datastore-emulator')
if not os.path.isdir(gcd_dir):
raise NoGCDError()
return gcd_dir
def __init__(self, address, config_file=None, broker_dir=None):
"""Constructor.
Args:
address: (str) The host or host-port of the broker server. The server may
already be running.
config_file: (str) The full path to the broker config file.
broker_dir: (str) A custom path to the broker directory.
"""
if config_file is not None:
assert os.path.isabs(config_file)
self._address = address
self._config_file = config_file
if broker_dir:
self._broker_dir = broker_dir
else:
self._broker_dir = os.path.join(util.GetCloudSDKRoot(), 'bin', 'broker')
self._host_port = arg_parsers.HostPort.Parse(address)
self._current_platform = platforms.Platform.Current()
self._process = None
self._comm_thread = None