def test_get(self) -> None:
fqid = FullQualifiedId(Collection("fakeModel"), 1)
fields = ["a", "b", "c"]
command = commands.Get(fqid=fqid, mappedFields=fields)
self.engine.retrieve.return_value = (
json.dumps({
"f": 1,
"meta_deleted": False,
"meta_position": 1
}),
200,
)
partial_model = self.db.get(fqid, fields)
assert command.get_raw_data() == {
"fqid": str(fqid),
"mapped_fields": fields
}
assert partial_model is not None
self.engine.retrieve.assert_called_with("get", command.data)
def test_get(self) -> None:
fqid = FullQualifiedId(Collection("fakeModel"), 1)
fields = set(["a", "b", "c"])
command = commands.Get(fqid=fqid, mapped_fields=fields)
self.engine.retrieve.return_value = (
json.dumps({"f": 1, "meta_deleted": False, "meta_position": 1}),
200,
)
partial_model = self.db.get(fqid, list(fields))
raw_data = command.get_raw_data()
assert raw_data["fqid"] == str(fqid)
assert isinstance(raw_data["mapped_fields"], list)
assert set(raw_data["mapped_fields"]) == fields
assert partial_model is not None
self.engine.retrieve.assert_called()
call_args = self.engine.retrieve.call_args[0]
assert call_args[0] == "get"
data = json.loads(call_args[1])
assert data["fqid"] == str(fqid)
assert set(data["mapped_fields"]) == fields