Ejemplo n.º 1
0
    async def test_run(self, event_loop):
        change = SetAnnotationsChange(1, [],
                                      params={
                                          "id": "id",
                                          "entity-type": "entity_type",
                                          "annotations": "annotations"
                                      })

        entity = mock.Mock()
        entity.set_annotations = base.AsyncMock(return_value="annotations1")

        state = mock.Mock()
        state.get_entity = mock.Mock(return_value=entity)

        model = mock.Mock()
        model.state = state

        context = mock.Mock()
        context.resolve = mock.Mock(side_effect=['application1'])
        context.model = model

        result = await change.run(context)
        assert result == "annotations1"

        entity.set_annotations.assert_called_once()
        entity.set_annotations.assert_called_with("annotations")
Ejemplo n.º 2
0
 def test_list_params(self):
     change = SetAnnotationsChange(1, [], params=["id", "entity_type", "annotations"])
     self.assertEqual({"change_id": 1,
                       "requires": [],
                       "id": "id",
                       "entity_type": "entity_type",
                       "annotations": "annotations"}, change.__dict__)
Ejemplo n.º 3
0
 def test_dict_params_missing_data(self):
     change = SetAnnotationsChange(1, [], params={"id": "id",
                                                  "entity-type": "entity_type"})
     self.assertEqual({"change_id": 1,
                       "requires": [],
                       "id": "id",
                       "entity_type": "entity_type",
                       "annotations": None}, change.__dict__)
Ejemplo n.º 4
0
 def test_method(self):
     self.assertEqual("setAnnotations", SetAnnotationsChange.method())