test_docker.py 文件源码

def test_pull_image(self):
        cases = dict(
            no_errors=dict(
                side_effect=(SucceededResult(), ),
                expected_pull_error=None,
            ),
            error=dict(
                side_effect=(RuntimeError(), ),
                expected_pull_error=True,
            ),
        )
        for case, test_data in cases.items():
            with self.subTest(case=case):
                service = docker.Service(name='service', image='image')
                with mock.patch.object(
                    fabricio,
                    'run',
                    side_effect=test_data['side_effect']
                ):
                    service.pull_image()
                    self.assertEqual(
                        test_data['expected_pull_error'],
                        service.pull_errors.get(fab.env.host),
                    )