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)