implemented getPositions

src/Mtlstats/Helpers/Position.hs

@@ -23,8 +23,12 @@ module Mtlstats.Helpers.Position
   ( posSearch
   , posSearchExact
   , posCallback
+  , getPositions
   ) where
 
+import qualified Data.Set as S
+import Lens.Micro ((^.))
+
 import Mtlstats.Types
 
 -- | Searches the 'Database' for all the positions used
@@ -57,3 +61,9 @@ posCallback
   -> Action ()
   -- ^ The action to perform
 posCallback = undefined
+
+-- | Extracts a list of positions from a 'Database'
+getPositions :: Database -> [String]
+getPositions = do
+  raw <- map (^.pPosition) . (^.dbPlayers)
+  return $ S.toList $ S.fromList raw

test/Helpers/PositionSpec.hs

@@ -0,0 +1,44 @@
+{-
+
+mtlstats
+Copyright (C) 1984, 1985, 2019, 2020 Rhéal Lamothe
+<rheal.lamothe@gmail.com>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or (at
+your option) any later version.
+
+This program is distributed in the hope that it will be useful, but
+WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+-}
+
+module Helpers.PositionSpec (spec) where
+
+import Lens.Micro ((&), (.~))
+import Test.Hspec (Spec, describe, it, shouldBe)
+
+import Mtlstats.Helpers.Position
+import Mtlstats.Types
+
+spec :: Spec
+spec = describe "Position"
+  getPositionsSpec
+
+getPositionsSpec :: Spec
+getPositionsSpec = describe "getPositions" $ let
+  db = newDatabase & dbPlayers .~
+    [ newPlayer 2 "Joe"  "foo"
+    , newPlayer 3 "Bob"  "bar"
+    , newPlayer 5 "Bill" "foo"
+    , newPlayer 8 "Ed"   "baz"
+    ]
+  expected = ["bar", "baz", "foo"]
+  in it ("should be " ++ show expected) $
+    getPositions db `shouldBe` expected

test/HelpersSpec.hs

@@ -25,8 +25,10 @@ import Test.Hspec (Spec, describe)
 
 import qualified Helpers.GoalieSpec as Goalie
 import qualified Helpers.PlayerSpec as Player
+import qualified Helpers.PositionSpec as Position
 
 spec :: Spec
 spec = describe "Helper" $ do
   Player.spec
   Goalie.spec
+  Position.spec