From 6238b2fbfa6008fdf9bc6c85409fbd13e9b5e4e9 Mon Sep 17 00:00:00 2001 From: Eugen Wissner Date: Wed, 3 Jul 2019 17:54:50 +0200 Subject: Add nullable types --- tests/Test/StarWars/Schema.hs | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'tests/Test/StarWars/Schema.hs') diff --git a/tests/Test/StarWars/Schema.hs b/tests/Test/StarWars/Schema.hs index 0abda7b..daa000e 100644 --- a/tests/Test/StarWars/Schema.hs +++ b/tests/Test/StarWars/Schema.hs @@ -24,15 +24,18 @@ schema = hero :| [human, droid] hero :: MonadPlus m => Resolver m hero = Schema.objectA "hero" $ \case [] -> character artoo - [Argument "episode" (ValueInt n)] -> character . getHero $ fromIntegral n [Argument "episode" (ValueEnum "NEWHOPE")] -> character $ getHero 4 [Argument "episode" (ValueEnum "EMPIRE" )] -> character $ getHero 5 [Argument "episode" (ValueEnum "JEDI" )] -> character $ getHero 6 _ -> ActionT $ throwE "Invalid arguments." human :: MonadPlus m => Resolver m -human = Schema.objectA "human" $ \case - [Argument "id" (ValueString i)] -> character =<< lift (getHuman i) +human = Schema.nullableObject "human" $ \case + [Argument "id" (ValueString i)] -> do + humanCharacter <- lift $ return $ getHuman i >>= Just + case humanCharacter of + Nothing -> return Nothing + Just e -> Just <$> character e _ -> ActionT $ throwE "Invalid arguments." droid :: MonadPlus m => Resolver m -- cgit v1.2.3