passman/test/Spec/PWSearch.hs

{-

passman
Copyright (C) Jonathan Lamothe
<jonathan@jlamothe.net>

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.PWSearch (tests) where

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

import Password

tests :: Test
tests = TestLabel "pwSearch" $ TestList $ map test'
  [ ( "no results",   "quux", []                    )
  , ( "some results", "A",    ["bar", "baz"]        )
  , ( "all results",  "",     ["foo", "bar", "baz"] )
  ]

test' :: (String, String, [String]) -> Test
test' (label, str, expect) = TestLabel label $ TestList $
  TestLabel "length" (length result ~?= length expect) :
  map (\x -> TestLabel ("has " ++ x) $ TestCase $
    assertBool "not found" $ elem x expect) result
  where result = pwSearch str db

db :: M.Map String PWData
db = M.fromList
  [ ( "foo", foo )
  , ( "bar", bar )
  , ( "baz", baz )
  ]

foo :: PWData
g'  :: StdGen
(foo, g') = newPWData g

bar :: PWData
g'' :: StdGen
(bar, g'') = newPWData g'

baz :: PWData
(baz, _) = newPWData g''

g :: StdGen
g = mkStdGen 1

--jl