Move common libs to libs/common

libs/common/rebulk/test/test_rules.py

@@ -0,0 +1,197 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# pylint: disable=no-self-use, pointless-statement, missing-docstring, invalid-name, no-member, len-as-condition
+import pytest
+from rebulk.test.default_rules_module import RuleRemove0, RuleAppend0, RuleRename0, RuleAppend1, RuleRemove1, \
+    RuleRename1, RuleAppend2, RuleRename2, RuleAppend3, RuleRename3, RuleAppendTags0, RuleRemoveTags0, \
+    RuleAppendTags1, RuleRemoveTags1
+
+from ..rules import Rules
+from ..match import Matches, Match
+
+from .rules_module import Rule1, Rule2, Rule3, Rule0, Rule1Disabled
+from . import rules_module as rm
+
+
+def test_rule_priority():
+    matches = Matches([Match(1, 2)])
+
+    rules = Rules(Rule1, Rule2())
+
+    rules.execute_all_rules(matches, {})
+    assert len(matches) == 0
+    matches = Matches([Match(1, 2)])
+
+    rules = Rules(Rule1(), Rule0)
+
+    rules.execute_all_rules(matches, {})
+    assert len(matches) == 1
+    assert matches[0] == Match(3, 4)
+
+
+def test_rules_duplicates():
+    matches = Matches([Match(1, 2)])
+
+    rules = Rules(Rule1, Rule1)
+
+    with pytest.raises(ValueError):
+        rules.execute_all_rules(matches, {})
+
+
+def test_rule_disabled():
+    matches = Matches([Match(1, 2)])
+
+    rules = Rules(Rule1Disabled(), Rule2())
+
+    rules.execute_all_rules(matches, {})
+    assert len(matches) == 2
+    assert matches[0] == Match(1, 2)
+    assert matches[1] == Match(3, 4)
+
+
+def test_rule_when():
+    matches = Matches([Match(1, 2)])
+
+    rules = Rules(Rule3())
+
+    rules.execute_all_rules(matches, {'when': False})
+    assert len(matches) == 1
+    assert matches[0] == Match(1, 2)
+
+    matches = Matches([Match(1, 2)])
+
+    rules.execute_all_rules(matches, {'when': True})
+    assert len(matches) == 2
+    assert matches[0] == Match(1, 2)
+    assert matches[1] == Match(3, 4)
+
+
+class TestDefaultRules(object):
+    def test_remove(self):
+        rules = Rules(RuleRemove0)
+
+        matches = Matches([Match(1, 2)])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches) == 0
+
+        rules = Rules(RuleRemove1)
+
+        matches = Matches([Match(1, 2)])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches) == 0
+
+    def test_append(self):
+        rules = Rules(RuleAppend0)
+
+        matches = Matches([Match(1, 2)])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches) == 2
+
+        rules = Rules(RuleAppend1)
+
+        matches = Matches([Match(1, 2)])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches) == 2
+
+        rules = Rules(RuleAppend2)
+
+        matches = Matches([Match(1, 2)])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches) == 2
+        assert len(matches.named('renamed')) == 1
+
+        rules = Rules(RuleAppend3)
+
+        matches = Matches([Match(1, 2)])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches) == 2
+        assert len(matches.named('renamed')) == 1
+
+    def test_rename(self):
+        rules = Rules(RuleRename0)
+
+        matches = Matches([Match(1, 2, name='original')])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches.named('original')) == 1
+        assert len(matches.named('renamed')) == 0
+
+        rules = Rules(RuleRename1)
+
+        matches = Matches([Match(5, 10, name='original')])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches.named('original')) == 0
+        assert len(matches.named('renamed')) == 1
+
+        rules = Rules(RuleRename2)
+
+        matches = Matches([Match(5, 10, name='original')])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches.named('original')) == 0
+        assert len(matches.named('renamed')) == 1
+
+        rules = Rules(RuleRename3)
+
+        matches = Matches([Match(5, 10, name='original')])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches.named('original')) == 0
+        assert len(matches.named('renamed')) == 1
+
+    def test_append_tags(self):
+        rules = Rules(RuleAppendTags0)
+
+        matches = Matches([Match(1, 2, name='tags', tags=['other'])])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches.named('tags')) == 1
+        assert matches.named('tags', index=0).tags == ['other', 'new-tag']
+
+        rules = Rules(RuleAppendTags1)
+
+        matches = Matches([Match(1, 2, name='tags', tags=['other'])])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches.named('tags')) == 1
+        assert matches.named('tags', index=0).tags == ['other', 'new-tag']
+
+    def test_remove_tags(self):
+        rules = Rules(RuleRemoveTags0)
+
+        matches = Matches([Match(1, 2, name='tags', tags=['other', 'new-tag'])])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches.named('tags')) == 1
+        assert matches.named('tags', index=0).tags == ['other']
+
+        rules = Rules(RuleRemoveTags1)
+
+        matches = Matches([Match(1, 2, name='tags', tags=['other', 'new-tag'])])
+        rules.execute_all_rules(matches, {})
+
+        assert len(matches.named('tags')) == 1
+        assert matches.named('tags', index=0).tags == ['other']
+
+
+def test_rule_module():
+    rules = Rules(rm)
+
+    matches = Matches([Match(1, 2)])
+    rules.execute_all_rules(matches, {})
+
+    assert len(matches) == 1
+
+
+def test_rule_repr():
+    assert str(Rule0()) == "<Rule0>"
+    assert str(Rule1()) == "<Rule1>"
+    assert str(Rule2()) == "<Rule2>"
+    assert str(Rule1Disabled()) == "<Disabled Rule1>"