def populateExpectations(tree, file_name, port_number, expectations):
if not tree:
tree = createTree()
tree = addNodeToTree(tree, port_number, port_number, ROOT, PORT)
for expectation in expectations:
request_parameters = expectations.get(expectation).get(EXP_REQUEST)
if request_parameters is None:
request_parameters = {}
parent = port_number
method = request_parameters.get(EXP_METHOD)
if method is None:
method = "GET"
node_id = parent + ":" + method
tree = addNodeToTree(tree, method, node_id, parent, EXP_METHOD)
parent = node_id
path = request_parameters.get(EXP_PATH)
if path is None:
path = "/"
node_id = parent + ">" + path
tree = addNodeToTree(tree, path, node_id, parent, EXP_PATH)
parent = node_id
queryList = request_parameters.get(EXP_QUERY)
if queryList is not None:
queryList.sort()
query = query_delimiter.join(queryList)
node_id = parent + "?" + query
tree = addNodeToTree(tree, query, node_id, parent, EXP_QUERY)
parent = node_id
headersList = request_parameters.get(EXP_HEADERS)
if headersList is not None:
headersList.sort()
headers = header_delimiter.join(headersList)
node_id = parent + "|" + headers
tree = addNodeToTree(tree, headers, node_id, parent, EXP_HEADERS)
parent = node_id
requestBody = request_parameters.get(EXP_BODY)
if requestBody is not None:
node_id = parent + "|" + requestBody
tree = addNodeToTree(tree, requestBody, node_id, parent, EXP_BODY)
parent = node_id
response = expectations.get(expectation).get(EXP_RESPONSE)
if response is None:
response = {EXP_STATUS_CODE: 200}
tree = addLeafNodeToTree(
tree, response, file_name + ":" + port_number + "/" + expectation, parent, EXP_RESPONSE
)
return tree
def getResponseFromExpectations(tree, request):
print "Finding response for request {" + request.toString() + "}"
port, method, path, query, headers, body = (
request.port,
request.method,
request.path,
request.query,
request.headers,
request.body,
)
query = query_delimiter.join(sorted(query.split(query_delimiter)))
nodes = []
if tree.contains(port):
nodes = [getNodeFromTreeById(port, tree)]
print "Port: [" + port + "] <> Nodes under observation: " + str([getIdForNode(node) for node in nodes])
nodes = getNodesThatMatchRequestComponent(nodes, EXP_METHOD, method, tree)
print "Method: [" + method + "] <> Nodes under observation: " + str([getIdForNode(node) for node in nodes])
nodes = getNodesThatMatchRequestComponent(nodes, EXP_PATH, path, tree)
print "Path: [" + path + "] <> Nodes under observation: " + str([getIdForNode(node) for node in nodes])
nodes = getNodesThatMatchRequestComponent(nodes, EXP_QUERY, query, tree)
print "Query: [" + query + "] <> Nodes under observation: " + str([getIdForNode(node) for node in nodes])
nodes = getNodesThatMatchRequestComponent(nodes, EXP_HEADERS, headers, tree, True)
print "Headers: [" + headers + "] <> Nodes under observation: " + str([getIdForNode(node) for node in nodes])
nodes = getNodesThatMatchRequestComponent(nodes, EXP_BODY, body, tree)
print "Body: [" + body + "] <> Nodes under observation: " + str([getIdForNode(node) for node in nodes])
leaf_nodes = getDirectChildrenForNodeList(nodes, tree, EXP_RESPONSE)
print "Response candidates: " + str([getIdForNode(node) for node in leaf_nodes])
if len(leaf_nodes) == 1:
print "Returning response: " + str(getIdForNode(leaf_nodes[0])) + " for request - {" + request.toString() + "}"
response = getTagForNode(leaf_nodes[0])
responseBody = response[EXP_BODY]
if responseBody.endswith(".json"):
basePath = os.path.dirname(getIdForNode(leaf_nodes[0]).split(":")[0])
responseBody = Arguments.getPathForFile(responseBody, basePath)
with open(responseBody, "r") as response_file:
responseBody = response_file.read()
response[EXP_BODY] = responseBody
return response
elif len(leaf_nodes) < 1:
raise ValueError("No match found for request - {" + request.toString() + "}")
else:
node_names = [getIdForNode(leaf_node) for leaf_node in leaf_nodes]
raise ValueError(
"Multiple matches found for request - {" + request.toString() + "} >> Matching nodes: " + str(node_names)
)
