passman/test/Spec/PWRemoveService.hs

{-

passman
Copyright (C) 2018 Jonathan Lamothe
<jlamothe1980@gmail.com>

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser 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
Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public
License along with this program.  If not, see
<https://www.gnu.org/licenses/>.

-}

module Spec.PWRemoveService (tests) where

import qualified Data.Map.Lazy as M
import System.Random (mkStdGen)
import Test.HUnit (Test (..), assertBool, (~?=))

import Password

tests = TestLabel "pwRemoveService" $ TestList
  [ emptyDB
  , existingService
  , missingService
  ]

emptyDB = TestLabel "empty database" $
  pwRemoveService "foo" newPWDatabase ~?= newPWDatabase

existingService = TestLabel "existing service" $
  test' "foo" ["bar", "baz"]

missingService = TestLabel "missing service" $
  test' "quux" ["foo", "bar", "baz"]

test' serv keys = let db' = pwRemoveService serv db in
  TestList $
  TestLabel "key count" (length keys ~?= length (M.keys db')) :
  map
  (\x -> TestLabel x $ TestCase $ assertBool "service missing" $ pwHasService x db')
  keys

db = M.fromList
  [ ( "foo", foo )
  , ( "bar", bar )
  , ( "baz", baz )
  ]

(foo, g') = newPWData g

(bar, g'') = newPWData g'

(baz, _) = newPWData g''

g = mkStdGen 1

--jl