def test_move_up_skips_end_node(self):
traversal = DiagnosisTraversal.objects.create(
graph_version="v1",
current_node_id="INSCOPE",
nodes=[
self.mocked_graph.get_node_dict("2a"),
self.mocked_graph.get_node_dict("3ab"),
self.mocked_graph.get_node_dict("INSCOPE"),
],
category=self.category,
matter_type1=self.matter_type1,
matter_type2=self.matter_type2,
state=DIAGNOSIS_SCOPE.INSCOPE,
)
serializer = DiagnosisSerializer(instance=traversal)
serializer.move_up()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, "2a")
self.assertItemsEqual(traversal.nodes,
[self.mocked_graph.get_node_dict("2a")])
self.assertEqual(traversal.category, None)
self.assertEqual(traversal.matter_type1, None)
self.assertEqual(traversal.matter_type2, None)
self.assertEqual(traversal.state, DIAGNOSIS_SCOPE.UNKNOWN)
def test_move_down_processes_end_node_directly(self):
traversal = DiagnosisTraversal.objects.create(
graph_version='v1',
current_node_id='2a',
nodes=[
self.mocked_graph.get_node_dict('2a')
]
)
serializer = DiagnosisSerializer(
instance=traversal, data={
'current_node_id': '3ab'
}
)
self.assertTrue(serializer.is_valid())
serializer.save()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, 'INSCOPE')
self.assertItemsEqual(traversal.nodes, [
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3ab'),
self.mocked_graph.get_node_dict('INSCOPE')
])
self.assertEqual(traversal.category.code, 'debt')
self.assertEqual(traversal.state, DIAGNOSIS_SCOPE.INSCOPE)
def test_move_up_skips_end_node(self):
traversal = DiagnosisTraversal.objects.create(
graph_version="v1",
current_node_id="INSCOPE",
nodes=[
self.mocked_graph.get_node_dict("2a"),
self.mocked_graph.get_node_dict("3ab"),
self.mocked_graph.get_node_dict("INSCOPE"),
],
category=self.category,
matter_type1=self.matter_type1,
matter_type2=self.matter_type2,
state=DIAGNOSIS_SCOPE.INSCOPE,
)
serializer = DiagnosisSerializer(instance=traversal)
serializer.move_up()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, "2a")
self.assertItemsEqual(traversal.nodes, [self.mocked_graph.get_node_dict("2a")])
self.assertEqual(traversal.category, None)
self.assertEqual(traversal.matter_type1, None)
self.assertEqual(traversal.matter_type2, None)
self.assertEqual(traversal.state, DIAGNOSIS_SCOPE.UNKNOWN)
def test_move_down_processes_end_node_directly(self):
traversal = DiagnosisTraversal.objects.create(
graph_version="v1",
current_node_id="2a",
nodes=[self.mocked_graph.get_node_dict("2a")])
serializer = DiagnosisSerializer(instance=traversal,
data={"current_node_id": "3ab"})
self.assertTrue(serializer.is_valid())
serializer.save()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, "INSCOPE")
self.assertItemsEqual(
traversal.nodes,
[
self.mocked_graph.get_node_dict("2a"),
self.mocked_graph.get_node_dict("3ab"),
self.mocked_graph.get_node_dict("INSCOPE"),
],
)
self.assertEqual(traversal.state, DIAGNOSIS_SCOPE.INSCOPE)
self.assertEqual(traversal.category.code, self.category.code)
self.assertEqual(traversal.matter_type1.code, self.matter_type1.code)
self.assertEqual(traversal.matter_type2.code, self.matter_type2.code)
def test_move_down_updates_traversal(self):
"""
Tests that moving down updates the traversal object
"""
traversal = DiagnosisTraversal.objects.create(
graph_version='v1',
current_node_id='2a',
nodes=[
self.mocked_graph.get_node_dict('2a')
]
)
serializer = DiagnosisSerializer(
instance=traversal, data={
'current_node_id': '3aa'
}
)
self.assertTrue(serializer.is_valid())
serializer.save()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, '3aa')
self.assertItemsEqual(traversal.nodes, [
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3aa')
])
def test_move_up_updates_traversal(self):
traversal = DiagnosisTraversal.objects.create(
graph_version="v1",
current_node_id="3aa",
nodes=[self.mocked_graph.get_node_dict("2a"), self.mocked_graph.get_node_dict("3aa")],
)
serializer = DiagnosisSerializer(instance=traversal)
serializer.move_up()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, "2a")
self.assertItemsEqual(traversal.nodes, [self.mocked_graph.get_node_dict("2a")])
def test_move_up_updates_traversal(self):
traversal = DiagnosisTraversal.objects.create(
graph_version='v1',
current_node_id='3aa',
nodes=[
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3aa')
])
serializer = DiagnosisSerializer(instance=traversal)
serializer.move_up()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, '2a')
self.assertItemsEqual(traversal.nodes,
[self.mocked_graph.get_node_dict('2a')])
def test_get_with_inscope_diagnosis_different_version(self):
"""
Tests that 'version_in_conflict' is False for inscope
old diagnosis
"""
traversal = DiagnosisTraversal.objects.create(
current_node_id="INSCOPE",
graph_version="old_v1",
nodes=[
self.mocked_graph.get_node_dict("2a"),
self.mocked_graph.get_node_dict("3ab"),
self.mocked_graph.get_node_dict("INSCOPE"),
],
category=self.category,
matter_type1=self.matter_type1,
matter_type2=self.matter_type2,
state=DIAGNOSIS_SCOPE.INSCOPE,
)
serializer = DiagnosisSerializer(instance=traversal)
data = serializer.data
self.assertItemsEqual(
data["nodes"],
[
self.mocked_graph.get_node_dict("2a"),
self.mocked_graph.get_node_dict("3ab"),
self.mocked_graph.get_node_dict("INSCOPE"),
],
)
self.assertItemsEqual(data["choices"], [])
self.assertItemsEqual(data["category"], self.category.code)
self.assertItemsEqual(data["matter_type1"], self.matter_type1.code)
self.assertItemsEqual(data["matter_type2"], self.matter_type2.code)
self.assertEqual(data["version_in_conflict"], False)
def test_move_down_updates_traversal(self):
"""
Tests that moving down updates the traversal object
"""
traversal = DiagnosisTraversal.objects.create(
graph_version="v1", current_node_id="2a", nodes=[self.mocked_graph.get_node_dict("2a")]
)
serializer = DiagnosisSerializer(instance=traversal, data={"current_node_id": "3aa"})
self.assertTrue(serializer.is_valid())
serializer.save()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, "3aa")
self.assertItemsEqual(
traversal.nodes, [self.mocked_graph.get_node_dict("2a"), self.mocked_graph.get_node_dict("3aa")]
)
def test_move_up_updates_traversal(self):
traversal = DiagnosisTraversal.objects.create(
graph_version="v1",
current_node_id="3aa",
nodes=[
self.mocked_graph.get_node_dict("2a"),
self.mocked_graph.get_node_dict("3aa")
],
)
serializer = DiagnosisSerializer(instance=traversal)
serializer.move_up()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, "2a")
self.assertItemsEqual(traversal.nodes,
[self.mocked_graph.get_node_dict("2a")])
def test_move_up_updates_traversal(self):
traversal = DiagnosisTraversal.objects.create(
graph_version='v1',
current_node_id='3aa',
nodes=[
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3aa')
]
)
serializer = DiagnosisSerializer(instance=traversal)
serializer.move_up()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, '2a')
self.assertItemsEqual(traversal.nodes, [
self.mocked_graph.get_node_dict('2a')
])
def test_move_down_processes_end_node_directly(self):
traversal = DiagnosisTraversal.objects.create(
graph_version='v1',
current_node_id='2a',
nodes=[self.mocked_graph.get_node_dict('2a')])
serializer = DiagnosisSerializer(instance=traversal,
data={'current_node_id': '3ab'})
self.assertTrue(serializer.is_valid())
serializer.save()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, 'INSCOPE')
self.assertItemsEqual(traversal.nodes, [
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3ab'),
self.mocked_graph.get_node_dict('INSCOPE')
])
self.assertEqual(traversal.category.code, 'debt')
self.assertEqual(traversal.state, DIAGNOSIS_SCOPE.INSCOPE)
def test_move_down_updates_traversal(self):
"""
Tests that moving down updates the traversal object
"""
traversal = DiagnosisTraversal.objects.create(
graph_version="v1",
current_node_id="2a",
nodes=[self.mocked_graph.get_node_dict("2a")])
serializer = DiagnosisSerializer(instance=traversal,
data={"current_node_id": "3aa"})
self.assertTrue(serializer.is_valid())
serializer.save()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, "3aa")
self.assertItemsEqual(traversal.nodes, [
self.mocked_graph.get_node_dict("2a"),
self.mocked_graph.get_node_dict("3aa")
])
def test_move_down_updates_traversal(self):
"""
Tests that moving down updates the traversal object
"""
traversal = DiagnosisTraversal.objects.create(
graph_version='v1',
current_node_id='2a',
nodes=[self.mocked_graph.get_node_dict('2a')])
serializer = DiagnosisSerializer(instance=traversal,
data={'current_node_id': '3aa'})
self.assertTrue(serializer.is_valid())
serializer.save()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, '3aa')
self.assertItemsEqual(traversal.nodes, [
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3aa')
])
def test_move_up_skips_end_node(self):
traversal = DiagnosisTraversal.objects.create(
graph_version='v1',
current_node_id='INSCOPE',
nodes=[
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3ab'),
self.mocked_graph.get_node_dict('INSCOPE'),
],
category=self.category,
state=DIAGNOSIS_SCOPE.INSCOPE)
serializer = DiagnosisSerializer(instance=traversal)
serializer.move_up()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, '2a')
self.assertItemsEqual(traversal.nodes,
[self.mocked_graph.get_node_dict('2a')])
self.assertEqual(traversal.category, None)
self.assertEqual(traversal.state, DIAGNOSIS_SCOPE.UNKNOWN)
def test_move_down_processes_end_node_directly(self):
traversal = DiagnosisTraversal.objects.create(
graph_version="v1", current_node_id="2a", nodes=[self.mocked_graph.get_node_dict("2a")]
)
serializer = DiagnosisSerializer(instance=traversal, data={"current_node_id": "3ab"})
self.assertTrue(serializer.is_valid())
serializer.save()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, "INSCOPE")
self.assertItemsEqual(
traversal.nodes,
[
self.mocked_graph.get_node_dict("2a"),
self.mocked_graph.get_node_dict("3ab"),
self.mocked_graph.get_node_dict("INSCOPE"),
],
)
self.assertEqual(traversal.state, DIAGNOSIS_SCOPE.INSCOPE)
self.assertEqual(traversal.category.code, self.category.code)
self.assertEqual(traversal.matter_type1.code, self.matter_type1.code)
self.assertEqual(traversal.matter_type2.code, self.matter_type2.code)
def test_move_up_skips_end_node(self):
traversal = DiagnosisTraversal.objects.create(
graph_version='v1',
current_node_id='INSCOPE',
nodes=[
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3ab'),
self.mocked_graph.get_node_dict('INSCOPE'),
],
category=self.category,
state=DIAGNOSIS_SCOPE.INSCOPE
)
serializer = DiagnosisSerializer(instance=traversal)
serializer.move_up()
traversal = DiagnosisTraversal.objects.get(pk=traversal.pk)
self.assertEqual(traversal.current_node_id, '2a')
self.assertItemsEqual(traversal.nodes, [
self.mocked_graph.get_node_dict('2a')
])
self.assertEqual(traversal.category, None)
self.assertEqual(traversal.state, DIAGNOSIS_SCOPE.UNKNOWN)
def test_get_with_ongoing_diagnosis_different_version(self):
"""
Tests that the get request from the serializer returns
[] when in the middle of a diagnosis
"""
traversal = DiagnosisTraversal.objects.create(
current_node_id='2a',
graph_version='old_v1',
nodes=[self.mocked_graph.get_node_dict('2a')])
serializer = DiagnosisSerializer(instance=traversal)
data = serializer.data
self.assertItemsEqual(data['nodes'],
[self.mocked_graph.get_node_dict('2a')])
self.assertItemsEqual(data['choices'], [])
self.assertEqual(data['current_node_id'], '2a')
self.assertEqual(data['state'], DIAGNOSIS_SCOPE.UNKNOWN)
self.assertEqual(data['category'], None)
self.assertEqual(data['version_in_conflict'], True)
def test_get_with_new_diagnosis(self):
"""
Tests that the get request from the serializer returns
expected values when new diagnosis
"""
traversal = DiagnosisTraversal.objects.create(graph_version='v1')
serializer = DiagnosisSerializer(instance=traversal)
data = serializer.data
self.assertEqual(data['nodes'], None)
self.assertItemsEqual(data['choices'], [
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('2b')
])
self.assertEqual(data['current_node_id'], '')
self.assertEqual(data['state'], DIAGNOSIS_SCOPE.UNKNOWN)
self.assertEqual(data['category'], None)
self.assertEqual(data['version_in_conflict'], False)
def test_get_with_new_diagnosis(self):
"""
Tests that the get request from the serializer returns
expected values when new diagnosis
"""
traversal = DiagnosisTraversal.objects.create(graph_version="v1")
serializer = DiagnosisSerializer(instance=traversal)
data = serializer.data
self.assertEqual(data["nodes"], None)
self.assertItemsEqual(data["choices"], [
self.mocked_graph.get_node_dict("2a"),
self.mocked_graph.get_node_dict("2b")
])
self.assertEqual(data["current_node_id"], "")
self.assertEqual(data["state"], DIAGNOSIS_SCOPE.UNKNOWN)
self.assertEqual(data["category"], None)
self.assertEqual(data["matter_type1"], None)
self.assertEqual(data["matter_type2"], None)
self.assertEqual(data["version_in_conflict"], False)
def test_get_with_ongoing_diagnosis_different_version(self):
"""
Tests that the get request from the serializer returns
[] when in the middle of a diagnosis
"""
traversal = DiagnosisTraversal.objects.create(
current_node_id="2a",
graph_version="old_v1",
nodes=[self.mocked_graph.get_node_dict("2a")])
serializer = DiagnosisSerializer(instance=traversal)
data = serializer.data
self.assertItemsEqual(data["nodes"],
[self.mocked_graph.get_node_dict("2a")])
self.assertItemsEqual(data["choices"], [])
self.assertEqual(data["current_node_id"], "2a")
self.assertEqual(data["state"], DIAGNOSIS_SCOPE.UNKNOWN)
self.assertEqual(data["category"], None)
self.assertEqual(data["matter_type1"], None)
self.assertEqual(data["matter_type2"], None)
self.assertEqual(data["version_in_conflict"], True)
def test_get_with_inscope_diagnosis_different_version(self):
"""
Tests that 'version_in_conflict' is False for inscope
old diagnosis
"""
traversal = DiagnosisTraversal.objects.create(
current_node_id='INSCOPE',
graph_version='old_v1',
nodes=[
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3ab'),
self.mocked_graph.get_node_dict('INSCOPE'),
],
state=DIAGNOSIS_SCOPE.INSCOPE)
serializer = DiagnosisSerializer(instance=traversal)
data = serializer.data
self.assertItemsEqual(data['nodes'], [
self.mocked_graph.get_node_dict('2a'),
self.mocked_graph.get_node_dict('3ab'),
self.mocked_graph.get_node_dict('INSCOPE'),
])
self.assertItemsEqual(data['choices'], [])
self.assertEqual(data['version_in_conflict'], False)
def test_move_up_fails_if_current_node_is_root(self):
traversal = DiagnosisTraversal.objects.create(graph_version='v1')
serializer = DiagnosisSerializer(instance=traversal)
self.assertRaises(ParseError, serializer.move_up)