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_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_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_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_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)
