def handle_node_request(args, node_type):
node = graph.find_node(node_type, args["id"])
if node:
return jsonify(
name=node.properties["name"],
id=node.properties["node_id"]
)
return jsonify(error="No matching node: {0}".format(args["id"]))
def get_user(args):
node = graph.find_node("User", args["id"])
if node:
return jsonify(
id=node.properties["node_id"],
name=node.properties["name"],
city=node.properties["city"]
)
return jsonify(error="No matching user: {0}".format(args["id"]))
def test_create_new(self):
# make sure user does not exist
graph.delete_user(self.user)
# submit POST request to create new user
rv = self.app.post(self.endpoint, data=self.data)
# confirm JSON response matches what we expect
response = json.loads(rv.data)
expected = dict(
node_id=self.user,
success="User <{0}> created".format(self.user)
)
self.assertEqual(response, expected)
# query graph directly and verify node exists
node = graph.find_node("User", self.user)
self.assertEqual(node.properties["node_id"], self.user)
# clean up
graph.delete_user(self.user)
