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_get_many(reader: ReaderService, read_db: SqlReadDatabaseBackendService): model = MagicMock() result = {"c/1": model} read_db.get_many = get_many = MagicMock(return_value=result) parts = [ GetManyRequestPart("a", "1", ["field1"]), GetManyRequestPart("b", "1", ["field2"]), ] request = GetManyRequest(parts, ["field"]) assert reader.get_many(request) == {"a": {}, "b": {}, "c": {"1": model}} read_db.get_context.assert_called() get_many.assert_called_with( ["a/1", "b/1"], {"a/1": ["field1", "field"], "b/1": ["field2", "field"]}, DeletedModelsBehaviour.NO_DELETED, )
def test_get_many_with_position( reader: ReaderService, read_db: SqlReadDatabaseBackendService ): model = MagicMock() result = {"c/1": model} deleted_map = {"a/1": False, "b/1": False} read_db.get_deleted_status = gds = MagicMock(return_value=deleted_map) read_db.build_models_ignore_deleted = bmid = MagicMock(return_value=result) reader.apply_mapped_fields_multi = amfm = MagicMock(return_value=result) parts = [ GetManyRequestPart("a", "1", ["field1"]), GetManyRequestPart("b", "1", ["field2"]), ] request = GetManyRequest(parts, ["field"], 42) assert reader.get_many(request) == {"a": {}, "b": {}, "c": {"1": model}} gds.assert_called_with(["a/1", "b/1"], 42) bmid.assert_called_with(["a/1", "b/1"], 42) amfm.assert_called_with( result, {"a/1": ["field1", "field"], "b/1": ["field2", "field"]} )