def setUp(self):
"""Create a docker client"""
super(DockerClientTestCase, self).setUp()
self.client = DockerClient()
CONF.set_override(
'cmd_test',
"knife cookbook test {cookbook_name}",
group='clients_chef')
CONF.set_override(
'cmd_install',
" knife cookbook site install {cookbook_name}",
group='clients_chef')
CONF.set_override('cmd_inject', "cmdinject {}", group='clients_chef')
CONF.set_override('cmd_launch', "cmdlaunch {}", group='clients_chef')
class DockerClientTestCase(tb.ValidatorTestCase):
"""Docker Client unit tests"""
def setUp(self):
"""Create a docker client"""
super(DockerClientTestCase, self).setUp()
self.client = DockerClient()
CONF.set_override(
'cmd_test',
"knife cookbook test {cookbook_name}",
group='clients_chef')
CONF.set_override(
'cmd_install',
" knife cookbook site install {cookbook_name}",
group='clients_chef')
CONF.set_override('cmd_inject', "cmdinject {}", group='clients_chef')
CONF.set_override('cmd_launch', "cmdlaunch {}", group='clients_chef')
def test_create_client(self):
"""Test client creation"""
self.assertRaises(DockerException, DockerClient, 'fakeurl')
self.assertIsInstance(self.client.dc, docker.client.Client)
def test_run_container(self):
"""Test container deployment"""
self.client.dc = mock.MagicMock()
self.client.run_container('validimage')
self.client.dc.create_container.assert_called_once_with(
'validimage',
name=u'validimage-validate',
tty=True)
self.client.dc.start.assert_called_once_with(
container=self.client.container
)
def test_stop_container(self):
"""Test stopping and removing a container"""
self.client.dc = self.m.CreateMockAnything()
self.client.dc.stop(self.client.container)
self.client.dc.remove_container(self.client.container)
self.m.ReplayAll()
self.client.remove_container()
self.m.VerifyAll()
def test_run_deploy(self):
self.client.remove_container = mock.MagicMock()
self.client.execute_command = mock.MagicMock()
self.client.execute_command.return_value = "Alls good"
obs = self.client.run_deploy("mycookbook", "myrecipe")
expected = "{'response': u'Alls good', 'success': True}"
self.assertEqual(expected, str(obs))
def test_run_install(self):
self.client.container = "1234"
self.client.remove_container = mock.MagicMock()
self.client.execute_command = mock.MagicMock()
self.client.execute_command.return_value = "Alls good"
cookbook = "testing"
obs = self.client.run_install(cookbook)
expected = "{'response': u'Alls good', 'success': True}"
self.assertEqual(expected, str(obs))
def test_run_test(self):
self.client.remove_container = mock.MagicMock()
self.client.execute_command = mock.MagicMock()
self.client.execute_command.return_value = "Alls good"
test_input = "testing"
cookbook = test_input
obs = self.client.run_test(cookbook)
expected = "{'response': u'Alls good', 'success': True}"
self.assertEqual(expected, str(obs))
self.m.VerifyAll()
def test_execute_command(self):
"""Test a command execution in container"""
self.client.dc = self.m.CreateMockAnything()
self.client.container = "1234"
self.client.dc.exec_create(
cmd='/bin/bash -c "mycommand"',
container=u'1234'
).AndReturn("validcmd")
self.client.dc.exec_start("validcmd").AndReturn("OK")
self.m.ReplayAll()
obs = self.client.execute_command("mycommand")
self.assertEqual("OK", obs)
self.m.VerifyAll()
def tearDown(self):
"""Cleanup environment"""
super(DockerClientTestCase, self).tearDown()
self.m.UnsetStubs()
self.m.ResetAll()
