def test_find_sources(self, fake_manager):
type_id = 'test_1'
unit_key = 1
destination = '/tmp/123'
url = 'http://redhat.com/repository'
primary = PrimarySource(None)
alternatives = dict([(s, ContentSource(s, d)) for s, d in DESCRIPTOR])
fake_manager().find.return_value = CATALOG
# test
request = Request(type_id, unit_key, url, destination)
request.find_sources(primary, alternatives)
# validation
# validate sources sorted by priority with the primary last.
# should only have matched on s-1 and s-3.
request.sources = list(request.sources)
self.assertEqual(len(request.sources), 5)
self.assertEqual(request.sources[0][0].id, 's-3')
self.assertEqual(request.sources[0][1], CATALOG[2][constants.URL])
self.assertEqual(request.sources[1][0].id, 's-3')
self.assertEqual(request.sources[1][1], CATALOG[3][constants.URL])
self.assertEqual(request.sources[2][0].id, 's-1')
self.assertEqual(request.sources[2][1], CATALOG[0][constants.URL])
self.assertEqual(request.sources[3][0].id, 's-1')
self.assertEqual(request.sources[3][1], CATALOG[1][constants.URL])
self.assertEqual(request.sources[4][0].id, primary.id)
self.assertEqual(request.sources[4][1], url)
def test_find_sources(self, fake_manager):
type_id = 'test_1'
unit_key = 1
destination = '/tmp/123'
url = 'http://redhat.com/repository'
primary = PrimarySource(None)
alternatives = dict([(s, ContentSource(s, d)) for s, d in DESCRIPTOR])
fake_manager().find.return_value = CATALOG
# test
request = Request(type_id, unit_key, url, destination)
request.find_sources(primary, alternatives)
# validation
# validate sources sorted by priority with the primary last.
# should only have matched on s-1 and s-3.
request.sources = list(request.sources)
self.assertEqual(len(request.sources), 5)
self.assertEqual(request.sources[0][0].id, 's-3')
self.assertEqual(request.sources[0][1], CATALOG[2][constants.URL])
self.assertEqual(request.sources[1][0].id, 's-3')
self.assertEqual(request.sources[1][1], CATALOG[3][constants.URL])
self.assertEqual(request.sources[2][0].id, 's-1')
self.assertEqual(request.sources[2][1], CATALOG[0][constants.URL])
self.assertEqual(request.sources[3][0].id, 's-1')
self.assertEqual(request.sources[3][1], CATALOG[1][constants.URL])
self.assertEqual(request.sources[4][0].id, primary.id)
self.assertEqual(request.sources[4][1], url)
def test_next_source(self):
sources = [1, 2, 3]
request = Request('', {}, '', '')
request.sources = sources
# test and validation
for i, source in enumerate(request.sources):
self.assertEqual(source, sources[i])
def test_next_source(self):
sources = [1, 2, 3]
request = Request('', {}, '', '')
request.sources = sources
# test and validation
for i, source in enumerate(request.sources):
self.assertEqual(source, sources[i])
def test_has_source(self):
request = Request('', {}, '', '')
request.sources = [1]
# test and validation
self.assertTrue(request.has_source())
request.index = 1
self.assertFalse(request.has_source())
def test_has_source(self):
request = Request('', {}, '', '')
request.sources = [1]
# test and validation
self.assertTrue(request.has_source())
request.index = 1
self.assertFalse(request.has_source())
def test_next_source(self):
request = Request('', {}, '', '')
request.sources = [1, 2, 3]
# test and validation
for n in request.sources:
source = request.next_source()
self.assertEqual(source, n)
self.assertTrue(request.next_source() is None)
def test_next_source(self):
request = Request('', {}, '', '')
request.sources = [1, 2, 3]
# test and validation
for n in request.sources:
source = request.next_source()
self.assertEqual(source, n)
self.assertTrue(request.next_source() is None)
