12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182 |
- #!/usr/bin/python3
-
- import subprocess
- from nose.tools import with_setup
-
-
- def _run(cmd):
- subprocess.call(cmd,
- stdout=subprocess.DEVNULL,
- stderr=subprocess.DEVNULL)
-
-
- def setup():
- _run(['docker', 'run', '--name=testA', '-d', 'hello-world'])
- _run(['docker', 'run', '--name=testB', '-d', 'hello-world'])
-
-
- def teardown():
- _run(['docker', 'rm', '-f', 'testA'])
- _run(['docker', 'rm', '-f', 'testB'])
-
-
- def setup_each():
- _run(['docker', 'rm', '-f', 'test123'])
-
-
- def teardown_each():
- _run(['docker', 'rm', '-f', 'test123'])
-
-
- @with_setup(setup, teardown)
- def test_command_matches():
- yield check, ['docker', 'run', '--name=test123', '-d', 'hello-world']
- yield check, ['docker', 'run', '--name=test123', '-d', 'hello-world:latest']
- yield check, ['docker', 'run', '--name=test123', '-d', 'hello-world', '/hello', 'world...']
- yield check, ['docker', 'run', '--env=PATH=/root', '--env=Z=X', '--name=test123', '-d', 'hello-world']
- yield check, ['docker', 'run', '--env=FOO=bar baz', '--name=test123', '-d', 'hello-world']
- yield check, ['docker', 'run', '--name=test123', '--restart=always', '-d', 'hello-world']
- yield check, ['docker', 'run', '--name=test123', '--restart=on-failure:10', '-d', 'hello-world']
- yield check, ['docker', 'run', '--name=test123', '--net=host', '-d', 'hello-world']
- yield check, ['docker', 'run', '--name=test123', '-d', '-p=127.0.0.1:443:443', '-p=127.0.0.1::1336/udp', 'hello-world']
- yield check, ['docker', 'run', '--name=test123', '-d', '-p=443', 'hello-world']
- yield check, ['docker', 'run', '--name=test123', '--user=root', '-d', 'hello-world', '/hello', 'foobar']
- yield check, ['docker', 'run', '--name=test123', '--net=host', '--user=root:root', '-d', 'hello-world']
- yield check, ['docker', 'run', '--name=test123', '--volume=/dev/null:/null', '--volume=/dev/urandom:/mnt/random', '-d', 'hello-world']
- yield check, ['docker', 'run', '--label=com.example=123 456', '--name=test123', '-d', 'hello-world']
- yield check, ['docker', 'run', '--label=com.example.1', '--label=com.example.2=345', '--name=test123', '-d', 'hello-world']
- yield check, ['docker', 'run', '--link=testA:bar', '--link=testB', '--name=test123', '-d', 'hello-world']
-
-
- @with_setup(setup, teardown)
- def test_modifiers():
- yield (check,
- ['docker', 'run', '--name=test123', '-d', 'hello-world'],
- ['--image', 'tutum/hello-world'],
- ['docker', 'run', '--name=test123', '-d', 'tutum/hello-world'])
- yield (check,
- ['docker', 'run', '--name=test123', '-d', 'hello-world'],
- ['--image', 'hello-world:latest'],
- ['docker', 'run', '--name=test123', '-d', 'hello-world:latest'])
- yield (check,
- ['docker', 'run', '--name=test123', '-d', 'hello-world'],
- ['--port', '123:456'],
- ['docker', 'run', '--name=test123', '-d', '-p=123:456', 'hello-world'])
- yield (check,
- ['docker', 'run', '--name=test123', '-d', 'hello-world'],
- ['--port', '123:456', '-p', '80'],
- ['docker', 'run', '--name=test123', '-d', '-p=123:456', '-p=80', 'hello-world'])
- yield (check,
- ['docker', 'run', '--name=test123', '-d', 'hello-world'],
- ['--tag', 'latest'],
- ['docker', 'run', '--name=test123', '-d', 'hello-world:latest'])
-
-
- @with_setup(setup_each, teardown_each)
- def check(command, args=[], expected=None):
- _run(command)
- output = subprocess.check_output(['./docker-rerun', '--dry-run', 'test123'] + args)
- output = output.decode('utf-8').strip().splitlines()
- expected = ' '.join(expected or command)
- assert output[3] == expected, 'Expected "%s" but got "%s"' % (expected, output[3])
|