From 43b3b5c1fc184224fffc34f124f190794d8599ea Mon Sep 17 00:00:00 2001
From: Michele Tartara <mtartara@google.com>
Date: Tue, 13 Nov 2012 12:31:13 +0100
Subject: [PATCH] Added attoparsec unit test for Unicode parsing

Attoparsec is known to have had issues with parsing non-ASCII strings.
This test makes sure that parsing of Unicode characters works fine.

Signed-off-by: Michele Tartara <mtartara@google.com>
[iustin: small doc string fixes]
Reviewed-by: Iustin Pop <iustin@google.com>
---
 Makefile.am                     |    1 +
 htest/Test/Ganeti/Attoparsec.hs |   66 +++++++++++++++++++++++++++++++++++++++
 htest/test.hs                   |    2 ++
 3 files changed, 69 insertions(+)
 create mode 100644 htest/Test/Ganeti/Attoparsec.hs

diff --git a/Makefile.am b/Makefile.am
index ea38407..d6a0e28 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -473,6 +473,7 @@ HS_LIB_SRCS = \
 	htools/Ganeti/Utils.hs
 
 HS_TEST_SRCS = \
+	htest/Test/Ganeti/Attoparsec.hs \
 	htest/Test/Ganeti/BasicTypes.hs \
 	htest/Test/Ganeti/Common.hs \
 	htest/Test/Ganeti/Confd/Utils.hs \
diff --git a/htest/Test/Ganeti/Attoparsec.hs b/htest/Test/Ganeti/Attoparsec.hs
new file mode 100644
index 0000000..20aab6e
--- /dev/null
+++ b/htest/Test/Ganeti/Attoparsec.hs
@@ -0,0 +1,66 @@
+{-# LANGUAGE TemplateHaskell #-}
+
+{-| Unittests for Attoparsec support for unicode -}
+
+{-
+
+Copyright (C) 2012 Google Inc.
+
+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 2 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, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
+02110-1301, USA.
+
+-}
+
+module Test.Ganeti.Attoparsec (testAttoparsec) where
+
+import Test.QuickCheck
+
+import Test.Ganeti.TestHelper
+import Test.Ganeti.TestCommon
+
+import qualified Data.Attoparsec.Text as A
+import Data.Attoparsec.Text (Parser)
+import Data.Text (pack, unpack)
+
+-- | Unicode test string, first part.
+part1 :: String
+part1 = "äßĉ"
+
+-- | Unicode test string, second part.
+part2 :: String
+part2 = "ðèق"
+
+-- | Simple parser able to split a string in two parts, name and
+-- value, separated by a '=' sign.
+simpleParser :: Parser (String, String)
+simpleParser = do
+  n <- A.takeTill (\c -> A.isHorizontalSpace c || c == '=')
+  A.skipWhile A.isHorizontalSpace
+  _ <- A.char '='
+  A.skipWhile A.isHorizontalSpace
+  v <- A.takeTill A.isEndOfLine
+  return (unpack n, unpack v)
+
+-- | Tests whether a Unicode string is still Unicode after being
+-- parsed.
+prop_parserSupportsUnicode :: Property
+prop_parserSupportsUnicode = case A.parseOnly simpleParser text of
+    Right (name, value) -> (name ==? part1) .&&. (value ==? part2)
+    Left msg -> failTest $ "Failed to parse: " ++ msg
+  where text = Data.Text.pack $ part1 ++ "  = \t" ++ part2
+
+testSuite "Attoparsec"
+          [ 'prop_parserSupportsUnicode
+          ]
diff --git a/htest/test.hs b/htest/test.hs
index 5180878..b41fb99 100644
--- a/htest/test.hs
+++ b/htest/test.hs
@@ -30,6 +30,7 @@ import Test.Framework
 import System.Environment (getArgs)
 
 import Test.Ganeti.TestImports ()
+import Test.Ganeti.Attoparsec
 import Test.Ganeti.BasicTypes
 import Test.Ganeti.Common
 import Test.Ganeti.Confd.Utils
@@ -74,6 +75,7 @@ defOpts = TestOptions
 allTests :: [Test]
 allTests =
   [ testBasicTypes
+  , testAttoparsec
   , testCommon
   , testConfd_Utils
   , testDaemon
-- 
1.7.10.4