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"