diff options
author | Alexander Gerasiov <gerasiov@yandex-team.ru> | 2018-07-10 14:38:19 +0300 |
---|---|---|
committer | Alexander Gerasiov <gerasiov@yandex-team.ru> | 2018-07-10 14:38:19 +0300 |
commit | 5e06faab1ecaa2d30100f24b39f81e307329a19e (patch) | |
tree | 8845322ece509cd0bb82c81135923d48ca5404bf /tests/unit/fake_api_client.py | |
parent | 6951127c68046ae46df6972a37c6cd80cae1d622 (diff) | |
parent | 598a95848dfa5d5ba13c99741bb54310697a2e78 (diff) |
Merge tag 'upstream/3.4.1'
Upstream version 3.4.1
Diffstat (limited to 'tests/unit/fake_api_client.py')
-rw-r--r-- | tests/unit/fake_api_client.py | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/tests/unit/fake_api_client.py b/tests/unit/fake_api_client.py index 15b60ea..2147bfd 100644 --- a/tests/unit/fake_api_client.py +++ b/tests/unit/fake_api_client.py @@ -20,15 +20,18 @@ class CopyReturnMagicMock(mock.MagicMock): return ret -def make_fake_api_client(): +def make_fake_api_client(overrides=None): """ Returns non-complete fake APIClient. This returns most of the default cases correctly, but most arguments that change behaviour will not work. """ + + if overrides is None: + overrides = {} api_client = docker.APIClient() - mock_client = CopyReturnMagicMock(**{ + mock_attrs = { 'build.return_value': fake_api.FAKE_IMAGE_ID, 'commit.return_value': fake_api.post_fake_commit()[1], 'containers.return_value': fake_api.get_fake_containers()[1], @@ -47,15 +50,18 @@ def make_fake_api_client(): 'networks.return_value': fake_api.get_fake_network_list()[1], 'start.return_value': None, 'wait.return_value': {'StatusCode': 0}, - }) + } + mock_attrs.update(overrides) + mock_client = CopyReturnMagicMock(**mock_attrs) + mock_client._version = docker.constants.DEFAULT_DOCKER_API_VERSION return mock_client -def make_fake_client(): +def make_fake_client(overrides=None): """ Returns a Client with a fake APIClient. """ client = docker.DockerClient() - client.api = make_fake_api_client() + client.api = make_fake_api_client(overrides) return client |