diff options
author | Michael Hwang <mhwang@research.att.com> | 2017-08-23 14:26:36 -0400 |
---|---|---|
committer | Michael Hwang <mhwang@research.att.com> | 2017-08-23 14:31:51 -0400 |
commit | 21af561cafe31681f94479e8c70f157f6e6ecc53 (patch) | |
tree | f41f69e1419867fd70af1f7697f78e3490d41b87 /docker/tests/test_discovery.py | |
parent | 94cbaca0f5d9447afe9b0f392f248470420422e5 (diff) |
Add docker and relationships plugin
Change-Id: I323599ae2965f081f2061b6791635bbeddb09811
Issue-Id: DCAEGEN2-79
Signed-off-by: Michael Hwang <mhwang@research.att.com>
Diffstat (limited to 'docker/tests/test_discovery.py')
-rw-r--r-- | docker/tests/test_discovery.py | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/docker/tests/test_discovery.py b/docker/tests/test_discovery.py new file mode 100644 index 0000000..9a18519 --- /dev/null +++ b/docker/tests/test_discovery.py @@ -0,0 +1,40 @@ +# ============LICENSE_START======================================================= +# org.onap.dcae +# ================================================================================ +# Copyright (c) 2017 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. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============LICENSE_END========================================================= +# +# ECOMP is a trademark and service mark of AT&T Intellectual Property. + +import pytest +from functools import partial +import requests +from dockerplugin import discovery as dis + + +def test_wrap_consul_call(): + def foo(a, b, c="default"): + return " ".join([a, b, c]) + + wrapped_foo = partial(dis._wrap_consul_call, foo) + assert wrapped_foo("hello", "world") == "hello world default" + assert wrapped_foo("hello", "world", c="new masters") == "hello world new masters" + + def foo_connection_error(a, b, c): + raise requests.exceptions.ConnectionError("simulate failed connection") + + wrapped_foo = partial(dis._wrap_consul_call, foo_connection_error) + with pytest.raises(dis.DiscoveryConnectionError): + wrapped_foo("a", "b", "c") |