cinder/cinder/tests/test_evaluator.py

# Copyright (c) 2014 Hewlett-Packard Development Company, L.P.
# All Rights Reserved.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

from cinder import exception
from cinder.scheduler.evaluator.evaluator import evaluate
from cinder import test


class EvaluatorTestCase(test.TestCase):
    def test_simple_integer(self):
        self.assertEqual(2, evaluate("1+1"))
        self.assertEqual(9, evaluate("2+3+4"))
        self.assertEqual(23, evaluate("11+12"))
        self.assertEqual(30, evaluate("5*6"))
        self.assertEqual(2, evaluate("22/11"))
        self.assertEqual(38, evaluate("109-71"))
        self.assertEqual(493, evaluate("872 - 453 + 44 / 22 * 4 + 66"))

    def test_simple_float(self):
        self.assertEqual(2.0, evaluate("1.0 + 1.0"))
        self.assertEqual(2.5, evaluate("1.5 + 1.0"))
        self.assertEqual(3.0, evaluate("1.5 * 2.0"))

    def test_int_float_mix(self):
        self.assertEqual(2.5, evaluate("1.5 + 1"))
        self.assertEqual(4.25, evaluate("8.5 / 2"))
        self.assertEqual(5.25, evaluate("10/4+0.75    + 2"))

    def test_negative_numbers(self):
        self.assertEqual(-2, evaluate("-2"))
        self.assertEqual(-1, evaluate("-2+1"))
        self.assertEqual(3, evaluate("5+-2"))

    def test_exponent(self):
        self.assertEqual(8, evaluate("2^3"))
        self.assertEqual(-8, evaluate("-2 ^ 3"))
        self.assertEqual(15.625, evaluate("2.5 ^ 3"))
        self.assertEqual(8, evaluate("4 ^ 1.5"))

    def test_function(self):
        self.assertEqual(5, evaluate("abs(-5)"))
        self.assertEqual(2, evaluate("abs(2)"))
        self.assertEqual(1, evaluate("min(1, 100)"))
        self.assertEqual(100, evaluate("max(1, 100)"))

    def test_parentheses(self):
        self.assertEqual(1, evaluate("(1)"))
        self.assertEqual(-1, evaluate("(-1)"))
        self.assertEqual(2, evaluate("(1+1)"))
        self.assertEqual(15, evaluate("(1+2) * 5"))
        self.assertEqual(3, evaluate("(1+2)*(3-1)/((1+(2-1)))"))
        self.assertEqual(-8.0, evaluate("((1.0 / 0.5) * (2)) *(-2)"))

    def test_comparisons(self):
        self.assertEqual(True, evaluate("1 < 2"))
        self.assertEqual(True, evaluate("2 > 1"))
        self.assertEqual(True, evaluate("2 != 1"))
        self.assertEqual(False, evaluate("1 > 2"))
        self.assertEqual(False, evaluate("2 < 1"))
        self.assertEqual(False, evaluate("2 == 1"))
        self.assertEqual(True, evaluate("(1 == 1) == !(1 == 2)"))

    def test_logic_ops(self):
        self.assertEqual(True, evaluate("(1 == 1) AND (2 == 2)"))
        self.assertEqual(True, evaluate("(1 == 1) and (2 == 2)"))
        self.assertEqual(True, evaluate("(1 == 1) && (2 == 2)"))
        self.assertEqual(False, evaluate("(1 == 1) && (5 == 2)"))

        self.assertEqual(True, evaluate("(1 == 1) OR (5 == 2)"))
        self.assertEqual(True, evaluate("(1 == 1) or (5 == 2)"))
        self.assertEqual(True, evaluate("(1 == 1) || (5 == 2)"))
        self.assertEqual(False, evaluate("(5 == 1) || (5 == 2)"))

        self.assertEqual(False, evaluate("(1 == 1) AND NOT (2 == 2)"))
        self.assertEqual(False, evaluate("(1 == 1) AND not (2 == 2)"))
        self.assertEqual(False, evaluate("(1 == 1) AND !(2 == 2)"))
        self.assertEqual(True, evaluate("(1 == 1) AND NOT (5 == 2)"))
        self.assertEqual(True,
                         evaluate("(1 == 1) OR NOT (2 == 2) AND (5 == 5)"))

    def test_ternary_conditional(self):
        self.assertEqual(5, evaluate("(1 < 2) ? 5 : 10"))
        self.assertEqual(10, evaluate("(1 > 2) ? 5 : 10"))

    def test_variables_dict(self):
        stats = {'iops': 1000, 'usage': 0.65, 'count': 503, 'free_space': 407}
        request = {'iops': 500, 'size': 4}
        self.assertEqual(1500, evaluate("stats.iops + request.iops",
                                        stats=stats, request=request))

    def test_missing_var(self):
        stats = {'iops': 1000, 'usage': 0.65, 'count': 503, 'free_space': 407}
        request = {'iops': 500, 'size': 4}
        self.assertRaises(exception.EvaluatorParseException,
                          evaluate,
                          "foo.bob + 5",
                          stats=stats, request=request)
        self.assertRaises(exception.EvaluatorParseException,
                          evaluate,
                          "stats.bob + 5",
                          stats=stats, request=request)
        self.assertRaises(exception.EvaluatorParseException,
                          evaluate,
                          "fake.var + 1",
                          stats=stats, request=request, fake=None)

    def test_bad_expression(self):
        self.assertRaises(exception.EvaluatorParseException,
                          evaluate,
                          "1/*1")

    def test_nonnumber_comparison(self):
        nonnumber = {'test': 'foo'}
        request = {'test': 'bar'}
        self.assertRaises(
            exception.EvaluatorParseException,
            evaluate,
            "nonnumber.test != request.test",
            nonnumber=nonnumber, request=request)

    def test_div_zero(self):
        self.assertRaises(exception.EvaluatorParseException,
                          evaluate,
                          "7 / 0")