Ejemplos de DockerOperator.container en Python

Ejemplo n.º 1

Archivo: test_docker_operator.py Proyecto: zuiwanting/flink-ai-extended

    def test_on_kill():
        client_mock = mock.Mock(spec=APIClient)

        operator = DockerOperator(image='ubuntu', owner='unittest', task_id='unittest')
        operator.cli = client_mock
        operator.container = {'Id': 'some_id'}

        operator.on_kill()

        client_mock.stop.assert_called_with('some_id')

Ejemplo n.º 2

Archivo: test_docker_operator.py Proyecto: MiguelPeralvo/incubator-airflow

    def test_on_kill():
        client_mock = mock.Mock(spec=APIClient)

        operator = DockerOperator(image='ubuntu', owner='unittest', task_id='unittest')
        operator.cli = client_mock
        operator.container = {'Id': 'some_id'}

        operator.on_kill()

        client_mock.stop.assert_called_with('some_id')