Synnefo » snf-ganeti

#!/usr/bin/python

#

# Copyright (C) 2010 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.qlang"""

import unittest

from ganeti import utils

from ganeti import errors

from ganeti import qlang

import testutils

class TestReadSimpleFilter(unittest.TestCase):

  def _Test(self, filter_, expected):

    self.assertEqual(qlang.ReadSimpleFilter("name", filter_), expected)

  def test(self):

    self._Test(None, [])

    self._Test(["|", ["=", "name", "xyz"]], ["xyz"])

    for i in [1, 3, 10, 25, 140]:

      self._Test(["|"] + [["=", "name", "node%s" % j] for j in range(i)],

                 ["node%s" % j for j in range(i)])

  def testErrors(self):

    for i in [123, True, False, "", "Hello World", "a==b",

              [], ["x"], ["x", "y", "z"], ["|"],

              ["|", ["="]], ["|", "x"], ["|", 123],

              ["|", ["=", "otherfield", "xyz"]],

              ["|", ["=", "name", "xyz"], "abc"],

              ["|", ["=", "name", "xyz", "too", "long"]],

              ["|", ["=", "name", []]],

              ["|", ["=", "name", 999]],

              ["|", ["=", "name", "abc"], ["=", "otherfield", "xyz"]]]:

      self.assertRaises(errors.ParameterError, qlang.ReadSimpleFilter,

                        "name", i)

class TestMakeSimpleFilter(unittest.TestCase):

  def _Test(self, field, names, expected, parse_exp=None):

    if parse_exp is None:

      parse_exp = names

    filter_ = qlang.MakeSimpleFilter(field, names)

    self.assertEqual(filter_, expected)

    self.assertEqual(qlang.ReadSimpleFilter(field, filter_), parse_exp)

  def test(self):

    self._Test("name", None, None, parse_exp=[])

    self._Test("name", [], None)

    self._Test("name", ["node1.example.com"],

               ["|", ["=", "name", "node1.example.com"]])

    self._Test("xyz", ["a", "b", "c"],

               ["|", ["=", "xyz", "a"], ["=", "xyz", "b"], ["=", "xyz", "c"]])

if __name__ == "__main__":

  testutils.GanetiTestProgram()