diff options
Diffstat (limited to 'dcae-policy/tests')
-rw-r--r-- | dcae-policy/tests/test_discovery.py | 8 | ||||
-rw-r--r-- | dcae-policy/tests/test_tasks.py | 6 |
2 files changed, 7 insertions, 7 deletions
diff --git a/dcae-policy/tests/test_discovery.py b/dcae-policy/tests/test_discovery.py index 129ad99..0c41fe3 100644 --- a/dcae-policy/tests/test_discovery.py +++ b/dcae-policy/tests/test_discovery.py @@ -1,5 +1,5 @@ # ================================================================================ -# Copyright (c) 2018 AT&T Intellectual Property. All rights reserved. +# Copyright (c) 2019 AT&T Intellectual Property. All rights reserved. # ================================================================================ # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -35,7 +35,7 @@ from tests.mock_setup import (MONKEYED_POLICY_ID, POLICY_ID, MonkeyedNode, POLICY_HANDLER_FROM_KV = "http://policy_handler_from_kv:25577" -def monkeyed_discovery_get_failure(full_path): +def monkeyed_discovery_get_failure(full_path, **kwargs): """monkeypatch for the GET to consul""" raise requests.ConnectionError("monkey-boom") @@ -65,7 +65,7 @@ def test_discovery_failure(monkeypatch): current_ctx.clear() -def monkeyed_discovery_get_kv(full_path): +def monkeyed_discovery_get_kv(full_path, **kwargs): """monkeypatch for the GET to consul""" if full_path.startswith(discovery.CONSUL_SERVICE_URL.format("")): return MonkeyedResponse(full_path) @@ -102,7 +102,7 @@ def test_discovery_kv(monkeypatch): current_ctx.clear() -def monkeyed_discovery_get(full_path): +def monkeyed_discovery_get(full_path, **kwargs): """monkeypatch for the GET to consul""" return MonkeyedResponse(full_path, {}, [{"ServiceAddress": "monkey-policy-handler-address", "ServicePort": "9999"}]) diff --git a/dcae-policy/tests/test_tasks.py b/dcae-policy/tests/test_tasks.py index f2ca205..dc89d65 100644 --- a/dcae-policy/tests/test_tasks.py +++ b/dcae-policy/tests/test_tasks.py @@ -1,5 +1,5 @@ # ================================================================================ -# Copyright (c) 2017-2018 AT&T Intellectual Property. All rights reserved. +# Copyright (c) 2017-2019 AT&T Intellectual Property. All rights reserved. # ================================================================================ # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -36,7 +36,7 @@ from tests.mock_setup import (CONFIG_NAME, MONKEYED_POLICY_ID, POLICY_BODY, LATEST_POLICIES = "latest_policies" -def monkeyed_policy_handler_get(full_path, headers=None): +def monkeyed_policy_handler_get(full_path, headers=None, **kwargs): """monkeypatch for the GET to policy-engine""" return MonkeyedResponse(full_path, headers, MonkeyedPolicyBody.create_policy(MONKEYED_POLICY_ID)) @@ -105,7 +105,7 @@ def test_policy_get_fail(monkeypatch): current_ctx.clear() -def monkeyed_policy_handler_find(full_path, json, headers): +def monkeyed_policy_handler_find(full_path, json, headers, **kwargs): """monkeypatch for the GET to policy-engine""" return MonkeyedResponse( full_path, headers, |