Issue 395 - Filter hits statistics backend

sitescripts/filterhits/test/geometrical_mean_tests.py

 # coding: utf-8
 
 # This file is part of the Adblock Plus web scripts,
 # Copyright (C) 2006-2015 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
 from datetime import datetime
-
-import MySQLdb
 
 from sitescripts.filterhits.test import test_helpers
 from sitescripts.filterhits import db, geometrical_mean
 
 test_data =  [{
   "filters": {
     "##.top-box-right-ad": {
       "firstParty": {
         "acxiom-online.com": { "hits": 6, "latest": 1414817340948 },
         "google.com": { "hits": 50, "latest": 1414849084678 },
@@ skipping 30 matching lines @@
     },
     "stevedeace.com##.topAddHolder": {
       "firstParty": {
         "mathtag.com": { "hits": 14, "latest": 1415032601175 },
         "amazonaws.com": { "hits": 18, "latest": 1414977342966 }
       }
     }
   }
 }]
 
-class GeometricalMeanTestCase(unittest.TestCase):
+class GeometricalMeanTestCase(test_helpers.FilterhitsTestCase):
   longMessage = True
   maxDiff = None
 
   def geometrical(self, interval, new, new_timestamp, old, old_timestamp):
     delta_divby_interval = (new_timestamp - old_timestamp) / 1000 / float(interval)
     return long(round(old ** (1 - delta_divby_interval) * new ** delta_divby_interval))
 
-  def clear_rows(self):
-    if self.db:
-      db.write(self.db, (("DELETE FROM frequencies",),
-                         ("DELETE FROM filters",)))
-
-  def setUp(self):
-    self.config = test_helpers.setup_config()
-    try:
-      self.db = db.connect()
-    except MySQLdb.Error:
-      self.db = None
-    self.clear_rows()
-
-  def tearDown(self):
-    test_helpers.restore_config()
-    if self.db:
-      self.clear_rows()
-      self.db.close()
-      self.db = None
-
   def test_calculations(self):
-    if not self.db:
-      raise unittest.SkipTest("Not connected to test DB.")
-
     interval = 86400
 
     # Tables should be empty to start with
     self.assertEqual(db.query(self.db, "SELECT * FROM filters"), ())
     self.assertEqual(db.query(self.db, "SELECT * FROM frequencies"), ())
     # First batch
     db.write(self.db, geometrical_mean.update(interval, test_data[0]))
     self.assertEqual(db.query(self.db, "SELECT * FROM filters"),
                      (("8c5ea548436c61f05536e205a29ada6204f603b0".decode("hex"),
                        u"##.top-box-right-ad"),))
@@ skipping 68 matching lines @@
         14L, datetime.utcfromtimestamp(1415032601175 / 1000)),
        ("8c5ea548436c61f05536e205a29ada6204f603b0".decode("hex"), u"yahoo.com",
         self.geometrical(interval, 43, 1415045194098,
                          self.geometrical(interval, 27, 1414917270343,
                                           14, 1414859271125),
                          1414917270343),
         datetime.utcfromtimestamp(1415045194098 / 1000))))
 
 if __name__ == "__main__":
   unittest.main()