def test_get_source_method_as_json(self, mocker): mocked_get_elastic = mocker.patch.object(ModelMixin, "get_elastic") mocked_get_elastic.return_value = dict( _id=1, _index="dor", _source=dict(name="some-variable")) model_mixin = ModelMixin() result = model_mixin.get_source(as_json=True) assert result == json.dumps({"name": "some-variable"})
def test_get_source_method(self, mocker): mocked_get_elastic = mocker.patch.object(ModelMixin, "get_elastic") mocked_get_elastic.return_value = dict( _id=1, _index="dor", _source=dict(name="some-variable")) model_mixin = ModelMixin() result = model_mixin.get_source() assert result["name"] == "some-variable"