Add unit tests for repository rules

gramps/gen/filters/rules/test/repository_rules_test.py

@@ -0,0 +1,131 @@
+#
+# Gramps - a GTK+/GNOME based genealogy program
+#
+# Copyright (C) 2016 Tom Samstag
+#
+# 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.
+#
+
+"""
+Unittest that tests repository-specific filter rules
+"""
+import unittest
+
+from gramps.gen.merge.diff import import_as_dict
+from gramps.cli.user import User
+from gramps.gen.filters import GenericFilterFactory
+
+from gramps.gen.filters.rules.repository import (
+    AllRepos, HasIdOf, RegExpIdOf, HasNoteRegexp, HasReferenceCountOf,
+    RepoPrivate, ChangedSince, MatchesNameSubstringOf, HasTag)
+
+GenericRepositoryFilter = GenericFilterFactory('Repository')
+
+class BaseTest(unittest.TestCase):
+    """
+    Repository rule tests.
+    """
+
+    @classmethod
+    def setUpClass(cls):
+        """
+        Import example database.
+        """
+        cls.db = import_as_dict("example/gramps/example.gramps", User())
+
+    def filter_with_rule(self, rule):
+        """
+        Apply a filter with the given rule.
+        """
+        filter_ = GenericRepositoryFilter()
+        filter_.add_rule(rule)
+        results = filter_.apply(self.db)
+        return set(results)
+
+    def test_allrepos(self):
+        """
+        Test AllRepos rule.
+        """
+        rule = AllRepos([])
+        self.assertEqual(len(self.filter_with_rule(rule)),
+                         self.db.get_number_of_repositories())
+
+    def test_hasidof(self):
+        """
+        Test HasIdOf rule.
+        """
+        rule = HasIdOf(['R0000'])
+        self.assertEqual(self.filter_with_rule(rule),
+                         set([b'b39fe38593f3f8c4f12']))
+
+    def test_regexpidof(self):
+        """
+        Test RegExpIdOf rule.
+        """
+        rule = RegExpIdOf(['R000.'], use_regex=True)
+        self.assertEqual(self.filter_with_rule(rule), set([
+            b'a701ead12841521cd4d', b'a701e99f93e5434f6f3',
+            b'b39fe38593f3f8c4f12']))
+
+    def test_hasnoteregexp(self):
+        """
+        Test HasNoteRegexp rule.
+        """
+        rule = HasNoteRegexp(['.'], use_regex=True)
+        self.assertEqual(self.filter_with_rule(rule), set([
+            b'a701ead12841521cd4d', b'b39fe38593f3f8c4f12']))
+
+    def test_hasreferencecountof(self):
+        """
+        Test HasReferenceCountOf rule.
+        """
+        rule = HasReferenceCountOf(['greater than', '1'])
+        self.assertEqual(self.filter_with_rule(rule),
+                         set([b'a701e99f93e5434f6f3']))
+
+    def test_repoprivate(self):
+        """
+        Test RepoPrivate rule.
+        """
+        rule = RepoPrivate([])
+        self.assertEqual(self.filter_with_rule(rule), set([]))
+
+    def test_changedsince(self):
+        """
+        Test ChangedSince rule.
+        """
+        rule = ChangedSince(['2010-01-01', '2016-01-01'])
+        self.assertEqual(self.filter_with_rule(rule),
+                         set([b'a701e99f93e5434f6f3']))
+
+    def test_matchesnamesubstringof(self):
+        """
+        Test MatchesNameSubstringOf rule.
+        """
+        rule = MatchesNameSubstringOf(['Martha'])
+        self.assertEqual(self.filter_with_rule(rule),
+                         set([b'a701ead12841521cd4d']))
+
+    def test_hastag(self):
+        """
+        Test HasTag rule.
+        """
+        rule = HasTag(['ToDo'])
+        self.assertEqual(self.filter_with_rule(rule),
+                         set())
+
+
+if __name__ == "__main__":
+    unittest.main()

po/POTFILES.skip

@@ -87,6 +87,7 @@ gramps/gen/filters/rules/test/media_rules_test.py
 gramps/gen/filters/rules/test/note_rules_test.py
 gramps/gen/filters/rules/test/person_rules_test.py
 gramps/gen/filters/rules/test/place_rules_test.py
+gramps/gen/filters/rules/test/repository_rules_test.py
 #
 # gen.filters.rules.person package
 #