def test_set_elastic_method_with_update(self, mocker):
mocked_es_update = mocker.patch.object(Elasticsearch, "update")
model_mixin = ModelMixin()
model_mixin.id = 1
result = model_mixin.set_elastic(body={"field": "updated-value"},
update=True)
mocked_es_update.assert_called_once()
def test_get_attribute_method_fails(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_attribute("alternative", None)
assert result is None
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"
def test_get_elastic_method(self, mocker):
mocked_es_get = mocker.patch.object(Elasticsearch, "get")
model_mixin = ModelMixin()
model_mixin.id = 1
model_mixin.get_elastic()
mocked_es_get.assert_called_once()
def test_set_elastic_method(self, mocker):
mocked_es_index = mocker.patch.object(Elasticsearch, "index")
model_mixin = ModelMixin()
model_mixin.id = 1
result = model_mixin.set_elastic(body={"field": "value"})
mocked_es_index.assert_called_once()
def test_delete_elastic_method(self, mocker):
mocked_es_delete = mocker.patch.object(Elasticsearch, "delete")
model_mixin = ModelMixin()
model_mixin.id = 1
model_mixin.delete_elastic()
mocked_es_delete.assert_called_once()
def test_get_elastic_method_without_search(self):
model_mixin = ModelMixin()
model_mixin.elastic = dict(_id=1, status="not found")
result = model_mixin.get_elastic()
assert result == model_mixin.elastic
def test_get_elastic_method_as_json(self):
model_mixin = ModelMixin()
model_mixin.elastic = dict(_id=1, status="not found")
expected = json.dumps(model_mixin.elastic)
result = model_mixin.get_elastic(as_json=True)
assert result == expected