class TestSolrDockerRunner(unittest.TestCase):
    """
    Test the docker runner specifically for Solr
    """

    @mock.patch('mc.builders.Client')
    def setUp(self, mocked):
        instance = mocked.return_value
        instance.create_container.return_value = {'Id': 'mocked'}
        instance.port.return_value = [{'HostIp': '127.0.0.1', 'HostPort': 1234}]
        instance.containers.return_value = [
            {
                u'Command': u'/entrypoint.sh redis-server',
                u'Created': 1443632967,
                u'Id': u'mocked',
                u'Image': u'redis',
                u'Labels': {},
                u'Names': [u'/livetest-redis-tLJpZ'],
                u'Ports': [{u'PrivatePort': 6379, u'Type': u'tcp'}],
                u'Status': u'Up About a minute'
            }
        ]
        self.instance = instance
        self.builder = SolrDockerRunner(network_mode="host")

    def test_ready(self):
        """
        Tests the health check of the service
        """

        with mock.patch.object(mc.builders.requests,
                               'get',
                               side_effect=requests.ConnectionError):
            self.assertFalse(self.builder.ready)

        with HTTMock(response_500):
            self.assertFalse(self.builder.ready)

        with HTTMock(response_404):
            self.assertTrue(self.builder.ready)

    @mock.patch('mc.builders.PostgresDockerRunner.service_provisioner')
    def test_can_provision(self, mocked):
        """
        Tests that there is a method that allows provisioning
        """
        self.builder.provision(services=['adsaws'])

        mocked.assert_has_calls([
            mock.call(container=self.builder, requirements=None, services=['adsaws']),
            mock.call()()
        ])

    @mock.patch('mc.builders.SolrDockerRunner.service_provisioner')
    def test_can_provision(self, mocked):
        """
        Tests that there is a method that allows provisioning
        """
        self.builder.provision(services=['recommender'])
        mocked.assert_has_calls([
            mock.call(container=self.builder, requirements=None, services=['recommender']),
            mock.call()()
        ])

    @mock.patch('mc.builders.SolrDockerRunner.service_provisioner')
    def test_does_not_raise_for_non_existing_service(self, mocked):
        """
        Tests that it skips services it does not know about
        """
        instance = mocked.return_value
        try:
            self.builder.provision(services=['unknown_service'])
        except Exception as e:
            self.fail('Provisioning failed: {}'.format(e))
        self.assertTrue(instance.called)
Beispiel #2
0
class TestSolrDockerRunner(unittest.TestCase):
    """
    Test the docker runner specifically for Solr
    """
    @mock.patch('mc.builders.Client')
    def setUp(self, mocked):
        instance = mocked.return_value
        instance.create_container.return_value = {'Id': 'mocked'}
        instance.port.return_value = [{
            'HostIp': '127.0.0.1',
            'HostPort': 1234
        }]
        instance.containers.return_value = [{
            u'Command':
            u'/entrypoint.sh redis-server',
            u'Created':
            1443632967,
            u'Id':
            u'mocked',
            u'Image':
            u'redis',
            u'Labels': {},
            u'Names': [u'/livetest-redis-tLJpZ'],
            u'Ports': [{
                u'PrivatePort': 6379,
                u'Type': u'tcp'
            }],
            u'Status':
            u'Up About a minute'
        }]
        self.instance = instance
        self.builder = SolrDockerRunner(network_mode="host")

    def test_ready(self):
        """
        Tests the health check of the service
        """

        with mock.patch.object(mc.builders.requests,
                               'get',
                               side_effect=requests.ConnectionError):
            self.assertFalse(self.builder.ready)

        with HTTMock(response_500):
            self.assertFalse(self.builder.ready)

        with HTTMock(response_404):
            self.assertTrue(self.builder.ready)

    @mock.patch('mc.builders.PostgresDockerRunner.service_provisioner')
    def test_can_provision(self, mocked):
        """
        Tests that there is a method that allows provisioning
        """
        self.builder.provision(services=['adsaws'])

        mocked.assert_has_calls([
            mock.call(container=self.builder,
                      requirements=None,
                      services=['adsaws']),
            mock.call()()
        ])

    @mock.patch('mc.builders.SolrDockerRunner.service_provisioner')
    def test_can_provision(self, mocked):
        """
        Tests that there is a method that allows provisioning
        """
        self.builder.provision(services=['recommender'])
        mocked.assert_has_calls([
            mock.call(container=self.builder,
                      requirements=None,
                      services=['recommender']),
            mock.call()()
        ])

    @mock.patch('mc.builders.SolrDockerRunner.service_provisioner')
    def test_does_not_raise_for_non_existing_service(self, mocked):
        """
        Tests that it skips services it does not know about
        """
        instance = mocked.return_value
        try:
            self.builder.provision(services=['unknown_service'])
        except Exception as e:
            self.fail('Provisioning failed: {}'.format(e))
        self.assertTrue(instance.called)