Refactor test_shell
This file has become difficult to maintain with conditionals everywhere to handle the different http clients. This change replaces the http mocking code with calls to a handful of mock methods, and deletes what is left unused. Change-Id: Id2339b63a90f4805afe94f89f4c2715834588c61
This commit is contained in:
parent
c212875d7e
commit
a654226dd1
|
@ -13,215 +13,6 @@
|
||||||
|
|
||||||
from oslo_serialization import jsonutils
|
from oslo_serialization import jsonutils
|
||||||
|
|
||||||
from heatclient.common import http
|
|
||||||
from heatclient import exc
|
|
||||||
|
|
||||||
|
|
||||||
def script_heat_list(url=None, show_nested=False, client=http.HTTPClient):
|
|
||||||
if url is None:
|
|
||||||
url = '/stacks?'
|
|
||||||
|
|
||||||
resp_dict = {"stacks": [
|
|
||||||
{
|
|
||||||
"id": "1",
|
|
||||||
"stack_name": "teststack",
|
|
||||||
"stack_owner": "testowner",
|
|
||||||
"project": "testproject",
|
|
||||||
"stack_status": 'CREATE_COMPLETE',
|
|
||||||
"creation_time": "2012-10-25T01:58:47Z"
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "2",
|
|
||||||
"stack_name": "teststack2",
|
|
||||||
"stack_owner": "testowner",
|
|
||||||
"project": "testproject",
|
|
||||||
"stack_status": 'IN_PROGRESS',
|
|
||||||
"creation_time": "2012-10-25T01:58:47Z"
|
|
||||||
}]
|
|
||||||
}
|
|
||||||
if show_nested:
|
|
||||||
nested = {
|
|
||||||
"id": "3",
|
|
||||||
"stack_name": "teststack_nested",
|
|
||||||
"stack_status": 'IN_PROGRESS',
|
|
||||||
"creation_time": "2012-10-25T01:58:47Z",
|
|
||||||
"parent": "theparentof3"
|
|
||||||
}
|
|
||||||
resp_dict["stacks"].append(nested)
|
|
||||||
resp = FakeHTTPResponse(200,
|
|
||||||
'success, you',
|
|
||||||
{'content-type': 'application/json'},
|
|
||||||
jsonutils.dumps(resp_dict))
|
|
||||||
if client == http.SessionClient:
|
|
||||||
client.request(url, 'GET').AndReturn(resp)
|
|
||||||
else:
|
|
||||||
client.json_request('GET', url).AndReturn((resp, resp_dict))
|
|
||||||
|
|
||||||
|
|
||||||
def mock_script_heat_list(show_nested=False):
|
|
||||||
resp_dict = {"stacks": [
|
|
||||||
{
|
|
||||||
"id": "1",
|
|
||||||
"stack_name": "teststack",
|
|
||||||
"stack_owner": "testowner",
|
|
||||||
"project": "testproject",
|
|
||||||
"stack_status": 'CREATE_COMPLETE',
|
|
||||||
"creation_time": "2012-10-25T01:58:47Z"
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "2",
|
|
||||||
"stack_name": "teststack2",
|
|
||||||
"stack_owner": "testowner",
|
|
||||||
"project": "testproject",
|
|
||||||
"stack_status": 'IN_PROGRESS',
|
|
||||||
"creation_time": "2012-10-25T01:58:47Z"
|
|
||||||
}]
|
|
||||||
}
|
|
||||||
if show_nested:
|
|
||||||
nested = {
|
|
||||||
"id": "3",
|
|
||||||
"stack_name": "teststack_nested",
|
|
||||||
"stack_status": 'IN_PROGRESS',
|
|
||||||
"creation_time": "2012-10-25T01:58:47Z",
|
|
||||||
"parent": "theparentof3"
|
|
||||||
}
|
|
||||||
resp_dict["stacks"].append(nested)
|
|
||||||
resp = FakeHTTPResponse(200,
|
|
||||||
'success, you',
|
|
||||||
{'content-type': 'application/json'},
|
|
||||||
jsonutils.dumps(resp_dict))
|
|
||||||
return resp, resp_dict
|
|
||||||
|
|
||||||
|
|
||||||
def mock_script_event_list(
|
|
||||||
stack_name="teststack", resource_name=None,
|
|
||||||
rsrc_eventid1="7fecaeed-d237-4559-93a5-92d5d9111205",
|
|
||||||
rsrc_eventid2="e953547a-18f8-40a7-8e63-4ec4f509648b",
|
|
||||||
action="CREATE", final_state="COMPLETE", fakehttp=True):
|
|
||||||
|
|
||||||
resp_dict = {"events": [
|
|
||||||
{"event_time": "2013-12-05T14:14:31",
|
|
||||||
"id": rsrc_eventid1,
|
|
||||||
"links": [{"href": "http://heat.example.com:8004/foo",
|
|
||||||
"rel": "self"},
|
|
||||||
{"href": "http://heat.example.com:8004/foo2",
|
|
||||||
"rel": "resource"},
|
|
||||||
{"href": "http://heat.example.com:8004/foo3",
|
|
||||||
"rel": "stack"}],
|
|
||||||
"logical_resource_id": "myDeployment",
|
|
||||||
"physical_resource_id": None,
|
|
||||||
"resource_name": resource_name if resource_name else "testresource",
|
|
||||||
"resource_status": "%s_IN_PROGRESS" % action,
|
|
||||||
"resource_status_reason": "state changed"},
|
|
||||||
{"event_time": "2013-12-05T14:14:32",
|
|
||||||
"id": rsrc_eventid2,
|
|
||||||
"links": [{"href": "http://heat.example.com:8004/foo",
|
|
||||||
"rel": "self"},
|
|
||||||
{"href": "http://heat.example.com:8004/foo2",
|
|
||||||
"rel": "resource"},
|
|
||||||
{"href": "http://heat.example.com:8004/foo3",
|
|
||||||
"rel": "stack"}],
|
|
||||||
"logical_resource_id": "myDeployment",
|
|
||||||
"physical_resource_id": "bce15ec4-8919-4a02-8a90-680960fb3731",
|
|
||||||
"resource_name": resource_name if resource_name else "testresource",
|
|
||||||
"resource_status": "%s_%s" % (action, final_state),
|
|
||||||
"resource_status_reason": "state changed"}]}
|
|
||||||
|
|
||||||
if resource_name is None:
|
|
||||||
# if resource_name is not specified,
|
|
||||||
# then request is made for stack events. Hence include the stack event
|
|
||||||
stack_event1 = "0159dccd-65e1-46e8-a094-697d20b009e5"
|
|
||||||
stack_event2 = "8f591a36-7190-4adb-80da-00191fe22388"
|
|
||||||
resp_dict["events"].insert(
|
|
||||||
0, {"event_time": "2013-12-05T14:14:30",
|
|
||||||
"id": stack_event1,
|
|
||||||
"links": [{"href": "http://heat.example.com:8004/foo",
|
|
||||||
"rel": "self"},
|
|
||||||
{"href": "http://heat.example.com:8004/foo2",
|
|
||||||
"rel": "resource"},
|
|
||||||
{"href": "http://heat.example.com:8004/foo3",
|
|
||||||
"rel": "stack"}],
|
|
||||||
"logical_resource_id": "aResource",
|
|
||||||
"physical_resource_id": None,
|
|
||||||
"resource_name": stack_name,
|
|
||||||
"resource_status": "%s_IN_PROGRESS" % action,
|
|
||||||
"resource_status_reason": "state changed"})
|
|
||||||
resp_dict["events"].append(
|
|
||||||
{"event_time": "2013-12-05T14:14:33",
|
|
||||||
"id": stack_event2,
|
|
||||||
"links": [{"href": "http://heat.example.com:8004/foo",
|
|
||||||
"rel": "self"},
|
|
||||||
{"href": "http://heat.example.com:8004/foo2",
|
|
||||||
"rel": "resource"},
|
|
||||||
{"href": "http://heat.example.com:8004/foo3",
|
|
||||||
"rel": "stack"}],
|
|
||||||
"logical_resource_id": "aResource",
|
|
||||||
"physical_resource_id": None,
|
|
||||||
"resource_name": stack_name,
|
|
||||||
"resource_status": "%s_%s" % (action, final_state),
|
|
||||||
"resource_status_reason": "state changed"})
|
|
||||||
|
|
||||||
resp = FakeHTTPResponse(
|
|
||||||
200,
|
|
||||||
'OK',
|
|
||||||
{'content-type': 'application/json'},
|
|
||||||
jsonutils.dumps(resp_dict)) if fakehttp else None
|
|
||||||
|
|
||||||
return resp, resp_dict
|
|
||||||
|
|
||||||
|
|
||||||
def script_heat_normal_error(client=http.HTTPClient):
|
|
||||||
resp_dict = {
|
|
||||||
"explanation": "The resource could not be found.",
|
|
||||||
"code": 404,
|
|
||||||
"error": {
|
|
||||||
"message": "The Stack (bad) could not be found.",
|
|
||||||
"type": "StackNotFound",
|
|
||||||
"traceback": "",
|
|
||||||
},
|
|
||||||
"title": "Not Found"
|
|
||||||
}
|
|
||||||
resp = FakeHTTPResponse(400,
|
|
||||||
'The resource could not be found',
|
|
||||||
{'content-type': 'application/json'},
|
|
||||||
jsonutils.dumps(resp_dict))
|
|
||||||
if client == http.SessionClient:
|
|
||||||
client.request('/stacks/bad', 'GET').AndRaise(exc.from_response(resp))
|
|
||||||
else:
|
|
||||||
client.json_request('GET',
|
|
||||||
'/stacks/bad').AndRaise(exc.from_response(resp))
|
|
||||||
|
|
||||||
|
|
||||||
def script_heat_error(resp_string, client=http.HTTPClient):
|
|
||||||
resp = FakeHTTPResponse(400,
|
|
||||||
'The resource could not be found',
|
|
||||||
{'content-type': 'application/json'},
|
|
||||||
resp_string)
|
|
||||||
if client == http.SessionClient:
|
|
||||||
client.request('/stacks/bad', 'GET').AndRaise(exc.from_response(resp))
|
|
||||||
else:
|
|
||||||
client.json_request('GET',
|
|
||||||
'/stacks/bad').AndRaise(exc.from_response(resp))
|
|
||||||
|
|
||||||
|
|
||||||
def fake_headers():
|
|
||||||
return {'X-Auth-Token': 'abcd1234',
|
|
||||||
'Content-Type': 'application/json',
|
|
||||||
'Accept': 'application/json',
|
|
||||||
'User-Agent': 'python-heatclient'}
|
|
||||||
|
|
||||||
|
|
||||||
class FakeServiceCatalog(object):
|
|
||||||
def url_for(self, endpoint_type, service_type):
|
|
||||||
return 'http://192.168.1.5:8004/v1/f14b41234'
|
|
||||||
|
|
||||||
|
|
||||||
class FakeKeystone(object):
|
|
||||||
service_catalog = FakeServiceCatalog()
|
|
||||||
|
|
||||||
def __init__(self, auth_token):
|
|
||||||
self.auth_token = auth_token
|
|
||||||
|
|
||||||
|
|
||||||
class FakeRaw(object):
|
class FakeRaw(object):
|
||||||
version = 110
|
version = 110
|
||||||
|
|
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue