From cb5aa63469821a5bfcfbf918a0e3bc1d72abc81f Mon Sep 17 00:00:00 2001
From: Jonathan Lamothe <jlamothe1980@gmail.com>
Date: Thu, 14 Nov 2019 01:32:20 -0500
Subject: [PATCH] implemented editGoalieYtdGoals

---
 src/Mtlstats/Actions/EditGoalie.hs |  2 +-
 test/Actions/EditGoalieSpec.hs     | 32 ++++++++++++++++++++++++++++++
 2 files changed, 33 insertions(+), 1 deletion(-)

diff --git a/src/Mtlstats/Actions/EditGoalie.hs b/src/Mtlstats/Actions/EditGoalie.hs
index 0a7e101..ec4a1de 100644
--- a/src/Mtlstats/Actions/EditGoalie.hs
+++ b/src/Mtlstats/Actions/EditGoalie.hs
@@ -72,7 +72,7 @@ editGoalieYtdGoals
   -- ^ The number of goals
   -> ProgState
   -> ProgState
-editGoalieYtdGoals = undefined
+editGoalieYtdGoals goals = editGoalie (gYtd.gsGoalsAllowed .~ goals) EGYtd
 
 editGoalie :: (Goalie -> Goalie) -> EditGoalieMode -> ProgState -> ProgState
 editGoalie f mode s = fromMaybe s $ do
diff --git a/test/Actions/EditGoalieSpec.hs b/test/Actions/EditGoalieSpec.hs
index 2543395..d407466 100644
--- a/test/Actions/EditGoalieSpec.hs
+++ b/test/Actions/EditGoalieSpec.hs
@@ -35,6 +35,7 @@ spec = describe "EditGoalie" $ do
   editGoalieNameSpec
   editGoalieYtdGamesSpec
   editGoalieYtdMinsSpec
+  editGoalieYtdGoalsSpec
 
 editGoalieNumberSpec :: Spec
 editGoalieNumberSpec = describe "editGoalieNumber" $ editTest
@@ -160,6 +161,37 @@ editGoalieYtdMinsSpec = describe "editGoalieYtdMins" $ editTest
     )
   ]
 
+editGoalieYtdGoalsSpec :: Spec
+editGoalieYtdGoalsSpec = describe "editGoalieYtdGoals" $ editTest
+  (editGoalieYtdGoals 1)
+  EGYtdGoals
+  (\(num, name, goals) -> newGoalie num name & gYtd.gsGoalsAllowed .~ goals)
+  [ ( "set Joe"
+    , Just 0
+    , ( 2, "Joe", 1 )
+    , ( 3, "Bob", 0 )
+    , EGYtd
+    )
+  , ( "set Bob"
+    , Just 1
+    , ( 2, "Joe", 0 )
+    , ( 3, "Bob", 1 )
+    , EGYtd
+    )
+  , ( "out of bounds"
+    , Just 2
+    , ( 2, "Joe", 0 )
+    , ( 3, "Bob", 0 )
+    , EGYtdGoals
+    )
+  , ( "no goalie selected"
+    , Nothing
+    , ( 2, "Joe", 0 )
+    , ( 3, "Bob", 0 )
+    , EGYtdGoals
+    )
+  ]
+
 editTest
   :: (ProgState -> ProgState)
   -> EditGoalieMode