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)