def test_service_registry_with_blank_dict(self): service_registry_ext_test = dict() dbaas_sr.get_custom_managers = Mock(return_value=service_registry_ext_test) test_dict = dbaas_sr.service_registry() self.assertEqual(2, len(test_dict)) self.assertEqual(test_dict.get("mysql"), "trove.guestagent.manager.mysql.Manager") self.assertEqual(test_dict.get("percona"), "trove.guestagent.manager.mysql.Manager")
def test_service_registry_with_extra_manager(self): service_registry_ext_test = {"test": "trove.guestagent.manager.test.Manager"} dbaas_sr.get_custom_managers = Mock(return_value=service_registry_ext_test) test_dict = dbaas_sr.service_registry() self.assertEqual(3, len(test_dict)) self.assertEqual(test_dict.get("test"), service_registry_ext_test.get("test")) self.assertEqual(test_dict.get("mysql"), "trove.guestagent.manager.mysql.Manager") self.assertEqual(test_dict.get("percona"), "trove.guestagent.manager.mysql.Manager")
def test_service_registry_with_existing_manager(self): service_registry_ext_test = { 'mysql': 'trove.guestagent.manager.mysql.Manager123', } dbaas_sr.get_custom_managers = Mock(return_value= service_registry_ext_test) test_dict = dbaas_sr.service_registry() self.assertEqual(2, len(test_dict)) self.assertEqual(test_dict.get('mysql'), 'trove.guestagent.manager.mysql.Manager123') self.assertEqual(test_dict.get('percona'), 'trove.guestagent.manager.mysql.Manager')
def test_service_registry_with_blank_dict(self): service_registry_ext_test = dict() dbaas_sr.get_custom_managers = Mock(return_value= service_registry_ext_test) test_dict = dbaas_sr.service_registry() self.assertEqual(2, len(test_dict)) self.assertEqual(test_dict.get('mysql'), 'trove.guestagent.datastore.mysql.' 'manager.Manager') self.assertEqual(test_dict.get('percona'), 'trove.guestagent.datastore.mysql.' 'manager.Manager')
def test_service_registry_with_extra_manager(self): service_registry_ext_test = { 'test': 'trove.guestagent.datastore.test.manager.Manager', } dbaas_sr.get_custom_managers = Mock(return_value= service_registry_ext_test) test_dict = dbaas_sr.service_registry() self.assertEqual(3, len(test_dict)) self.assertEqual(test_dict.get('test'), service_registry_ext_test.get('test', None)) self.assertEqual(test_dict.get('mysql'), 'trove.guestagent.datastore.mysql.' 'manager.Manager') self.assertEqual(test_dict.get('percona'), 'trove.guestagent.datastore.mysql.' 'manager.Manager')