def test_are_nodes_connected(self):
self.assertTrue(are_nodes_connected(
get_state_by_name("New"),
get_state_by_name("First message")
))
self.assertTrue(are_nodes_connected(
get_state_by_name("Negative response"),
get_state_by_name("First message"))
)
self.assertFalse(are_nodes_connected(
get_state_by_name("Negative response"),
get_state_by_name("No response"))
)
self.assertFalse(are_nodes_connected(
get_state_by_name("Hired"),
get_state_by_name("New"))
)
def change_state(request):
try:
person_id = int(request.POST['person_id'])
new_state_id = int(request.POST['new_state_id'])
except KeyError:
raise Http404
person = get_object_or_404(Person, id=person_id)
if are_nodes_connected(new_state_id, person.state.pk):
person.update_state(new_state_id)
return HttpResponse(200, content_type="plain/text")
raise Http404
