def setUp(self):
self.service = 'my-db'
self.service_ip = '172.18.0.23'
self.nport = 3306
self.sport = 'mysql'
self.service_port = 27384
self.secret = 'my-secret'
self.secret_value = 'my=value'
self.client = docker.Client()
self.client.containers = MagicMock(return_value=[
dict(
Ports=[
dict(Type="tcp", PrivatePort=self.nport, PublicPort=self.service_port)
],
Names=['mock-container']
)
])
self.client.inspect_container = MagicMock(return_value = dict(
Config=dict(
Env=[
"MY_SECRET=my=value",
]
)
))
self.r = DockerRegistry(self.client, self.service_ip)
class DockerRegistryTest(TestCase):
def setUp(self):
self.service = 'my-db'
self.service_ip = '172.18.0.23'
self.nport = 3306
self.sport = 'mysql'
self.service_port = 27384
self.secret = 'my-secret'
self.secret_value = 'my=value'
self.client = docker.Client()
self.client.containers = MagicMock(return_value=[
dict(
Ports=[
dict(Type="tcp", PrivatePort=self.nport, PublicPort=self.service_port)
],
Names=['mock-container']
)
])
self.client.inspect_container = MagicMock(return_value = dict(
Config=dict(
Env=[
"MY_SECRET=my=value",
]
)
))
self.r = DockerRegistry(self.client, self.service_ip)
def test_register_creates_service(self):
svc = self.r.register(self.service, self.nport, self.sport)
self.assertIsNotNone(svc)
def test_registered_service_resolves_host(self):
svc = self.r.register(self.service, self.nport, self.sport)
self.assertEqual(svc.host, self.service_ip)
def test_registered_service_resolves_port(self):
svc = self.r.register(self.service, self.nport, self.sport)
self.assertEqual(svc.port, self.service_port)
def test_register_fails_if_no_service_match(self):
self.client.containers = MagicMock(return_value=[])
self.assertRaises(ValueError, self.r.register, 'service', 8000, 'http')
def test_register_fails_if_no_port_match(self):
self.assertRaises(ValueError, self.r.register, self.service, self.nport + 1, self.sport)
def test_register_fails_if_not_protocol_match(self):
self.assertRaises(ValueError, self.r.register, self.service, self.nport, self.sport, protocol="udp")
def test_register_resolves_named_ports(self):
svc = self.r.register(self.service, self.nport, self.sport)
self.assertEqual(svc.port, self.service_port)
def test_service_secret_resolves_secret(self):
svc = self.r.register(self.service, self.nport, self.sport, secrets=[self.secret])
self.assertEquals(svc.secrets[self.secret], self.secret_value)
def test_register_fails_on_missing_service_secret(self):
self.assertRaises(ValueError, self.r.register, self.service, self.nport, self.sport, secrets=['unknown-secret'])
