def CreateManagedDbFromSpec(spec_dict):
mock_db_spec = mock.Mock(
spec=benchmark_config_spec._ManagedRelationalDbSpec)
mock_db_spec.configure_mock(**spec_dict)
db_class = gcp_managed_relational_db.GCPManagedRelationalDb(mock_db_spec)
CreateMockClientVM(db_class)
return db_class
def testDelete(self):
with PatchCriticalObjects() as issue_command:
db = gcp_managed_relational_db.GCPManagedRelationalDb(self.mock_db_spec)
db._Delete()
self.assertEqual(issue_command.call_count, 1)
command_string = ' '.join(issue_command.call_args[0][0])
self.assertTrue(
command_string.startswith(
'gcloud sql instances delete pkb-db-instance-123'))
def CreateManagedDbFromSpec(spec_dict):
mock_db_spec = mock.Mock(
spec=benchmark_config_spec._ManagedRelationalDbSpec)
mock_db_spec.configure_mock(**spec_dict)
mocked_flags = mock_flags.MockFlags()
mocked_flags.run_uri = 'mock-run-uri'
with mock_flags.PatchFlags(mocked_flags):
db_class = gcp_managed_relational_db.GCPManagedRelationalDb(mock_db_spec)
return db_class
def testCreate(self):
with self._PatchCriticalObjects() as issue_command:
vm = gcp_managed_relational_db.GCPManagedRelationalDb(
self.mock_db_spec)
vm._Create()
self.assertEquals(issue_command.call_count, 1)
command_string = ' '.join(issue_command.call_args[0][0])
self.assertTrue(
command_string.startswith(
'gcloud beta sql instances create pkb-db-instance-123'),
command_string)
self.assertIn('--project fakeproject', command_string)
self.assertIn('--tier=db-n1-standard-1', command_string)
self.assertIn('--storage-size=50', command_string)
self.assertIn('--backup', command_string)
self.assertIn('--backup-start-time=07:00', command_string)
def testCreateWithBackupDisabled(self):
with PatchCriticalObjects() as issue_command:
spec = self.mock_db_spec
spec.backup_enabled = False
db = gcp_managed_relational_db.GCPManagedRelationalDb(self.mock_db_spec)
CreateMockClientVM(db)
db._Create()
self.assertEqual(issue_command.call_count, 1)
command_string = ' '.join(issue_command.call_args[0][0])
self.assertTrue(
command_string.startswith(
'gcloud beta sql instances create pkb-db-instance-123'),
command_string)
self.assertIn('--project fakeproject', command_string)
self.assertIn('--tier=db-n1-standard-1', command_string)
self.assertIn('--no-backup', command_string)
self.assertNotIn('--backup-start-time=07:00', command_string)
