def test_load_with_env(self):
"""Tests loading app manifest with resource allocation."""
manifest = {
'services': [
{
'name': 'web_server',
'command': '/bin/sleep 5',
'restart': {
'limit': 3,
'interval': 60,
}
}
],
'proid': 'foo',
'environment': 'dev',
'disk': '100G',
'cpu': '100%',
'memory': '100M',
'environ': [
{'name': 'xxx', 'value': 'yyy'}
],
}
treadmill.appcfg.manifest.read.return_value = manifest
event_filename0 = os.path.join(self.root, 'proid.myapp#0')
app0 = configure.load_runtime_manifest(
self.tm_env, event_filename0, 'linux')
self.assertEqual(app0['environ'], [{'name': 'xxx', 'value': 'yyy'}])
def test_load_tar_image(self):
"""Tests loading app manifest with a docker image defined."""
manifest = {
'services': [{
'name': 'web_server',
'command': '/bin/sleep 5',
'restart': {
'limit': 3,
'interval': 60,
}
}],
'proid':
'foo',
'environment':
'dev',
'disk':
'100G',
'cpu':
'100%',
'memory':
'100M',
'image':
'http://test'
}
treadmill.appcfg.manifest.read.return_value = manifest
event_filename0 = os.path.join(self.root, 'proid.myapp#0')
app0 = configure.load_runtime_manifest(self.tm_env, event_filename0,
'linux')
self.assertEqual(app0['image'], 'http://test')
self.assertEqual(app0['type'], 'tar')
manifest = {
'proid': 'foo',
'disk': '100G',
'cpu': '100%',
'memory': '100M',
'image': 'http://test'
}
treadmill.appcfg.manifest.read.return_value = manifest
with self.assertRaises(exc.InvalidInputError):
configure.load_runtime_manifest(self.tm_env, event_filename0,
'linux')
def test_load_docker_image(self):
"""Tests loading app manifest with a docker image defined."""
manifest = {
'proid': 'foo',
'environment': 'dev',
'disk': '100G',
'cpu': '100%',
'memory': '100M',
'image': 'docker://test'
}
treadmill.appcfg.manifest.read.return_value = manifest
event_filename0 = os.path.join(self.root, 'proid.myapp#0')
app0 = configure.load_runtime_manifest(
self.tm_env, event_filename0, 'docker')
self.assertEqual(app0['image'], 'docker://test')
self.assertEqual(app0['type'], 'docker')
def test_load(self):
"""Tests loading app manifest with resource allocation."""
manifest = {
'services': [
{
'name': 'web_server',
'command': '/bin/sleep 5',
'restart': {
'limit': 3,
'interval': 60,
},
},
],
'proid': 'foo',
'environment': 'dev',
'disk': '100G',
'cpu': '100%',
'memory': '100M'
}
treadmill.appcfg.manifest.read.return_value = manifest
event_filename0 = os.path.join(self.root, 'proid.myapp#0')
app0 = configure.load_runtime_manifest(
self.tm_env, event_filename0, 'linux')
treadmill.appcfg.manifest.read.assert_called_with(event_filename0,
'yaml')
treadmill.appcfg.gen_uniqueid.assert_called_with(event_filename0)
self.assertEqual(app0['name'], 'proid.myapp#0')
self.assertEqual(app0['type'], 'native')
self.assertEqual(app0['app'], 'proid.myapp')
self.assertEqual(app0['proid'], 'foo')
self.assertEqual(app0['environment'], 'dev')
self.assertEqual(app0['cpu'], 100)
self.assertEqual(app0['memory'], '100M')
self.assertEqual(app0['disk'], '100G')
self.assertEqual(app0['uniqueid'], '42')
self.assertEqual(app0['cell'], 'testcell')
self.assertEqual(app0['zookeeper'], 'zookeeper://foo@foo:123')
self.assertEqual(
app0['services'][0],
{
'proid': 'foo',
'root': False,
'name': 'web_server',
'command': '/bin/sleep 5',
'restart': {
'limit': 3,
'interval': 60,
},
'environ': [],
'config': None,
'downed': False,
'trace': True,
'logger': 's6.app-logger.run',
},
)
self.assertTrue(
any(x['name'] == 'sshd' for x in app0['services'])
)
self.assertTrue(
any(x['name'] == 'register' for x in app0['system_services'])
)
self.assertTrue(
any(x['name'] == 'hostaliases' for x in app0['system_services'])
)
self.assertTrue(
any(x['name'] == 'start_container'
for x in app0['system_services'])
)
self.assertEqual(
app0['endpoints'],
[
{
'name': 'ssh',
'port': 0,
'proto': 'tcp',
'type': 'infra',
},
]
)
def test_load_normalize(self):
"""Test the normalization of manifests.
"""
manifest = {
'services': [
{
'name': 'test1',
'command': '/bin/sleep 5',
'restart': {
'limit': 3,
'interval': 60,
}
},
{
'name': 'test2',
'command': '/bin/sleep 5',
'restart': {
'limit': 3,
'interval': 60,
}
},
],
'endpoints': [
{
'name': 'test1',
'proto': 'udp',
'port': 12345,
},
{
'name': 'test2',
'port': '0',
},
{
'name': 'test3',
'port': 32,
'type': 'infra',
},
],
'ephemeral_ports': {
'tcp': '2',
},
'proid': 'foo',
'environment': 'dev',
'disk': '100G',
'cpu': '100%',
'memory': '100M'
}
treadmill.appcfg.manifest.read.return_value = manifest
event_filename0 = os.path.join(self.root, 'proid.myapp#0')
app0 = configure.load_runtime_manifest(
self.tm_env, event_filename0, 'linux'
)
self.assertEqual(app0['type'], 'native')
self.assertEqual(
app0['services'],
[
{
'proid': 'foo',
'name': 'test1',
'command': '/bin/sleep 5',
'restart': {
'limit': 3,
'interval': 60,
},
'root': False,
'environ': [],
'config': None,
'downed': False,
'trace': True,
'logger': 's6.app-logger.run',
},
{
'proid': 'foo',
'name': 'test2',
'command': '/bin/sleep 5',
'restart': {
'limit': 3,
'interval': 60,
},
'root': False,
'environ': [],
'config': None,
'downed': False,
'trace': True,
'logger': 's6.app-logger.run',
},
{
'command': (
'exec /path/to/sshd'
' -D -f /etc/ssh/sshd_config'
' -p $TREADMILL_ENDPOINT_SSH'
),
'name': 'sshd',
'proid': 'root',
'restart': {
'limit': 5,
'interval': 60,
},
'root': True,
'environ': [],
'config': None,
'downed': False,
'trace': False,
},
]
)
self.assertEqual(app0['shared_ip'], False)
self.assertEqual(app0['shared_network'], False)
self.assertEqual(
app0['endpoints'],
[
{
'name': 'test1',
'type': None,
'port': 12345,
'proto': 'udp',
},
{
'name': 'test2',
'type': None,
'port': 0,
'proto': 'tcp',
},
{
'name': 'test3',
'type': 'infra',
'port': 32,
'proto': 'tcp',
},
{
'name': 'ssh',
'type': 'infra',
'port': 0,
'proto': 'tcp',
},
]
)
self.assertEqual(app0['passthrough'], [])
self.assertEqual(app0['vring']['cells'], [])
self.assertEqual(app0['identity'], None)
self.assertEqual(app0['identity_group'], None)
self.assertEqual(app0['environ'], [])
self.assertEqual(app0['ephemeral_ports'], {'tcp': 2, 'udp': 0})