Skip to content

Commit f6e78cc

Browse files
xvellotruthbk
authored andcommitted
[docker] use DockerUtil.image_name_extractor for all code paths (#3334)
* use image_name_extractor for _get_checks_to_refresh too * use image_name_extractor for flares too * use image_name_extractor for unit test
1 parent 5959dc4 commit f6e78cc

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
lines changed

tests/core/test_service_discovery.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
_TemplateCache, CONFIG_FROM_KUBE, CONFIG_FROM_TEMPLATE, CONFIG_FROM_AUTOCONF
1818
from utils.service_discovery.sd_backend import get_sd_backend
1919
from utils.service_discovery.sd_docker_backend import SDDockerBackend, _SDDockerBackendConfigFetchState
20+
from utils.dockerutil import DockerUtil
2021

2122

2223
def clear_singletons(agentConfig):
@@ -587,7 +588,7 @@ def test_get_config_id(self, mock_get_auto_confd_path):
587588
for c_ins, _, _, _, expected_ident in self.container_inspects:
588589
sd_backend = get_sd_backend(agentConfig=self.auto_conf_agentConfig)
589590
self.assertEqual(
590-
sd_backend.get_config_id(c_ins.get('Image'), c_ins.get('Config', {}).get('Labels', {})),
591+
sd_backend.get_config_id(DockerUtil().image_name_extractor(c_ins), c_ins.get('Config', {}).get('Labels', {})),
591592
expected_ident)
592593
clear_singletons(self.auto_conf_agentConfig)
593594

utils/configcheck.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,13 +72,14 @@ def get_sd_configcheck(agentConfig, configs):
7272

7373

7474
def print_containers():
75-
containers = DockerUtil().client.containers()
75+
dockerutil = DockerUtil()
76+
containers = dockerutil.client.containers()
7677
print("\nContainers info:\n")
7778
print("Number of containers found: %s" % len(containers))
7879
for co in containers:
7980
c_id = 'ID: %s' % co.get('Id')[:12]
80-
c_image = 'image: %s' % co.get('Image')
81-
c_name = 'name: %s' % DockerUtil.container_name_extractor(co)[0]
81+
c_image = 'image: %s' % dockerutil.image_name_extractor(co)
82+
c_name = 'name: %s' % dockerutil.container_name_extractor(co)[0]
8283
print("\t- %s %s %s" % (c_id, c_image, c_name))
8384
print('\n')
8485

utils/service_discovery/sd_docker_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def _get_checks_to_refresh(self, state, c_id):
144144
return
145145

146146
identifier = inspect.get('Config', {}).get('Labels', {}).get(DATADOG_ID) or \
147-
inspect.get('Config', {}).get('Image')
147+
self.dockerutil.image_name_extractor(inspect)
148148

149149
platform_kwargs = {}
150150
if Platform.is_k8s():

0 commit comments

Comments
 (0)