1. import unittest

  2. 

  3. from django.template.smartif import IfParser

  4. 

  5. 

  6. class SmartIfTests(unittest.TestCase):

  7.     def assertCalcEqual(self, expected, tokens):

  8.         self.assertEqual(expected, IfParser(tokens).parse().eval({}))

  9. 

  10.     # We only test things here that are difficult to test elsewhere

  11.     # Many other tests are found in the main tests for builtin template tags

  12.     # Test parsing via the printed parse tree

  13.     def test_not(self):

  14.         var = IfParser(["not", False]).parse()

  15.         self.assertEqual("(not (literal False))", repr(var))

  16.         self.assertTrue(var.eval({}))

  17. 

  18.         self.assertFalse(IfParser(["not", True]).parse().eval({}))

  19. 

  20.     def test_or(self):

  21.         var = IfParser([True, "or", False]).parse()

  22.         self.assertEqual("(or (literal True) (literal False))", repr(var))

  23.         self.assertTrue(var.eval({}))

  24. 

  25.     def test_in(self):

  26.         list_ = [1, 2, 3]

  27.         self.assertCalcEqual(True, [1, "in", list_])

  28.         self.assertCalcEqual(False, [1, "in", None])

  29.         self.assertCalcEqual(False, [None, "in", list_])

  30. 

  31.     def test_not_in(self):

  32.         list_ = [1, 2, 3]

  33.         self.assertCalcEqual(False, [1, "not", "in", list_])

  34.         self.assertCalcEqual(True, [4, "not", "in", list_])

  35.         self.assertCalcEqual(False, [1, "not", "in", None])

  36.         self.assertCalcEqual(True, [None, "not", "in", list_])

  37. 

  38.     def test_precedence(self):

  39.         # (False and False) or True == True   <- we want this one, like Python

  40.         # False and (False or True) == False

  41.         self.assertCalcEqual(True, [False, "and", False, "or", True])

  42. 

  43.         # True or (False and False) == True   <- we want this one, like Python

  44.         # (True or False) and False == False

  45.         self.assertCalcEqual(True, [True, "or", False, "and", False])

  46. 

  47.         # (1 or 1) == 2  -> False

  48.         # 1 or (1 == 2)  -> True   <- we want this one

  49.         self.assertCalcEqual(True, [1, "or", 1, "==", 2])

  50. 

  51.         self.assertCalcEqual(True, [True, "==", True, "or", True, "==", False])

  52. 

  53.         self.assertEqual(

  54.             "(or (and (== (literal 1) (literal 2)) (literal 3)) (literal 4))",

  55.             repr(IfParser([1, "==", 2, "and", 3, "or", 4]).parse()),

  56.         )
About Sign Up Leave Feedback Contact