def setUp(self): docker_dns.CONFIG = DEFAULT_CONFIG self.client = MockDockerClient() self.mapping = DockerMapping(self.client) self.resolver = DockerResolver(self.mapping)
def setUp(self): docker_dns.CONFIG = DEFAULT_CONFIG self.client = MockDockerClient() self.mapping = DockerMapping(self.client)
class DockerMappingTest(unittest.TestCase): def setUp(self): docker_dns.CONFIG = DEFAULT_CONFIG self.client = MockDockerClient() self.mapping = DockerMapping(self.client) # # TEST _ids_from_prop # def test__ids_from_prop_single_depth(self): ids_gen1, ids_gen2 = itertools.tee( self.mapping._ids_from_prop(['ID'], 'cidpandaslong')) self.assertEqual(sum(1 for _ in ids_gen1), 1) self.assertEqual(ids_gen2.next(), 'cidpandaslong') def test__ids_from_prop_multi_depth(self): ids_gen1, ids_gen2 = itertools.tee( self.mapping._ids_from_prop(['NetworkSettings', 'IPAddress'], '8.8.8.8')) self.assertEqual(sum(1 for _ in ids_gen1), 1) self.assertEqual(ids_gen2.next(), 'cidfoxeslong') def test__ids_from_prop_multi_match(self): # FIXME I can not believe how disgusting this is ids_gen1, ids_gen2, ids_gen3 = itertools.tee( self.mapping._ids_from_prop(['Same'], 'Value'), 3) self.assertEqual(sum(1 for _ in ids_gen1), 2) self.assertTrue(in_generator(ids_gen2, 'cidpandaslong')) self.assertTrue(in_generator(ids_gen3, 'cidfoxeslong')) # # TEST lookup_container # def test_lookup_container_hostname(self): self.assertEqual(self.mapping.lookup_container('cuddly-pandas'), self.client.inspect_container_pandas) def test_lookup_container_id(self): self.assertEqual(self.mapping.lookup_container('cidfoxes.docker'), self.client.inspect_container_foxes) def test_lookup_container_hostname_none(self): self.assertEqual( # Raises an APIError 404 self.mapping.lookup_container('invalid'), None) def test_lookup_container_id_none(self): self.assertEqual( # Raises an APIError 404 self.mapping.lookup_container('invalid.docker'), None) # # TEST get_a # def test_get_a_hostname(self): self.assertEqual(self.mapping.get_a('sneaky-foxes'), '8.8.8.8') def test_get_a_id(self): self.assertEqual(self.mapping.get_a('cidpandas.docker'), '127.0.0.1') def test_get_a_hostname_none(self): self.assertEqual(self.mapping.get_a('invalid'), None) def test_get_a_id_none(self): self.assertEqual(self.mapping.get_a('invalid.docker'), None)
class DockerMappingTest(unittest.TestCase): def setUp(self): docker_dns.CONFIG = DEFAULT_CONFIG self.client = MockDockerClient() self.mapping = DockerMapping(self.client) # # TEST _ids_from_prop # def test__ids_from_prop_single_depth(self): ids_gen1, ids_gen2 = itertools.tee( self.mapping._ids_from_prop( ['ID'], 'cidpandaslong' ) ) self.assertEqual(sum(1 for _ in ids_gen1), 1) self.assertEqual( ids_gen2.next(), 'cidpandaslong' ) def test__ids_from_prop_multi_depth(self): ids_gen1, ids_gen2 = itertools.tee( self.mapping._ids_from_prop( ['NetworkSettings', 'IPAddress'], '8.8.8.8' ) ) self.assertEqual(sum(1 for _ in ids_gen1), 1) self.assertEqual( ids_gen2.next(), 'cidfoxeslong' ) def test__ids_from_prop_multi_match(self): # FIXME I can not believe how disgusting this is ids_gen1, ids_gen2, ids_gen3 = itertools.tee( self.mapping._ids_from_prop( ['Same'], 'Value' ), 3 ) self.assertEqual(sum(1 for _ in ids_gen1), 2) self.assertTrue(in_generator(ids_gen2, 'cidpandaslong')) self.assertTrue(in_generator(ids_gen3, 'cidfoxeslong')) # # TEST lookup_container # def test_lookup_container_hostname(self): self.assertEqual( self.mapping.lookup_container('cuddly-pandas'), self.client.inspect_container_pandas ) def test_lookup_container_id(self): self.assertEqual( self.mapping.lookup_container('cidfoxes.docker'), self.client.inspect_container_foxes ) def test_lookup_container_hostname_none(self): self.assertEqual( # Raises an APIError 404 self.mapping.lookup_container('invalid'), None ) def test_lookup_container_id_none(self): self.assertEqual( # Raises an APIError 404 self.mapping.lookup_container('invalid.docker'), None ) # # TEST get_a # def test_get_a_hostname(self): self.assertEqual( self.mapping.get_a('sneaky-foxes'), '8.8.8.8' ) def test_get_a_id(self): self.assertEqual( self.mapping.get_a('cidpandas.docker'), '127.0.0.1' ) def test_get_a_hostname_none(self): self.assertEqual( self.mapping.get_a('invalid'), None ) def test_get_a_id_none(self): self.assertEqual( self.mapping.get_a('invalid.docker'), None )