Highlights.hs

{-# LANGUAGE OverloadedStrings #-}

module Test.Highlights where

import qualified Data.Map as M
import Test.Common
import Test.Import

initHighlights :: Text -> BH IO (Either EsError (Maybe HitHighlight))
initHighlights fieldName = do
  _ <- insertData
  _ <- insertOther
  let query = QueryMatchQuery $ mkMatchQuery (FieldName fieldName) (QueryString "haskell")
  let testHighlight = Highlights Nothing [FieldHighlight (FieldName fieldName) Nothing]
  let search = mkHighlightSearch (Just query) testHighlight
  searchTweetHighlight search

spec :: Spec
spec =
  describe "Highlights API" $ do
    it "returns highlight from query when there should be one" $
      withTestEnv $ do
        myHighlight <- initHighlights "message"
        liftIO $
          myHighlight
            `shouldBe` Right (Just (M.fromList [("message", ["Use <em>haskell</em>!"])]))

    it "doesn't return highlight from a query when it shouldn't" $
      withTestEnv $ do
        myHighlight <- initHighlights "user"
        liftIO $
          myHighlight
            `shouldBe` Right Nothing