Ejemplo n.º 1
0
def run_injector(docker_socket, docker_info_path):
    injector = DockerInjector(
        get_production_docker_wrapper(base_url=docker_socket),
        docker_info_path=docker_info_path)

    while True:
        injector.start()
        time.sleep(30)
def run_injector(docker_socket, docker_info_path):
    injector = DockerInjector(
        get_production_docker_wrapper(base_url=docker_socket),
        docker_info_path=docker_info_path)

    while True:
        injector.start()
        time.sleep(30)
 def test_inject(self):
     expected = 'file already exists'
     container = {'Id': 'c1'}
     wrapper_mock = Mock()
     wrapper_mock.get_containers.return_value = [container]
     wrapper_mock.run_command.return_value = 'file already exists'
     injector = DockerInjector(docker_wrapper=wrapper_mock, docker_info_path="/path/docker.info")
     results = injector.inject_context()
     for c, result in results:
         self.assertEqual(container['Id'], c)
         self.assertEqual(expected, result)
Ejemplo n.º 4
0
 def test_inject(self):
     expected = 'file already exists'
     container = {'Id': 'c1'}
     wrapper_mock = Mock()
     wrapper_mock.get_containers.return_value = [container]
     wrapper_mock.run_command.return_value = 'file already exists'
     injector = DockerInjector(docker_wrapper=wrapper_mock,
                               docker_info_path="/path/docker.info")
     results = injector.inject_context()
     for c, result in results:
         self.assertEqual(container['Id'], c)
         self.assertEqual(expected, result)
 def test_get_my_container_id_when_file_not_exists(self):
     template = 'Docker container name=/boring_brattain,Docker image=ttt,Docker container id={0},Docker host=galha-ubuntu'
     expected_id = 'cd9d134b64807148faa24a17519c8e1a2650b825d4d38944ac54281b2dd1d94e'
     data = template.format(expected_id)
     with patch('os.path.exists') as exists:
         exists.side_effect = [False, True]
         with patch.object(builtins, 'open', mock_open(read_data=data)):
             wrapper_mock = Mock()
             wrapper_mock.get_containers.return_value = [{'Id': 'c1'}]
             wrapper_mock.run_command.return_value = 'file already exists'
             injector = DockerInjector(docker_wrapper=wrapper_mock, docker_info_path="/path/docker.info")
             id = injector.get_my_container_id()
             self.assertEqual(2, wrapper_mock.run_command.call_count)
             self.assertEqual(expected_id, id)
Ejemplo n.º 6
0
 def test_get_my_container_id_when_file_not_exists(self):
     template = 'Docker container name=/boring_brattain,Docker image=ttt,Docker container id={0},Docker host=galha-ubuntu'
     expected_id = 'cd9d134b64807148faa24a17519c8e1a2650b825d4d38944ac54281b2dd1d94e'
     data = template.format(expected_id)
     with patch('os.path.exists') as exists:
         exists.side_effect = [False, True]
         with patch.object(builtins, 'open', mock_open(read_data=data)):
             wrapper_mock = Mock()
             wrapper_mock.get_containers.return_value = [{'Id': 'c1'}]
             wrapper_mock.run_command.return_value = 'file already exists'
             injector = DockerInjector(docker_wrapper=wrapper_mock,
                                       docker_info_path="/path/docker.info")
             id = injector.get_my_container_id()
             self.assertEqual(2, wrapper_mock.run_command.call_count)
             self.assertEqual(expected_id, id)
    def test_start(self):
        expected_c1 = {'Docker container id': 'c1', 'Docker host': 'host', 'Docker container name': 'name1', 'Docker image': 'image1'}
        expected_c2 = {'Docker container id': 'c2', 'Docker host': 'host', 'Docker container name': 'name2', 'Docker image': 'image2'}
        container = {'Id': 'c1', 'Image':'image1', 'Names':['name1']}

        wrapper_mock = Mock()
        wrapper_mock.get_containers.return_value = [container]
        wrapper_mock.get_host_name.return_value='host'
        wrapper_mock.run_command.return_value = 'file already exists'
        wrapper_mock.get_events.return_value = [{'time': 1439388853, 'Id': 'c2', 'id': 'c2', 'from': 'image2', 'status': 'start'}]
        wrapper_mock.get_inspection.return_value = {'Id': 'c2', 'status': 'start', 'Config':{'Image':'image2'}, 'Name':'name2'}

        def assert_func():
            res = wrapper_mock.run_command.mock_calls
            start = time.time()
            while len(res) < 4 and time.time() - start < 100:
                time.sleep(1)
            self.assertEqual(4, len(res))

        injector = DockerInjector(docker_wrapper=wrapper_mock, docker_info_path="/path/docker.info")
        with ThreadPoolExecutor(max_workers=3) as ex:
            ex.submit(lambda: injector.start())
            result = ex.submit(lambda: assert_func())
            result.result()
            calls_argument_keys = [keys for name, args, keys in wrapper_mock.run_command.mock_calls]
            c1_calls = [d for d in calls_argument_keys if 'container' in d and 'Id' in d['container'] and d['container']['Id'] == 'c1']
            c2_calls = [d for d in calls_argument_keys if 'container' in d and 'Id' in d['container'] and d['container']['Id'] == 'c2']
            self.assertEqual(2, len(c1_calls))
            self.assertEqual(2, len(c2_calls))
            c1_data = c1_calls[1]['cmd']
            m1 = re.search('printf \'(.+)\'', c1_data)
            self.assertTrue(m1)
            actual_c1= {key:val for key,val in [token.split('=') for token in m1.group(1).strip(' ').split(',')]}
            self.assertDictEqual(expected_c1, actual_c1)
            c2_data = c2_calls[1]['cmd']
            m2 = re.search('printf \'(.+)\'', c2_data)
            self.assertTrue(m2)
            actual_c2= {key:val for key,val in [token.split('=') for token in m2.group(1).strip(' ').split(',')]}
            self.assertDictEqual(expected_c2, actual_c2)
