Python DiagnosisSerializer Examples

Example #1

File: test_serializers.py Project: uk-gov-mirror/ministryofjustice.cla_backend

    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)

Example #2

File: test_serializers.py Project: doismellburning/cla_backend

    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)

Example #3

File: test_serializers.py Project: ministryofjustice/cla_backend

    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)

Example #4

File: test_serializers.py Project: uk-gov-mirror/ministryofjustice.cla_backend

    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)

Example #5

File: test_serializers.py Project: doismellburning/cla_backend

    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')
        ])

Example #6

File: test_serializers.py Project: ministryofjustice/cla_backend

    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")])

Example #7

    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')])

Example #8

File: test_serializers.py Project: uk-gov-mirror/ministryofjustice.cla_backend

 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)

Example #9

File: test_serializers.py Project: ministryofjustice/cla_backend

    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")]
        )

Example #10

File: test_serializers.py Project: uk-gov-mirror/ministryofjustice.cla_backend

    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")])

Example #11

File: test_serializers.py Project: doismellburning/cla_backend

    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')
        ])

Example #12

    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)

Example #13

File: test_serializers.py Project: uk-gov-mirror/ministryofjustice.cla_backend

    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")
        ])

Example #14

    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')
        ])

Example #15

    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)

Example #16

File: test_serializers.py Project: ministryofjustice/cla_backend

    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)

Example #17

File: test_serializers.py Project: doismellburning/cla_backend

    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)

Example #18

 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)

Example #19

    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)

Example #20

File: test_serializers.py Project: uk-gov-mirror/ministryofjustice.cla_backend

    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)

Example #21

File: test_serializers.py Project: uk-gov-mirror/ministryofjustice.cla_backend

 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)

Example #22

 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)

Example #23

    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)