Synnefo » snf-ganeti

{-# LANGUAGE TemplateHaskell #-}

{-# OPTIONS_GHC -fno-warn-orphans #-}

{-| Unittests for ganeti-htools.

-}

{-

Copyright (C) 2009, 2010, 2011, 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.JSON (testJSON) where

import Control.Monad

import Data.List

import Test.QuickCheck

import qualified Text.JSON as J

import Test.Ganeti.TestHelper

import Test.Ganeti.TestCommon

import Test.Ganeti.Types ()

import qualified Ganeti.BasicTypes as BasicTypes

import qualified Ganeti.JSON as JSON

instance (Arbitrary a) => Arbitrary (JSON.MaybeForJSON a) where

  arbitrary = liftM JSON.MaybeForJSON arbitrary

instance Arbitrary JSON.TimeAsDoubleJSON where

  arbitrary = liftM JSON.TimeAsDoubleJSON arbitrary

prop_toArray :: [Int] -> Property

prop_toArray intarr =

  let arr = map J.showJSON intarr in

  case JSON.toArray (J.JSArray arr) of

    BasicTypes.Ok arr' -> arr ==? arr'

    BasicTypes.Bad err -> failTest $ "Failed to parse array: " ++ err

prop_toArrayFail :: Int -> String -> Bool -> Property

prop_toArrayFail i s b =

  -- poor man's instance Arbitrary JSValue

  forAll (elements [J.showJSON i, J.showJSON s, J.showJSON b]) $ \item ->

  case JSON.toArray item::BasicTypes.Result [J.JSValue] of

    BasicTypes.Bad _ -> passTest

    BasicTypes.Ok result -> failTest $ "Unexpected parse, got " ++ show result

arrayMaybeToJson :: (J.JSON a) => [Maybe a] -> String -> JSON.JSRecord

arrayMaybeToJson xs k = [(k, J.JSArray $ map sh xs)]

  where

    sh x = case x of

      Just v -> J.showJSON v

      Nothing -> J.JSNull

prop_arrayMaybeFromObj :: String -> [Maybe Int] -> String -> Property

prop_arrayMaybeFromObj t xs k =

  case JSON.tryArrayMaybeFromObj t (arrayMaybeToJson xs k) k of

    BasicTypes.Ok xs' -> xs' ==? xs

    BasicTypes.Bad e -> failTest $ "Parsing failing, got: " ++ show e

prop_arrayMaybeFromObjFail :: String -> String -> Property

prop_arrayMaybeFromObjFail t k =

  case JSON.tryArrayMaybeFromObj t [] k of

    BasicTypes.Ok r -> fail $

                       "Unexpected result, got: " ++ show (r::[Maybe Int])

    BasicTypes.Bad e -> conjoin [ Data.List.isInfixOf t e ==? True

                                , Data.List.isInfixOf k e ==? True

                                ]

prop_MaybeForJSON_serialisation :: JSON.MaybeForJSON String -> Property

prop_MaybeForJSON_serialisation = testSerialisation

prop_TimeAsDoubleJSON_serialisation :: JSON.TimeAsDoubleJSON -> Property

prop_TimeAsDoubleJSON_serialisation = testSerialisation

testSuite "JSON"

          [ 'prop_toArray

          , 'prop_toArrayFail

          , 'prop_arrayMaybeFromObj

          , 'prop_arrayMaybeFromObjFail

          , 'prop_MaybeForJSON_serialisation

          , 'prop_TimeAsDoubleJSON_serialisation

          ]