Beispiel #1
0
 def setUp(self):
     """Create a docker client"""
     super(DockerClientTestCase, self).setUp()
     self.client = PuppetClient()
     CONF.set_override('cmd_syntax',
                       "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_deploy', "cmdlaunch {}", group='clients_chef')
 def setUp(self):
     """Create a docker client"""
     super(DockerClientTestCase, self).setUp()
     self.client = PuppetClient()
     CONF.set_override(
         'cmd_syntax',
         "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_deploy', "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 = PuppetClient()
        CONF.set_override(
            'cmd_syntax',
            "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_deploy', "cmdlaunch {}", group='clients_chef')

    def test_create_client(self):
        """Test client creation"""
        self.assertRaises(DockerException, PuppetClient, '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()
Beispiel #4
0
class DockerClientTestCase(tb.ValidatorTestCase):
    """Docker Client unit tests"""
    def setUp(self):
        """Create a docker client"""
        super(DockerClientTestCase, self).setUp()
        self.client = PuppetClient()
        CONF.set_override('cmd_syntax',
                          "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_deploy', "cmdlaunch {}", group='clients_chef')

    def test_create_client(self):
        """Test client creation"""
        self.assertRaises(DockerException, PuppetClient, '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()