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