def test_get_conn_override_defaults(self, docker_client_mock): hook = DockerHook(docker_conn_id='docker_default', base_url='https://index.docker.io/v1/', version='1.23', tls='someconfig') hook.get_conn() docker_client_mock.assert_called_once_with( base_url='https://index.docker.io/v1/', version='1.23', tls='someconfig')
def test_get_conn_with_extra_config(self, _): try: hook = DockerHook(docker_conn_id='docker_with_extras', base_url='unix://var/run/docker.sock', version='auto') client = hook.get_conn() self.assertIsNotNone(client) except Exception: # pylint: disable=broad-except self.fail('Could not get connection from Airflow')
def test_get_conn_with_standard_config(self, _): try: hook = DockerHook(docker_conn_id='docker_default', base_url='unix://var/run/docker.sock', version='auto') client = hook.get_conn() assert client is not None except Exception: # pylint: disable=broad-except self.fail('Could not get connection from Airflow')
def test_conn_with_standard_config_passes_parameters(self, _): hook = DockerHook(docker_conn_id='docker_default', base_url='unix://var/run/docker.sock', version='auto') client = hook.get_conn() client.login.assert_called_once_with( # pylint: disable=no-member username='******', password='******', registry='some.docker.registry.com', reauth=True, email=None)
def test_conn_with_extra_config_passes_parameters(self, _): hook = DockerHook(docker_conn_id='docker_with_extras', base_url='unix://var/run/docker.sock', version='auto') client = hook.get_conn() client.login.assert_called_once_with( # pylint: disable=no-member username='******', password='******', registry='another.docker.registry.com:9876', reauth=False, email='*****@*****.**')