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')
示例#2
0
 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')
示例#3
0
 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='*****@*****.**')