mirror of
https://github.com/typeable/bloodhound.git
synced 2024-12-15 10:02:12 +03:00
81 lines
2.7 KiB
Haskell
81 lines
2.7 KiB
Haskell
{-# LANGUAGE DeriveGeneric #-}
|
|
|
|
module Main where
|
|
|
|
import Database.Bloodhound.Client
|
|
import Data.Aeson
|
|
import Data.DeriveTH
|
|
import Data.Either (Either(..))
|
|
import Data.Maybe (fromJust)
|
|
import Data.Time.Calendar (Day(..))
|
|
import Data.Time.Clock (secondsToDiffTime, UTCTime(..))
|
|
import Data.Text (Text)
|
|
import GHC.Generics (Generic)
|
|
import Network.HTTP.Conduit
|
|
import qualified Network.HTTP.Types.Status as NHTS
|
|
import Test.Hspec
|
|
|
|
|
|
testServer = Server "http://localhost:9200"
|
|
testIndex = "twitter"
|
|
validateStatus resp expected = (NHTS.statusCode $ responseStatus resp) `shouldBe` (expected :: Int)
|
|
createExampleIndex = createIndex testServer defaultIndexSettings testIndex
|
|
deleteExampleIndex = deleteIndex testServer testIndex
|
|
|
|
data Tweet = Tweet { user :: Text
|
|
, postDate :: UTCTime
|
|
, message :: Text }
|
|
deriving (Eq, Generic, Show)
|
|
|
|
instance ToJSON Tweet
|
|
instance FromJSON Tweet
|
|
|
|
exampleTweet = Tweet { user = "bitemyapp"
|
|
, postDate = UTCTime
|
|
(ModifiedJulianDay 55000)
|
|
(secondsToDiffTime 10)
|
|
, message = "Use haskell!" }
|
|
|
|
insertData :: IO ()
|
|
insertData = do
|
|
let encoded = encode exampleTweet
|
|
_ <- deleteExampleIndex
|
|
created <- createExampleIndex
|
|
docCreated <- indexDocument (Server "http://localhost:9200") "twitter" "tweet" exampleTweet "1"
|
|
_ <- refreshIndex testServer "twitter"
|
|
return ()
|
|
|
|
queryTweet :: IO (Either String Tweet)
|
|
queryTweet = do
|
|
let queryFilter = BoolFilter (MustMatch (Term "user" "bitemyapp") False)
|
|
<||> IdentityFilter
|
|
let search = Search Nothing (Just queryFilter)
|
|
reply <- searchByIndex testServer "twitter" search
|
|
let result = eitherDecode (responseBody reply) :: Either String (SearchResult Tweet)
|
|
let myTweet = fmap (hitSource . head . hits . searchHits) result
|
|
return myTweet
|
|
|
|
main :: IO ()
|
|
main = hspec $ do
|
|
describe "index create/delete API" $ do
|
|
it "creates and then deletes the requested index" $ do
|
|
-- priming state.
|
|
_ <- deleteExampleIndex
|
|
resp <- createExampleIndex
|
|
deleteResp <- deleteExampleIndex
|
|
validateStatus resp 200
|
|
validateStatus deleteResp 200
|
|
|
|
describe "document API" $ do
|
|
it "indexes, gets, and then deletes the generated document" $ do
|
|
_ <- insertData
|
|
docInserted <- getDocument (Server "http://localhost:9200") "twitter" "tweet" "1"
|
|
let newTweet = eitherDecode (responseBody docInserted) :: Either String (EsResult Tweet)
|
|
fmap _source newTweet `shouldBe` Right exampleTweet
|
|
|
|
describe "document filtering" $ do
|
|
it "returns documents expected from composed filters" $ do
|
|
_ <- insertData
|
|
myTweet <- queryTweet
|
|
myTweet `shouldBe` Right exampleTweet
|