mirror of https://github.com/docker/compose.git
104 lines
3.5 KiB
Python
104 lines
3.5 KiB
Python
from plum import Service
|
|
from .testcases import DockerClientTestCase
|
|
|
|
|
|
class ServiceTest(DockerClientTestCase):
|
|
def test_name_validations(self):
|
|
self.assertRaises(ValueError, lambda: Service(name=''))
|
|
|
|
self.assertRaises(ValueError, lambda: Service(name=' '))
|
|
self.assertRaises(ValueError, lambda: Service(name='/'))
|
|
self.assertRaises(ValueError, lambda: Service(name='!'))
|
|
self.assertRaises(ValueError, lambda: Service(name='\xe2'))
|
|
|
|
Service('a')
|
|
Service('foo')
|
|
Service('foo_bar')
|
|
Service('__foo_bar__')
|
|
Service('_')
|
|
Service('_____')
|
|
|
|
def test_containers(self):
|
|
foo = self.create_service('foo')
|
|
bar = self.create_service('bar')
|
|
|
|
foo.start()
|
|
|
|
self.assertEqual(len(foo.containers()), 1)
|
|
self.assertEqual(foo.containers()[0].name, '/foo_1')
|
|
self.assertEqual(len(bar.containers()), 0)
|
|
|
|
bar.scale(2)
|
|
|
|
self.assertEqual(len(foo.containers()), 1)
|
|
self.assertEqual(len(bar.containers()), 2)
|
|
|
|
names = [c.name for c in bar.containers()]
|
|
self.assertIn('/bar_1', names)
|
|
self.assertIn('/bar_2', names)
|
|
|
|
def test_up_scale_down(self):
|
|
service = self.create_service('scaling_test')
|
|
self.assertEqual(len(service.containers()), 0)
|
|
|
|
service.start()
|
|
self.assertEqual(len(service.containers()), 1)
|
|
|
|
service.start()
|
|
self.assertEqual(len(service.containers()), 1)
|
|
|
|
service.scale(2)
|
|
self.assertEqual(len(service.containers()), 2)
|
|
|
|
service.scale(1)
|
|
self.assertEqual(len(service.containers()), 1)
|
|
|
|
service.stop()
|
|
self.assertEqual(len(service.containers()), 0)
|
|
|
|
service.stop()
|
|
self.assertEqual(len(service.containers()), 0)
|
|
|
|
def test_start_container_passes_through_options(self):
|
|
db = self.create_service('db')
|
|
db.start_container(environment={'FOO': 'BAR'})
|
|
self.assertEqual(db.containers()[0].environment['FOO'], 'BAR')
|
|
|
|
def test_start_container_inherits_options_from_constructor(self):
|
|
db = self.create_service('db', environment={'FOO': 'BAR'})
|
|
db.start_container()
|
|
self.assertEqual(db.containers()[0].environment['FOO'], 'BAR')
|
|
|
|
def test_start_container_creates_links(self):
|
|
db = self.create_service('db')
|
|
web = self.create_service('web', links=[db])
|
|
db.start_container()
|
|
web.start_container()
|
|
self.assertIn('db_1', web.containers()[0].links())
|
|
db.stop()
|
|
web.stop()
|
|
|
|
def test_start_container_builds_images(self):
|
|
service = Service(
|
|
name='test',
|
|
client=self.client,
|
|
build='tests/fixtures/simple-dockerfile',
|
|
)
|
|
container = service.start()
|
|
container.wait()
|
|
self.assertIn('success', container.logs())
|
|
|
|
def test_start_container_creates_ports(self):
|
|
service = self.create_service('web', ports=[8000])
|
|
container = service.start_container().inspect()
|
|
self.assertIn('8000/tcp', container['HostConfig']['PortBindings'])
|
|
self.assertNotEqual(container['HostConfig']['PortBindings']['8000/tcp'][0]['HostPort'], '8000')
|
|
|
|
def test_start_container_creates_fixed_external_ports(self):
|
|
service = self.create_service('web', ports=['8000:8000'])
|
|
container = service.start_container().inspect()
|
|
self.assertIn('8000/tcp', container['HostConfig']['PortBindings'])
|
|
self.assertEqual(container['HostConfig']['PortBindings']['8000/tcp'][0]['HostPort'], '8000')
|
|
|
|
|