| import unittest |
| |
| from cStringIO import StringIO |
| |
| from ..backends import conditional |
| from ..node import BinaryExpressionNode, BinaryOperatorNode, VariableNode, NumberNode |
| |
| |
| class TestConditional(unittest.TestCase): |
| def compile(self, input_text): |
| return conditional.compile(input_text) |
| |
| def test_get_0(self): |
| data = """ |
| key: value |
| |
| [Heading 1] |
| other_key: |
| if a == 1: value_1 |
| if a == 2: value_2 |
| value_3 |
| """ |
| |
| manifest = self.compile(data) |
| |
| self.assertEquals(manifest.get("key"), "value") |
| children = list(item for item in manifest.iterchildren()) |
| self.assertEquals(len(children), 1) |
| section = children[0] |
| self.assertEquals(section.name, "Heading 1") |
| |
| self.assertEquals(section.get("other_key", {"a": 1}), "value_1") |
| self.assertEquals(section.get("other_key", {"a": 2}), "value_2") |
| self.assertEquals(section.get("other_key", {"a": 7}), "value_3") |
| self.assertEquals(section.get("key"), "value") |
| |
| def test_get_1(self): |
| data = """ |
| key: value |
| |
| [Heading 1] |
| other_key: |
| if a == "1": value_1 |
| if a == 2: value_2 |
| value_3 |
| """ |
| |
| manifest = self.compile(data) |
| |
| children = list(item for item in manifest.iterchildren()) |
| section = children[0] |
| |
| self.assertEquals(section.get("other_key", {"a": "1"}), "value_1") |
| self.assertEquals(section.get("other_key", {"a": 1}), "value_3") |
| |
| def test_get_2(self): |
| data = """ |
| key: |
| if a[1] == "b": value_1 |
| if a[1] == 2: value_2 |
| value_3 |
| """ |
| |
| manifest = self.compile(data) |
| |
| self.assertEquals(manifest.get("key", {"a": "ab"}), "value_1") |
| self.assertEquals(manifest.get("key", {"a": [1, 2]}), "value_2") |
| |
| def test_get_3(self): |
| data = """ |
| key: |
| if a[1] == "ab"[1]: value_1 |
| if a[1] == 2: value_2 |
| value_3 |
| """ |
| |
| manifest = self.compile(data) |
| |
| self.assertEquals(manifest.get("key", {"a": "ab"}), "value_1") |
| self.assertEquals(manifest.get("key", {"a": [1, 2]}), "value_2") |
| |
| def test_set_0(self): |
| data = """ |
| key: |
| if a == "a": value_1 |
| if a == "b": value_2 |
| value_3 |
| """ |
| manifest = self.compile(data) |
| |
| manifest.set("new_key", "value_new") |
| |
| self.assertEquals(manifest.get("new_key"), "value_new") |
| |
| def test_set_1(self): |
| data = """ |
| key: |
| if a == "a": value_1 |
| if a == "b": value_2 |
| value_3 |
| """ |
| |
| manifest = self.compile(data) |
| |
| manifest.set("key", "value_new") |
| |
| self.assertEquals(manifest.get("key"), "value_new") |
| self.assertEquals(manifest.get("key", {"a": "a"}), "value_1") |
| |
| def test_set_2(self): |
| data = """ |
| key: |
| if a == "a": value_1 |
| if a == "b": value_2 |
| value_3 |
| """ |
| |
| manifest = self.compile(data) |
| |
| expr = BinaryExpressionNode(BinaryOperatorNode("=="), |
| VariableNode("a"), |
| NumberNode("1")) |
| |
| manifest.set("key", "value_new", expr) |
| |
| self.assertEquals(manifest.get("key", {"a": 1}), "value_new") |
| self.assertEquals(manifest.get("key", {"a": "a"}), "value_1") |
| |
| def test_api_0(self): |
| data = """ |
| key: |
| if a == 1.5: value_1 |
| value_2 |
| key_1: other_value |
| """ |
| manifest = self.compile(data) |
| |
| self.assertFalse(manifest.is_empty) |
| self.assertEquals(manifest.root, manifest) |
| self.assertTrue(manifest.has_key("key_1")) |
| self.assertFalse(manifest.has_key("key_2")) |
| |
| self.assertEquals(set(manifest.iterkeys()), set(["key", "key_1"])) |
