def testAlwaysExists(self):
kwargs = copy.copy(_BASE_SNOWFLAKE_SPEC)
spec = benchmark_config_spec._EdwServiceSpec('NAME', **kwargs)
snowflake_local = snowflake.Snowflake(spec)
self.assertTrue(snowflake_local._Exists())
def testRunCommandHelper(self):
kwargs = copy.copy(_BASE_SNOWFLAKE_SPEC)
spec = benchmark_config_spec._EdwServiceSpec('NAME', **kwargs)
snowflake_local = snowflake.Snowflake(spec)
self.assertEqual('--connection fake_connection',
snowflake_local.RunCommandHelper())
def testCreateRequestError(self):
kwargs = copy.copy(_BASE_SNOWFLAKE_SPEC)
spec = benchmark_config_spec._EdwServiceSpec('NAME', **kwargs)
snowflake_local = snowflake.Snowflake(spec)
with self.assertRaises(NotImplementedError):
snowflake_local._Create()
def testInstallAndAuthenticateRunner(self):
kwargs = copy.copy(_BASE_SNOWFLAKE_SPEC)
spec = benchmark_config_spec._EdwServiceSpec('NAME', **kwargs)
snowflake_local = snowflake.Snowflake(spec)
snowflake_local.InstallAndAuthenticateRunner(
vm=FakeRemoteVMCreateLambdaRole(), benchmark_name='fake_benchmark')