Ejemplo n.º 8
0
def run_collect_performance_counters(docker_socket, sdk_file, docker_info_file,
                                     collect_interval):
    docker_wrapper = get_production_docker_wrapper(base_url=docker_socket)
    docker_injector = DockerInjector(docker_wrapper=docker_wrapper,
                                     docker_info_path=docker_info_file)
    collector = DockerCollector(docker_wrapper=docker_wrapper,
                                docker_injector=docker_injector,
                                samples_in_each_metric=5,
                                sdk_file=sdk_file)

    while True:
        collector.collect_stats_and_send()
        time.sleep(float(collect_interval))
Ejemplo n.º 9
0
def run_collect_containers_events(docker_socket, docker_info_file, sdk_file):
    docker_wrapper = get_production_docker_wrapper(base_url=docker_socket)
    docker_injector = DockerInjector(docker_wrapper=docker_wrapper,
                                     docker_info_path=docker_info_file)
    collector = DockerCollector(docker_wrapper=docker_wrapper,
                                docker_injector=docker_injector,
                                samples_in_each_metric=5,
                                sdk_file=sdk_file)
    while True:
        try:
            collector.collect_container_events()
        except Exception as e:
            print(e, file=sys.stderr)
            time.sleep(10)
Ejemplo n.º 10
0
    def test_start(self):
        expected_c1 = {
            'Docker container id': 'c1',
            'Docker host': 'host',
            'Docker container name': 'name1',
            'Docker image': 'image1'
        }
        expected_c2 = {
            'Docker container id': 'c2',
            'Docker host': 'host',
            'Docker container name': 'name2',
            'Docker image': 'image2'
        }
        container = {'Id': 'c1', 'Image': 'image1', 'Names': ['name1']}

        wrapper_mock = Mock()
        wrapper_mock.get_containers.return_value = [container]
        wrapper_mock.get_host_name.return_value = 'host'
        wrapper_mock.run_command.return_value = 'file already exists'
        wrapper_mock.get_events.return_value = [{
            'time': 1439388853,
            'Id': 'c2',
            'id': 'c2',
            'from': 'image2',
            'status': 'start'
        }]
        wrapper_mock.get_inspection.return_value = {
            'Id': 'c2',
            'status': 'start',
            'Config': {
                'Image': 'image2'
            },
            'Name': 'name2'
        }

        def assert_func():
            res = wrapper_mock.run_command.mock_calls
            start = time.time()
            while len(res) < 4 and time.time() - start < 100:
                time.sleep(1)
            self.assertEqual(4, len(res))

        injector = DockerInjector(docker_wrapper=wrapper_mock,
                                  docker_info_path="/path/docker.info")
        with ThreadPoolExecutor(max_workers=3) as ex:
            ex.submit(lambda: injector.start())
            result = ex.submit(lambda: assert_func())
            result.result()
            calls_argument_keys = [
                keys
                for name, args, keys in wrapper_mock.run_command.mock_calls
            ]
            c1_calls = [
                d for d in calls_argument_keys if 'container' in d
                and 'Id' in d['container'] and d['container']['Id'] == 'c1'
            ]
            c2_calls = [
                d for d in calls_argument_keys if 'container' in d
                and 'Id' in d['container'] and d['container']['Id'] == 'c2'
            ]
            self.assertEqual(2, len(c1_calls))
            self.assertEqual(2, len(c2_calls))
            c1_data = c1_calls[1]['cmd']
            m1 = re.search('printf \'(.+)\'', c1_data)
            self.assertTrue(m1)
            actual_c1 = {
                key: val
                for key, val in [
                    token.split('=')
                    for token in m1.group(1).strip(' ').split(',')
                ]
            }
            self.assertDictEqual(expected_c1, actual_c1)
            c2_data = c2_calls[1]['cmd']
            m2 = re.search('printf \'(.+)\'', c2_data)
            self.assertTrue(m2)
            actual_c2 = {
                key: val
                for key, val in [
                    token.split('=')
                    for token in m2.group(1).strip(' ').split(',')
                ]
            }
            self.assertDictEqual(expected_c2, actual_c2)