Synnefo » snf-ganeti

#!/usr/bin/python

#

# Copyright (C) 2011 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.

"""Script for testing ganeti.rapi.baserlib"""

import unittest

import itertools

from ganeti import errors

from ganeti import opcodes

from ganeti import ht

from ganeti import http

from ganeti import compat

from ganeti.rapi import baserlib

import testutils

class TestFillOpcode(unittest.TestCase):

  class OpTest(opcodes.OpCode):

    OP_PARAMS = [

      ("test", None, ht.TMaybeString, None),

      ]

  def test(self):

    for static in [None, {}]:

      op = baserlib.FillOpcode(self.OpTest, {}, static)

      self.assertTrue(isinstance(op, self.OpTest))

      self.assertTrue(op.test is None)

  def testStatic(self):

    op = baserlib.FillOpcode(self.OpTest, {}, {"test": "abc"})

    self.assertTrue(isinstance(op, self.OpTest))

    self.assertEqual(op.test, "abc")

    # Overwrite static parameter

    self.assertRaises(http.HttpBadRequest, baserlib.FillOpcode,

                      self.OpTest, {"test": 123}, {"test": "abc"})

  def testType(self):

    self.assertRaises(http.HttpBadRequest, baserlib.FillOpcode,

                      self.OpTest, {"test": [1, 2, 3]}, {})

  def testStaticType(self):

    self.assertRaises(http.HttpBadRequest, baserlib.FillOpcode,

                      self.OpTest, {}, {"test": [1, 2, 3]})

  def testUnicode(self):

    op = baserlib.FillOpcode(self.OpTest, {u"test": "abc"}, {})

    self.assertTrue(isinstance(op, self.OpTest))

    self.assertEqual(op.test, "abc")

    op = baserlib.FillOpcode(self.OpTest, {}, {u"test": "abc"})

    self.assertTrue(isinstance(op, self.OpTest))

    self.assertEqual(op.test, "abc")

  def testUnknownParameter(self):

    self.assertRaises(http.HttpBadRequest, baserlib.FillOpcode,

                      self.OpTest, {"othervalue": 123}, None)

  def testInvalidBody(self):

    self.assertRaises(http.HttpBadRequest, baserlib.FillOpcode,

                      self.OpTest, "", None)

    self.assertRaises(http.HttpBadRequest, baserlib.FillOpcode,

                      self.OpTest, range(10), None)

  def testRenameBothSpecified(self):

    self.assertRaises(http.HttpBadRequest, baserlib.FillOpcode,

                      self.OpTest, { "old": 123, "new": 999, }, None,

                      rename={ "old": "new", })

  def testRename(self):

    value = "Hello World"

    op = baserlib.FillOpcode(self.OpTest, { "data": value, }, None,

                             rename={ "data": "test", })

    self.assertEqual(op.test, value)

  def testRenameStatic(self):

    self.assertRaises(http.HttpBadRequest, baserlib.FillOpcode,

                      self.OpTest, { "data": 0, }, { "test": None, },

                      rename={ "data": "test", })

class TestOpcodeResource(unittest.TestCase):

  @staticmethod

  def _MakeClass(method, attrs):

    return type("Test%s" % method, (baserlib.OpcodeResource, ), attrs)

  @staticmethod

  def _GetMethodAttributes(method):

    attrs = ["%s_OPCODE" % method, "%s_RENAME" % method,

             "Get%sOpInput" % method.capitalize()]

    assert attrs == dict((opattrs[0], list(opattrs[1:]))

                         for opattrs in baserlib.OPCODE_ATTRS)[method]

    return attrs

  def test(self):

    for method in baserlib._SUPPORTED_METHODS:

      # Empty handler

      obj = self._MakeClass(method, {})(None, {}, None)

      for attr in itertools.chain(*baserlib.OPCODE_ATTRS):

        self.assertFalse(hasattr(obj, attr))

      # Direct handler function

      obj = self._MakeClass(method, {

        method: lambda _: None,

        })(None, {}, None)

      self.assertFalse(compat.all(hasattr(obj, attr)

                                  for i in baserlib._SUPPORTED_METHODS

                                  for attr in self._GetMethodAttributes(i)))

      # Let metaclass define handler function

      for opcls in [None, object()]:

        obj = self._MakeClass(method, {

          "%s_OPCODE" % method: opcls,

          })(None, {}, None)

        self.assertTrue(callable(getattr(obj, method)))

        self.assertEqual(getattr(obj, "%s_OPCODE" % method), opcls)

        self.assertFalse(hasattr(obj, "%s_RENAME" % method))

        self.assertFalse(compat.any(hasattr(obj, attr)

                                    for i in baserlib._SUPPORTED_METHODS

                                      if i != method

                                    for attr in self._GetMethodAttributes(i)))

  def testIllegalRename(self):

    class _TClass(baserlib.OpcodeResource):

      PUT_RENAME = None

      def PUT(self): pass

    self.assertRaises(AssertionError, _TClass, None, None, None)

  def testEmpty(self):

    class _Empty(baserlib.OpcodeResource):

      pass

    obj = _Empty(None, {}, None)

    for attr in itertools.chain(*baserlib.OPCODE_ATTRS):

      self.assertFalse(hasattr(obj, attr))

if __name__ == "__main__":

  testutils.GanetiTestProgram()