def test_get_types(self): # Setup type_def_1 = TypeDefinition('type-1', 'Type 1', 'Type 1', [], [], []) type_def_2 = TypeDefinition('type-2', 'Type 2', 'Type 2', [], [], []) types_db._create_or_update_type(type_def_1) types_db._create_or_update_type(type_def_2) # Test status, body = self.get('/v2/plugins/types/') # Verify self.assertEqual(200, status) self.assertEqual(2, len(body)) # the order isn't guaranteed, so we adjust here if necessary if body[0]['id'] == 'type-2': body.reverse() self.assertEqual(body[0]['_href'], '/v2/plugins/types/type-1/') self.assertEqual(body[0]['id'], 'type-1') self.assertEqual(body[0]['display_name'], 'Type 1') self.assertEqual(body[1]['_href'], '/v2/plugins/types/type-2/') self.assertEqual(body[1]['id'], 'type-2') self.assertEqual(body[1]['display_name'], 'Type 2')
def test_create_or_update_type_collection(self): """ Tests the call to create a new type collection works. """ # Setup type_def = TypeDefinition('rpm', 'RPM', 'RPM Packages', ['name'], ['name'], []) # Test types_db._create_or_update_type(type_def) # Verify # Present in types collection all_types = list(ContentType.get_collection().find()) self.assertEqual(1, len(all_types)) found = all_types[0] self.assertEqual(type_def.id, found['id']) self.assertEqual(type_def.display_name, found['display_name']) self.assertEqual(type_def.description, found['description']) self.assertEqual(type_def.unit_key, found['unit_key']) self.assertEqual(type_def.search_indexes, found['search_indexes']) # Type collection exists collection_name = types_db.unit_collection_name(type_def.id) self.assertTrue( collection_name in pulp_db.database().collection_names())
def test_create_or_update_existing_type_collection(self): """ Tests calling create_or_update with a change to an existing type collection is successful. """ # Setup type_def = TypeDefinition('rpm', 'RPM', 'RPM Packages', ['name'], ['name'], []) types_db._create_or_update_type(type_def) # Test type_def.display_name = 'new-name' type_def.description = 'new-description' type_def.unit_key = 'new-key' type_def.search_indexes = None types_db._create_or_update_type(type_def) # Verify # Present in types collection all_types = list(ContentType.get_collection().find()) self.assertEqual(1, len(all_types)) found = all_types[0] self.assertEqual(type_def.id, found['id']) self.assertEqual(type_def.display_name, found['display_name']) self.assertEqual(type_def.description, found['description']) self.assertEqual(type_def.unit_key, found['unit_key']) self.assertEqual(type_def.search_indexes, found['search_indexes']) # Type collection exists collection_name = types_db.unit_collection_name(type_def.id) self.assertTrue( collection_name in pulp_db.database().collection_names())
def test_create_or_update_type_collection(self): """ Tests the call to create a new type collection works. """ # Setup type_def = TypeDefinition('rpm', 'RPM', 'RPM Packages', ['name'], ['name'], []) # Test types_db._create_or_update_type(type_def) # Verify # Present in types collection all_types = list(ContentType.get_collection().find()) self.assertEqual(1, len(all_types)) found = all_types[0] self.assertEqual(type_def.id, found['id']) self.assertEqual(type_def.display_name, found['display_name']) self.assertEqual(type_def.description, found['description']) self.assertEqual(type_def.unit_key, found['unit_key']) self.assertEqual(type_def.search_indexes, found['search_indexes']) # Type collection exists collection_name = types_db.unit_collection_name(type_def.id) self.assertTrue(collection_name in pulp_db.get_database().collection_names())
def test_create_or_update_existing_type_collection(self): """ Tests calling create_or_update with a change to an existing type collection is successful. """ # Setup type_def = TypeDefinition('rpm', 'RPM', 'RPM Packages', ['name'], ['name'], []) types_db._create_or_update_type(type_def) # Test type_def.display_name = 'new-name' type_def.description = 'new-description' type_def.unit_key = 'new-key' type_def.search_indexes = None types_db._create_or_update_type(type_def) # Verify # Present in types collection all_types = list(ContentType.get_collection().find()) self.assertEqual(1, len(all_types)) found = all_types[0] self.assertEqual(type_def.id, found['id']) self.assertEqual(type_def.display_name, found['display_name']) self.assertEqual(type_def.description, found['description']) self.assertEqual(type_def.unit_key, found['unit_key']) self.assertEqual(type_def.search_indexes, found['search_indexes']) # Type collection exists collection_name = types_db.unit_collection_name(type_def.id) self.assertTrue(collection_name in pulp_db.get_database().collection_names())
def test_types(self): """ Tests retrieving all types in the database. """ # Setup type_def_1 = TypeDefinition('type-1', 'Type 1', 'Type 1', [], [], []) type_def_2 = TypeDefinition('type-2', 'Type 2', 'Type 2', [], [], []) types_db._create_or_update_type(type_def_1) types_db._create_or_update_type(type_def_2) # Test found_defs = self.manager.types() # Verify self.assertEqual(2, len(found_defs)) for type_def in [type_def_1, type_def_2]: found_def = [t for t in found_defs if t['id'] == type_def.id][0] self.assertEqual(found_def['id'], type_def.id) self.assertEqual(found_def['display_name'], type_def.display_name) self.assertEqual(found_def['description'], type_def.description) self.assertEqual(found_def['unit_key'], type_def.unit_key) self.assertEqual(found_def['search_indexes'], type_def.search_indexes) self.assertEqual(found_def['referenced_types'], type_def.referenced_types)
def test_create_or_update_existing_type_collection(self): """ Tests calling create_or_update with a change to an existing type collection is successful. """ # Setup type_def = TypeDefinition("rpm", "RPM", "RPM Packages", ["name"], ["name"], []) types_db._create_or_update_type(type_def) # Test type_def.display_name = "new-name" type_def.description = "new-description" type_def.unit_key = "new-key" type_def.search_indexes = None types_db._create_or_update_type(type_def) # Verify # Present in types collection all_types = list(ContentType.get_collection().find()) self.assertEqual(1, len(all_types)) found = all_types[0] self.assertEqual(type_def.id, found["id"]) self.assertEqual(type_def.display_name, found["display_name"]) self.assertEqual(type_def.description, found["description"]) self.assertEqual(type_def.unit_key, found["unit_key"]) self.assertEqual(type_def.search_indexes, found["search_indexes"]) # Type collection exists collection_name = types_db.unit_collection_name(type_def.id) self.assertTrue(collection_name in pulp_db.get_database().collection_names())
def test_types(self): """ Tests retrieving all types in the database. """ # Setup types_db.clean() type_def_1 = TypeDefinition('type-1', 'Type 1', 'Type 1', [], [], []) type_def_2 = TypeDefinition('type-2', 'Type 2', 'Type 2', [], [], []) types_db._create_or_update_type(type_def_1) types_db._create_or_update_type(type_def_2) # Test found_defs = self.manager.types() # Verify self.assertEqual(2, len(found_defs)) for type_def in [type_def_1, type_def_2]: found_def = [t for t in found_defs if t['id'] == type_def.id][0] self.assertEqual(found_def['id'], type_def.id) self.assertEqual(found_def['display_name'], type_def.display_name) self.assertEqual(found_def['description'], type_def.description) self.assertEqual(found_def['unit_key'], type_def.unit_key) self.assertEqual(found_def['search_indexes'], type_def.search_indexes) self.assertEqual(found_def['referenced_types'], type_def.referenced_types)
def test_get_type(self): # Setup type_def_1 = TypeDefinition('type-1', 'Type 1', 'Type 1', [], [], []) types_db._create_or_update_type(type_def_1) # Test status, body = self.get('/v2/plugins/types/type-1/') # Verify self.assertEqual(200, status) self.assertEqual(body['_href'], '/v2/plugins/types/type-1/') self.assertEqual(body['id'], 'type-1') self.assertEqual(body['display_name'], 'Type 1')
def test_type_units_unit_key(self): """ Tests the syntactic sugar method for retrieving unit key on a type. """ # Setup type_def = TypeDefinition('rpm', 'RPM', 'RPM Packages', ['unique_1', 'unique_2'], ['name'], []) types_db._create_or_update_type(type_def) # Test unit_key = types_db.type_units_unit_key('rpm') # Verify self.assertEqual(type_def.unit_key, unit_key)
def test_type_units_unit_key(self): """ Tests the syntactic sugar method for retrieving unit key on a type. """ # Setup type_def = TypeDefinition("rpm", "RPM", "RPM Packages", ["unique_1", "unique_2"], ["name"], []) types_db._create_or_update_type(type_def) # Test unit_key = types_db.type_units_unit_key("rpm") # Verify self.assertEqual(type_def.unit_key, unit_key)
def test_all_type_collection_names(self): """ Tests listing all type collections. """ # Setup type_def = TypeDefinition('rpm', 'RPM', 'RPM Packages', ['name'], ['name'], []) types_db._create_or_update_type(type_def) # Test all_names = types_db.all_type_collection_names() # Verify self.assertEqual(1, len(all_names)) self.assertEqual(types_db.unit_collection_name(type_def.id), all_names[0])
def test_all_type_ids(self): """ Tests listing all type IDs. """ # Setup types_db._create_or_update_type(TypeDefinition('a', 'A', 'A', [], [], [])) types_db._create_or_update_type(TypeDefinition('b', 'B', 'B', [], [], [])) # Test type_ids = types_db.all_type_ids() # Verify self.assertEqual(2, len(type_ids)) self.assertTrue('a' in type_ids) self.assertTrue('b' in type_ids)