Synnefo » snf-ganeti

from ganeti import ht

class TestLoadAndVerifyJson(unittest.TestCase):

  def testNoJson(self):

    self.assertRaises(errors.ParseError, serializer.LoadAndVerifyJson,

                      "", NotImplemented)

    self.assertRaises(errors.ParseError, serializer.LoadAndVerifyJson,

                      "}", NotImplemented)

  def testVerificationFails(self):

    self.assertRaises(errors.ParseError, serializer.LoadAndVerifyJson,

                      "{}", lambda _: False)

    verify_fn = ht.TListOf(ht.TNonEmptyString)

    try:

      serializer.LoadAndVerifyJson("{}", verify_fn)

    except errors.ParseError, err:

      self.assertTrue(str(err).endswith(str(verify_fn)))

    else:

      self.fail("Exception not raised")

  def testSuccess(self):

    self.assertEqual(serializer.LoadAndVerifyJson("{}", ht.TAny), {})

    self.assertEqual(serializer.LoadAndVerifyJson("\"Foo\"", ht.TAny), "Foo")