1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
# ================================================================================
# 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.
# 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 requests.exceptions import HTTPError, ConnectionError
class FakeResponse:
def __init__(self, status_code, thejson):
self.status_code = status_code
self.thejson = thejson
self.text = ""
def raise_for_status(self):
if self.status_code > 299:
raise HTTPError
def json(self):
return self.thejson
good_resp_all = FakeResponse(
status_code=200,
thejson={
"config": {"key_to_your_heart": 666},
"dti": {"some amazing": "dti stuff"},
"policies": {"event": {"foo": "bar"}, "items": [{"foo2": "bar2"}]},
"otherkey": {"foo3": "bar3"},
},
)
good_resp_conf = FakeResponse(status_code=200, thejson={"key_to_your_heart": 666})
@pytest.fixture
def monkeyed_requests_get():
"""
mock for the CBS get
"""
def _monkeyed_requests_get(url):
if url == "http://config-binding-service:10000/service_component_all/testhostname":
return good_resp_all
elif url == "http://config-binding-service:10000/service_component/testhostname":
return good_resp_conf
else:
raise Exception("Unexpected URL {0}!".format(url))
return _monkeyed_requests_get
@pytest.fixture
def monkeyed_requests_get_https():
"""
mock for the CBS get
"""
def _monkeyed_requests_get_https(url, verify=""):
if url == "https://config-binding-service:10443/service_component_all/testhostname":
return good_resp_all
elif url == "https://config-binding-service:10443/service_component/testhostname":
return good_resp_conf
else:
raise Exception("Unexpected URL {0}!".format(url))
return _monkeyed_requests_get_https
@pytest.fixture
def monkeyed_requests_get_404():
def _monkeyed_requests_get_404(url):
"""
get that pretends that key doesnt exist
"""
if url in [
"http://config-binding-service:10000/service_component_all/testhostname",
"http://config-binding-service:10000/service_component/testhostname",
]:
return FakeResponse(status_code=404, thejson={})
raise Exception("Unexpected URL {0}!".format(url))
return _monkeyed_requests_get_404
@pytest.fixture
def monkeyed_requests_get_unreachable():
def _monkeyed_requests_get_unreachable(_url):
raise ConnectionError()
return _monkeyed_requests_get_unreachable
|
