Python PodServiceMapper._fill_services_cache Examples

Example #1

File: test_kube_pod_service_mapper.py Project: dblackdblack/dd-agent

 def test_pod_to_service_no_match(self):
     jsons = self._load_json_array(['service_cache_services2.json'])
     with patch.object(self.kube, 'retrieve_json_auth', side_effect=jsons):
         mapper = PodServiceMapper(self.kube)
         mapper._fill_services_cache()
         no_match = self._build_pod_metadata(0, {'app': 'unknown'})
         self.assertEqual(0, len(mapper.match_services_for_pod(no_match)))

Example #2

File: test_kube_pod_service_mapper.py Project: vitbyst/dd-agent

 def test_pod_to_service_no_match(self):
     jsons = self._load_json_array(['service_cache_services2.json'])
     with patch.object(self.kube, 'retrieve_json_auth', side_effect=jsons):
         mapper = PodServiceMapper(self.kube)
         mapper._fill_services_cache()
         no_match = self._build_pod_metadata(0, {'app': 'unknown'})
         self.assertEqual(0, len(mapper.match_services_for_pod(no_match)))

Example #3

File: test_kube_pod_service_mapper.py Project: trilogy-group/kayako-dd-agent

    def test_403_disable(self):
        exception403 = requests.exceptions.HTTPError()
        exception403.response = Mock()
        exception403.response.status_code = 403
        self.assertEquals(403, exception403.response.status_code)
        self.assertTrue(isinstance(exception403,
                                   requests.exceptions.HTTPError))

        with patch.object(self.kube,
                          'retrieve_json_auth',
                          side_effect=exception403) as request_mock:
            # Fill pod label cache
            mapper = PodServiceMapper(self.kube)
            self.assertEqual(0, mapper._403_errors)

            for i in range(0, MAX_403_RETRIES):
                self.assertFalse(mapper._403_disable)
                mapper._fill_services_cache()

            self.assertTrue(mapper._403_disable)

            # No new requests to the apiserver
            request_mock.assert_called()
            request_mock.reset_mock()
            mapper._fill_services_cache()
            request_mock.assert_not_called()

Example #4

File: test_kube_pod_service_mapper.py Project: netsil/dd-agent

 def test_service_cache_invalidation_true(self):
     jsons = self._load_json_array(
         ['service_cache_events1.json', 'service_cache_services1.json', 'service_cache_events2.json'])
     with patch.object(self.kube, 'retrieve_json_auth', side_effect=jsons):
         mapper = PodServiceMapper(self.kube)
         mapper._fill_services_cache()
         mapper.check_services_cache_freshness()
         self.assertEqual(True, mapper._service_cache_invalidated)

Example #5

 def test_service_cache_invalidation_true(self):
     jsons = self._load_json_array([
         'service_cache_events1.json', 'service_cache_services1.json',
         'service_cache_events2.json'
     ])
     with patch.object(self.kube, 'retrieve_json_auth', side_effect=jsons):
         mapper = PodServiceMapper(self.kube)
         mapper._fill_services_cache()
         mapper.check_services_cache_freshness()
         self.assertEqual(True, mapper._service_cache_invalidated)

Example #6

File: test_kube_pod_service_mapper.py Project: dblackdblack/dd-agent

    def test_service_cache_fill(self):
        jsons = self._load_json_array(['service_cache_services2.json'])
        with patch.object(self.kube, 'retrieve_json_auth', side_effect=jsons):
            mapper = PodServiceMapper(self.kube)
            mapper._fill_services_cache()
        # Kubernetes service not imported because no selector
        self.assertEqual(3, len(mapper._service_cache_selectors))
        self.assertEqual(3, len(mapper._service_cache_names))

        self.assertEqual('redis-hello', mapper._service_cache_names['9474d98a-1aad-11e7-8b67-42010a840226'])
        redis = mapper._service_cache_selectors['9474d98a-1aad-11e7-8b67-42010a840226']
        self.assertEqual(2, len(redis))
        self.assertEqual('hello', redis['app'])
        self.assertEqual('db', redis['tier'])

Example #7

File: test_kube_pod_service_mapper.py Project: vitbyst/dd-agent

    def test_service_cache_fill(self):
        jsons = self._load_json_array(['service_cache_services2.json'])
        with patch.object(self.kube, 'retrieve_json_auth', side_effect=jsons):
            mapper = PodServiceMapper(self.kube)
            mapper._fill_services_cache()
        # Kubernetes service not imported because no selector
        self.assertEqual(3, len(mapper._service_cache_selectors))
        self.assertEqual(3, len(mapper._service_cache_names))

        self.assertEqual('redis-hello', mapper._service_cache_names['9474d98a-1aad-11e7-8b67-42010a840226'])
        redis = mapper._service_cache_selectors['9474d98a-1aad-11e7-8b67-42010a840226']
        self.assertEqual(2, len(redis))
        self.assertEqual('hello', redis['app'])
        self.assertEqual('db', redis['tier'])

Example #8

File: test_kube_pod_service_mapper.py Project: alq666/dd-agent

    def test_403_disable(self):
        exception403 = requests.exceptions.HTTPError()
        exception403.response = Mock()
        exception403.response.status_code = 403
        self.assertEquals(403, exception403.response.status_code)
        self.assertTrue(isinstance(exception403, requests.exceptions.HTTPError))

        with patch.object(self.kube, 'retrieve_json_auth', side_effect=exception403) as request_mock:
            # Fill pod label cache
            mapper = PodServiceMapper(self.kube)
            self.assertEqual(0, mapper._403_errors)

            for i in range(0, MAX_403_RETRIES):
                self.assertFalse(mapper._403_disable)
                mapper._fill_services_cache()

            self.assertTrue(mapper._403_disable)

            # No new requests to the apiserver
            request_mock.assert_called()
            request_mock.reset_mock()
            mapper._fill_services_cache()
            request_mock.assert_not_called()