code.delx.au - offlineimap/blob - testsrc/TestParserPrim.hs

   1 {-
   2 Copyright (C) 2002-2008 John Goerzen <jgoerzen@complete.org>
   3
   4 This program is free software; you can redistribute it and/or modify
   5 it under the terms of the GNU General Public License as published by
   6 the Free Software Foundation; either version 2 of the License, or
   7 (at your option) any later version.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program; if not, write to the Free Software
  16 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17 -}
  18 module TestParserPrim where
  19 import Data.List
  20 import qualified Test.HUnit as HU
  21 import Test.HUnit.Utils
  22 import Data.Word
  23 import Test.QuickCheck
  24 import TestInfrastructure
  25
  26 import Network.IMAP.Parser.Prim
  27 import Network.IMAP.Types
  28
  29 import TestInfrastructure
  30 import Text.ParserCombinators.Parsec
  31 import Text.ParserCombinators.Parsec.Error
  32
  33 {- | Test a parser, forcing it to apply to all input. -}
  34 p parser input =
  35     case parse parseTest "(none)" input of
  36       Left _ -> Nothing
  37       Right y -> Just y
  38     where parseTest = do r <- parser
  39                          eof
  40                          return r
  41
  42 prop_quoted :: String -> Result
  43 prop_quoted s =
  44     p quoted (gen_quoted s) @?= Just s
  45
  46 gen_quoted :: String -> String
  47 gen_quoted s = '"' : concatMap quoteChar s ++ "\""
  48     where quoteChar '\\' = "\\\\"
  49           quoteChar '"' = "\\\""
  50           quoteChar x = [x]
  51
  52 prop_literal :: String -> Result
  53 prop_literal s =
  54     p literal (gen_literal s) @?= Just s
  55
  56 gen_literal :: String -> String
  57 gen_literal s =
  58     "{" ++ show (length s) ++ "}\r\n" ++ s
  59
  60 gen_string3501 :: String -> Bool -> String
  61 gen_string3501 s True = gen_quoted s
  62 gen_string3501 s False = gen_literal s
  63
  64 prop_string3501 :: String -> Bool -> Result
  65 prop_string3501 s useQuoted = p string3501 (gen_string3501 s useQuoted) @?= Just s
  66
  67 prop_atom :: String -> Result
  68 prop_atom s =
  69     p atom s @?= if isvalid
  70                     then Just s
  71                     else Nothing
  72     where isvalid = not (null s) && all (`notElem` atomSpecials) s
  73
  74 prop_astring_basic :: String -> Result
  75 prop_astring_basic s =
  76     p astring s @?= if isValidAtom s
  77                        then Just s
  78                        else Nothing
  79
  80 isValidAtom s = not (null s) && all isValidChar s
  81     where isValidChar c =
  82               c `notElem` atomSpecials ||
  83               c `elem` respSpecials
  84
  85 prop_astring :: String -> Bool -> Result
  86 prop_astring s useQuoted =
  87     if isValidAtom s
  88        then p astring s @=? Just s
  89        else p astring (gen_string3501 s useQuoted) @=? Just s
  90
  91 allt = [q "quoted" prop_quoted,
  92         q "literal" prop_literal,
  93         q "string3501" prop_string3501,
  94         q "atom" prop_atom,
  95         q "astring basic" prop_astring_basic,
  96         q "prop_astring" prop_astring
  97        ]