implemented editPlayerStateL

src/Mtlstats/Types.hs

@@ -50,6 +50,7 @@ module Mtlstats.Types (
   gameStateL,
   createPlayerStateL,
   createGoalieStateL,
+  editPlayerStateL,
   -- ** GameState Lenses
   gameYear,
   gameMonth,
@@ -584,6 +585,13 @@ createGoalieStateL = lens
     _                -> newCreateGoalieState)
   (\_ cgs -> CreateGoalie cgs)
 
+editPlayerStateL :: Lens' ProgMode EditPlayerState
+editPlayerStateL = lens
+  (\case
+    EditPlayer eps -> eps
+    _              -> newEditPlayerState)
+  (\_ eps -> EditPlayer eps)
+
 -- | Constructor for a 'ProgState'
 newProgState :: ProgState
 newProgState = ProgState

test/TypesSpec.hs

@@ -47,6 +47,7 @@ spec = describe "Mtlstats.Types" $ do
   gameStateLSpec
   createPlayerStateLSpec
   createGoalieStateLSpec
+  editPlayerStateLSpec
   teamScoreSpec
   otherScoreSpec
   homeTeamSpec
@@ -141,6 +142,24 @@ createGoalieStateLSpec = describe "createGoalieStateL" $
       & cgsNumber ?~ 2
       & cgsName   .~ "Bob"
 
+editPlayerStateLSpec :: Spec
+editPlayerStateLSpec = describe "editPlayerStateL" $
+  lensSpec editPlayerStateL
+  -- getters
+  [ ( "missing state", MainMenu,        newEditPlayerState )
+  , ( "withState",     EditPlayer eps1, eps1               )
+  ]
+  -- setters
+  [ ( "set state",    MainMenu,        eps1               )
+  , ( "change state", EditPlayer eps1, eps2               )
+  , ( "clear state",  EditPlayer eps1, newEditPlayerState )
+  ]
+  where
+    eps1 = newEditPlayerState
+      & epsSelectedPlayer ?~ 1
+    eps2 = newEditPlayerState
+      & epsSelectedPlayer ?~ 2
+
 teamScoreSpec :: Spec
 teamScoreSpec = describe "teamScore" $ do
   let
@@ -734,6 +753,12 @@ instance Comparable CreatePlayerState where
       it ("should be " ++ expected^.cpsPosition) $
         actual^.cpsPosition `shouldBe` expected^.cpsPosition
 
+instance Comparable EditPlayerState where
+  compareTest actual expected =
+    describe "epsSelectedPlayer" $
+      it ("should be " ++ show (expected^.epsSelectedPlayer)) $
+        actual^.epsSelectedPlayer `shouldBe` expected^.epsSelectedPlayer
+
 instance Comparable CreateGoalieState where
   compareTest actual expected = do