def test_apply_mapped_fields_multi(reader: ReaderService):
result = {
"a/1": {
"f1": "a",
"f2": "b",
"f": "c"
},
"b/1": {
"f3": "a",
"f4": "b",
"f": "c"
},
}
assert reader.apply_mapped_fields_multi(result, {
"a/1": ["f1", "f"],
"b/1": ["f3", "f"]
}) == {
"a/1": {
"f1": "a",
"f": "c"
},
"b/1": {
"f3": "a",
"f": "c"
}
}
def test_get_many_with_position_deleted_no_deleted(
reader: ReaderService, read_db: SqlReadDatabaseBackendService):
deleted_map = {"a/1": True, "b/1": False}
read_db.get_deleted_status = gds = MagicMock(return_value=deleted_map)
read_db.build_models_ignore_deleted = bmid = MagicMock()
reader.apply_mapped_fields_multi = MagicMock()
parts = [
GetManyRequestPart("a", "1", ["field1"]),
GetManyRequestPart("b", "1", ["field2"]),
]
request = GetManyRequest(parts, ["field"], 42)
reader.get_many(request)
gds.assert_called_with(["a/1", "b/1"], 42)
bmid.assert_called_with(["b/1"], 42)
def test_apply_mapped_fields_multi_no_fields(reader: ReaderService):
result = MagicMock()
assert reader.apply_mapped_fields_multi(result, {}) == result
