git.iko.soy/typeable/bloodhound
on index-hack help sign in
typeable/bloodhound/tests/V1/Test/Suggest.hs
22 lines 973 B history raw
1{-# LANGUAGE OverloadedStrings #-}
3module Test.Suggest where
5import Test.Common
6import Test.Import
8spec :: Spec
9spec =
10 describe "Suggest" $
11 it "returns a search suggestion using the phrase suggester" $ withTestEnv $ do
12 _ <- insertData
13 let phraseSuggester = mkPhraseSuggester (FieldName "message")
14 namedSuggester = Suggest "Use haskel" "suggest_name" (SuggestTypePhraseSuggester phraseSuggester)
15 search' = mkSearch Nothing Nothing
16 search = search' { suggestBody = Just namedSuggester }
17 expectedText = Just "use haskell"
18 resp <- searchByIndex testIndex search
19 parsed <- parseEsResponse resp :: BH IO (Either EsError (SearchResult Tweet))
20 case parsed of
21 Left e -> liftIO $ expectationFailure ("Expected an search suggestion but got " <> show e)
22 Right sr -> liftIO $ (suggestOptionsText . head . suggestResponseOptions . head . nsrResponses <$> suggest sr) `shouldBe` expectedText