Issue 395 - Filter hits statistics backend

sitescripts/filterhits/test/db_tests.py

+# coding: utf-8
+
+# This file is part of the Adblock Plus web scripts,
+# Copyright (C) 2006-2014 Eyeo GmbH
+#
+# Adblock Plus is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 3 as
+# published by the Free Software Foundation.
+#
+# Adblock Plus 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 Adblock Plus.  If not, see <http://www.gnu.org/licenses/>.
+
+import unittest, MySQLdb
+from datetime import datetime
+from ConfigParser import NoOptionError
+
+from sitescripts.utils import get_config
+import sitescripts.filterhits.db as db
+
+class DbTestCase(unittest.TestCase):
+  longMessage = True
+  maxDiff = None
+
+  def clear_rows(self):
+    if self.db:
+      db.write("DELETE FROM filters;")
+
+  def setUp(self):
+    db.disconnect()
+    try:
+      self.config = get_config()
+      self.db = db.connect(
+        self.config.get("filterhitstats", "dbuser"),
+        self.config.get("filterhitstats", "dbpassword"),
+        self.config.get("filterhitstats", "test_database"))
+    except (MySQLdb.Error, NoOptionError):
+      self.db = None
+    self.clear_rows()
+
+  def tearDown(self):
+    self.clear_rows()
+    db.disconnect()
+
+  def test_query_and_write(self):
+    if not self.db:
+      raise unittest.SkipTest("Not connected to test DB.")
+
+    insert_sql = """INSERT INTO `filters` (filter, md5)
+                    VALUES ('%s', UNHEX(MD5(filter)));"""
+    select_sql = "SELECT filter FROM filters ORDER BY filter ASC;"
+
+    # Table should be empty to start with
+    self.assertEqual(db.query(select_sql), ())
+    # Write some data and query it back
+    db.write(insert_sql % "something")
+    self.assertEqual(db.query(select_sql), ((u"something",),))
+    # Write an array of SQL strings
+    db.write([insert_sql % "a", insert_sql % "b", insert_sql % "c"])
+    self.assertEqual(db.query(select_sql), ((u"a",), (u"b",), (u"c",), (u"something",)))
+    # Write an array of concatinated SQL strings
+    db.write([insert_sql % "d" + insert_sql % "e"])
+    self.assertEqual(db.query(select_sql), ((u"a",), (u"b",), (u"c",), (u"d",), (u"e",),
+                                            (u"something",)))
+    # Write a sequence of SQL but roll back when a problem arrises
+    with self.assertRaises(MySQLdb.ProgrammingError):
+      db.write([insert_sql % "f" + insert_sql % "g", insert_sql % "h", "GFDGks"])
+    self.assertEqual(db.query(select_sql), ((u"a",), (u"b",), (u"c",), (u"d",), (u"e",),
+                                            (u"something",)))
+
+if __name__ == '__main__':
+  unittest.main()