151 lines
3.3 KiB
Haskell
151 lines
3.3 KiB
Haskell
{-# LANGUAGE OverloadedStrings, QuasiQuotes #-}
|
|
|
|
module TypesSpec (spec) where
|
|
|
|
import Data.Aeson (decode, encode)
|
|
import Data.ByteString.Lazy (ByteString)
|
|
import Lens.Micro ((&), (.~))
|
|
import Test.Hspec (Spec, context, describe, it, shouldBe)
|
|
import Text.RawString.QQ (r)
|
|
import Mtlstats.Types
|
|
|
|
spec :: Spec
|
|
spec = describe "Mtlstats.Types" $ do
|
|
pPointsSpec
|
|
playerSpec
|
|
goalieSpec
|
|
databaseSpec
|
|
|
|
pPointsSpec :: Spec
|
|
pPointsSpec = describe "pPoints" $ mapM_
|
|
(\(goals, assists, points) -> let
|
|
desc = "goals: " ++ show goals ++
|
|
", assists: " ++ show assists
|
|
stats = newPlayerStats &
|
|
psGoals .~ goals &
|
|
psAssists .~ assists
|
|
in context desc $
|
|
it ("should be " ++ show points) $
|
|
pPoints stats `shouldBe` points)
|
|
-- goals, assists, points
|
|
[ ( 0, 0, 0 )
|
|
, ( 1, 0, 1 )
|
|
, ( 0, 1, 1 )
|
|
, ( 2, 3, 5 )
|
|
]
|
|
|
|
playerSpec :: Spec
|
|
playerSpec = describe "Player" $ do
|
|
|
|
describe "decode" $
|
|
it "should decode" $
|
|
decode playerJSON `shouldBe` Just player
|
|
|
|
describe "encode" $
|
|
it "should encode" $
|
|
decode (encode player) `shouldBe` Just player
|
|
|
|
goalieSpec :: Spec
|
|
goalieSpec = describe "Goalie" $ do
|
|
|
|
describe "decode" $
|
|
it "should decode" $
|
|
decode goalieJSON `shouldBe` Just goalie
|
|
|
|
describe "encode" $
|
|
it "should encode" $
|
|
decode (encode goalie) `shouldBe` Just goalie
|
|
|
|
databaseSpec :: Spec
|
|
databaseSpec = describe "Database" $ do
|
|
|
|
describe "decode" $
|
|
it "should decode" $
|
|
decode dbJSON `shouldBe` Just db
|
|
|
|
describe "encode" $
|
|
it "should encode" $
|
|
decode (encode db) `shouldBe` Just db
|
|
|
|
player :: Player
|
|
player = newPlayer 1 "Joe" "centre"
|
|
& pYtd . psGoals .~ 2
|
|
& pYtd . psAssists .~ 3
|
|
& pYtd . psPMin .~ 4
|
|
& pLifetime . psGoals .~ 5
|
|
& pLifetime . psAssists .~ 6
|
|
& pLifetime . psPMin .~ 7
|
|
|
|
goalie :: Goalie
|
|
goalie = newGoalie 1 "Joe"
|
|
& gYtd . gsGames .~ 2
|
|
& gYtd . gsMinsPlayed .~ 3
|
|
& gYtd . gsGoalsAllowed .~ 4
|
|
& gYtd . gsGoalsAgainst .~ 5
|
|
& gYtd . gsWins .~ 6
|
|
& gYtd . gsLosses .~ 7
|
|
& gYtd . gsTies .~ 8
|
|
& gLifetime . gsGames .~ 9
|
|
& gLifetime . gsMinsPlayed .~ 10
|
|
& gLifetime . gsGoalsAllowed .~ 11
|
|
& gLifetime . gsGoalsAgainst .~ 12
|
|
& gLifetime . gsWins .~ 13
|
|
& gLifetime . gsLosses .~ 14
|
|
& gLifetime . gsTies .~ 15
|
|
|
|
db :: Database
|
|
db = newDatabase
|
|
& dbPlayers .~ [player]
|
|
& dbGoalies .~ [goalie]
|
|
& dbGames .~ 1
|
|
|
|
playerJSON :: ByteString
|
|
playerJSON = [r|
|
|
{ "number": 1
|
|
, "name": "Joe"
|
|
, "position": "centre"
|
|
, "ytd":
|
|
{ "goals": 2
|
|
, "assists": 3
|
|
, "penalty_mins": 4
|
|
}
|
|
, "lifetime":
|
|
{ "goals": 5
|
|
, "assists": 6
|
|
, "penalty_mins": 7
|
|
}
|
|
}|]
|
|
|
|
goalieJSON :: ByteString
|
|
goalieJSON = [r|
|
|
{ "number": 1
|
|
, "name": "Joe"
|
|
, "ytd":
|
|
{ "games": 2
|
|
, "mins_played": 3
|
|
, "goals_allowed": 4
|
|
, "goals_against": 5
|
|
, "wins": 6
|
|
, "losses": 7
|
|
, "ties": 8
|
|
}
|
|
, "lifetime":
|
|
{ "games": 9
|
|
, "mins_played": 10
|
|
, "goals_allowed": 11
|
|
, "goals_against": 12
|
|
, "wins": 13
|
|
, "losses": 14
|
|
, "ties": 15
|
|
}
|
|
}|]
|
|
|
|
dbJSON :: ByteString
|
|
dbJSON = [r|
|
|
{ "players":
|
|
[ |] <> playerJSON <> [r| ]
|
|
, "goalies":
|
|
[ |] <> goalieJSON <> [r| ]
|
|
, "games": 1
|
|
}|]
|