Synnefo » snf-ganeti

	test/data/xen-xm-uptime-4.0.1.txt \

  , UptimeInfo(..)

-- Data type representing the information that can be obtained from @xm uptime@

data UptimeInfo = UptimeInfo

  { uInfoName   :: String

  , uInfoID     :: Int

  , uInfoUptime :: String

  } deriving (Eq, Show)

  , xmUptimeParser

  , uptimeLineParser

-- | A parser for parsing the output of the @xm uptime@ command.

xmUptimeParser :: Parser (Map.Map Int UptimeInfo)

xmUptimeParser = do

  _ <- headerParser

  uptimes <- uptimeLineParser `AC.manyTill` A.endOfInput

  return $ Map.fromList [(uInfoID u, u) | u <- uptimes]

    where headerParser = A.string "Name" <* A.skipSpace <* A.string "ID"

            <* A.skipSpace <* A.string "Uptime" <* A.skipSpace

-- | A helper for parsing a single line of the @xm uptime@ output.

uptimeLineParser :: Parser UptimeInfo

uptimeLineParser = do

  name <- A.takeTill isSpace <* A.skipSpace

  idNum <- A.decimal <* A.skipSpace

  uptime <- A.takeTill (`elem` "\n\r") <* A.skipSpace

  return . UptimeInfo (unpack name) idNum $ unpack uptime

Name                                ID Uptime

Domain-0                             0 98 days,  2:27:44

instance1.example.com               119 15 days, 20:57:07

import Text.Printf

-- | Generates an arbitrary @xm uptime@ output line.

instance Arbitrary UptimeInfo where

  arbitrary = do

    name <- genFQDN

    NonNegative idNum <- arbitrary :: Gen (NonNegative Int)

    NonNegative days <- arbitrary :: Gen (NonNegative Int)

    hours <- choose (0, 23) :: Gen Int

    mins <- choose (0, 59) :: Gen Int

    secs <- choose (0, 59) :: Gen Int

    let uptime :: String

        uptime =

          if days /= 0

            then printf "%d days, %d:%d:%d" days hours mins secs

            else printf "%d:%d:%d" hours mins secs

    return $ UptimeInfo name idNum uptime

  fileContent <- readTestData fileName

  case A.parseOnly xmListParser $ pack fileContent of

    Left msg -> assertFailure $ "Parsing failed: " ++ msg

    Right obtained -> assertEqual fileName expectedContent obtained

-- | Function for testing whether a @xm uptime@ output (stored in a file)

-- is parsed correctly.

testUptimeInfo :: String -> Map.Map Int UptimeInfo -> Assertion

testUptimeInfo fileName expectedContent = do

  fileContent <- readTestData fileName

  case A.parseOnly xmUptimeParser $ pack fileContent of

    Left msg -> assertFailure $ "Parsing failed: " ++ msg

    Right obtained -> assertEqual fileName expectedContent obtained

-- | Function to serialize UptimeInfos in such a way that they can be rebuilt

-- againg by the uptimeLineParser.

serializeUptime :: UptimeInfo -> String

serializeUptime (UptimeInfo name idNum uptime) =

  printf "%s\t%d\t%s" name idNum uptime

-- | Test whether a randomly generated UptimeInfo text line can be parsed.

prop_uptimeInfo :: UptimeInfo -> Property

prop_uptimeInfo uInfo =

  case A.parseOnly uptimeLineParser . pack . serializeUptime $ uInfo of

    Left msg -> fail $ "Parsing failed: " ++ msg

    Right obtained -> obtained ==? uInfo

-- | Test a Xen 4.0.1 @xm uptime@ output.

case_xen401uptime :: Assertion

case_xen401uptime = testUptimeInfo "xen-xm-uptime-4.0.1.txt" $

  Map.fromList

    [ (0, UptimeInfo "Domain-0" 0 "98 days,  2:27:44")

    , (119, UptimeInfo "instance1.example.com" 119 "15 days, 20:57:07")

    ]

          , 'prop_uptimeInfo

          , 'case_xen401uptime