[tracker/rss-enclosures] functional-tests: new tests for the stats API

[Roberto Guido <rguido src gnome org>]

commit 952d4ae031e06a7321cdbbae278a253dbdecb539
Author: Ivan Frade <ivan frade nokia com>
Date:   Wed Sep 15 20:38:20 2010 +0300

    functional-tests: new tests for the stats API

 tests/functional-tests/15-statistics.py |  121 +++++++++++++++++++++++++++++++
 tests/functional-tests/Makefile.am      |    1 +
 2 files changed, 122 insertions(+), 0 deletions(-)
---
diff --git a/tests/functional-tests/15-statistics.py b/tests/functional-tests/15-statistics.py
new file mode 100644
index 0000000..89eee4c
--- /dev/null
+++ b/tests/functional-tests/15-statistics.py
@@ -0,0 +1,121 @@
+#!/usr/bin/python
+#
+# Copyright (C) 2010, Nokia <ivan frade nokia com>
+#
+# 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.
+#
+
+"""
+Stand-alone tests cases for the store, checking the statistics
+are updated when different operations are executed on the store
+"""
+import time
+
+from common.utils import configuration as cfg
+import unittest2 as ut
+#import unittest as ut
+from common.utils.storetest import CommonTrackerStoreTest as CommonTrackerStoreTest
+
+RDFS_RESOURCE = "rdfs:Resource"
+NIE_IE = "nie:InformationElement"
+RDFS_CLASS = "rdfs:Class"
+
+class TrackerStoreStatisticsTests (CommonTrackerStoreTest):
+    """
+    Check initial statistics, add, remove, update content and check results stats
+    """
+    def __get_stats (self):
+        results = {}
+        for classname, count in self.tracker.get_stats ():
+            results [str(classname)] = int(count)
+        return results
+
+    def setUp (self):
+        """
+        Each test append to this list the used URIS, so they can be removed
+        in the tearDown
+        """
+        self.clean_up_instances = []
+
+    def tearDown (self):
+        for uri in self.clean_up_instances:
+            self.tracker.update ("DELETE { <%s> a rdfs:Resource. }" % (uri))
+        self.clean_up_instances = []
+        time.sleep (1)
+    
+    def test_stats_01_insert_base_class (self):
+        self.clean_up_instances.append ("test://stats-01")
+        
+        old_stats = self.__get_stats ()
+        self.tracker.update ("INSERT { <test://stats-01> a nie:InformationElement. }")
+        new_stats = self.__get_stats ()
+
+        increased_classes = [NIE_IE, RDFS_RESOURCE]
+
+        for k, v in new_stats.iteritems ():
+            if k in increased_classes:
+                self.assertEquals (old_stats[k]+1, new_stats[k])
+            else:
+                self.assertEquals (old_stats [k], new_stats [k],
+                                   "Class %s should have the same instances" % k)
+
+
+    def test_stats_02_insert_deep_class (self):
+        self.clean_up_instances.append ("test://stats-02")
+        old_stats = self.__get_stats ()
+        self.tracker.update ("INSERT { <test://stats-02> a nmm:Photo. }")
+        new_stats = self.__get_stats ()
+
+        increased_classes = [ NIE_IE, RDFS_RESOURCE]
+        new_classes = ["nmm:Photo", "nfo:Visual", "nfo:Image", "nfo:Media"]
+
+        # There were no instances of those classes before, check they are now
+        for c in new_classes:
+            self.assertIn (c, new_stats)
+        
+        for k, v in new_stats.iteritems ():
+            if k in increased_classes:
+                self.assertEquals (old_stats [k]+1, new_stats[k])
+            elif k in new_classes:
+                self.assertEquals (new_stats [k], 1)
+            else:
+                self.assertEquals (old_stats [k], new_stats[k])
+
+    def test_stats_03_delete_deep_class (self):
+        self.clean_up_instances.append ("test://stats-03")
+        self.tracker.update ("INSERT { <test://stats-03> a nmm:Photo. }")
+
+        old_stats = self.__get_stats ()
+        self.tracker.update ("DELETE { <test://stats-03> a rdfs:Resource. }")
+        new_stats = self.__get_stats ()
+
+        decreased_classes = [NIE_IE, RDFS_RESOURCE]
+        # These classes shouldn't have any instance, so they are not in the results
+        no_instances_classes = ["nmm:Photo", "nfo:Visual", "nfo:Image", "nfo:Media"]
+
+        for c in no_instances_classes:
+            self.assertNotIn (c, new_stats)
+
+        for k, v in new_stats.iteritems ():
+            if k in decreased_classes:
+                self.assertEquals (old_stats [k]-1, new_stats[k])
+            else:
+                self.assertEquals (old_stats [k], new_stats [k])
+
+if __name__ == "__main__":
+    ut.main ()
+
+    
diff --git a/tests/functional-tests/Makefile.am b/tests/functional-tests/Makefile.am
index e9c9dec..a44a4be 100644
--- a/tests/functional-tests/Makefile.am
+++ b/tests/functional-tests/Makefile.am
@@ -36,6 +36,7 @@ standard_tests +=			\
 	08-unique-insertions.py		\
 	09-concurrent-query.py		\
 	14-signals.py			\
+	15-statistics.py		\
 	200-backup-restore.py		\
 	300-miner-basic-ops.py
 if HAVE_TRACKER_FTS