55 lines
1.9 KiB
Haskell
55 lines
1.9 KiB
Haskell
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
import Test.Hspec
|
|
import UserManager
|
|
import Database.SQLite.Simple
|
|
import Control.Exception (bracket_)
|
|
import System.Directory (removeFile, doesFileExist)
|
|
import Data.String (fromString)
|
|
|
|
main :: IO ()
|
|
main = hspec $ do
|
|
describe "UserManager" $ do
|
|
beforeAll_ setupDB $
|
|
afterAll_ cleanupDB $ do
|
|
it "initializes the database" $ do
|
|
conn <- open "users.db"
|
|
tables <- query_ conn "SELECT name FROM sqlite_master WHERE type='table' AND name='users';" :: IO [Only String]
|
|
close conn
|
|
tables `shouldBe` [Only "users"]
|
|
|
|
it "adds a user" $ do
|
|
addUser "test_user" "test@example.com" "password"
|
|
users <- findUsers "test_user"
|
|
users `shouldBe` [User "test_user" "test@example.com" "password"]
|
|
|
|
it "lists users" $ do
|
|
addUser "test_user1" "test1@example.com" "password1"
|
|
users <- listUsers
|
|
users `shouldBe` [User "test_user" "test@example.com" "password", User "test_user1" "test1@example.com" "password1" ]
|
|
|
|
it "finds users by name" $ do
|
|
addUser "unique_user" "unique@example.com" "password"
|
|
users <- findUsers "unique_user"
|
|
users `shouldBe` [User "unique_user" "unique@example.com" "password"]
|
|
|
|
it "finds users by email" $ do
|
|
addUser "another_user" "another@example.com" "password"
|
|
users <- findUsers "another@example.com"
|
|
users `shouldBe` [User "another_user" "another@example.com" "password"]
|
|
|
|
it "deletes a user" $ do
|
|
addUser "delete_user" "delete@example.com" "password"
|
|
deleteUser "delete_user"
|
|
users <- listUsers
|
|
users `shouldSatisfy` all (\user -> userName user /= "delete_user")
|
|
|
|
setupDB :: IO ()
|
|
setupDB = do
|
|
conn <- open "users.db"
|
|
execute_ conn "CREATE TABLE IF NOT EXISTS users (name TEXT PRIMARY KEY, email TEXT, password TEXT)"
|
|
close conn
|
|
|
|
cleanupDB :: IO ()
|
|
cleanupDB = removeFile "users.db"
|