| OLD | NEW |
| (Empty) |
| 1 # coding: utf-8 | |
| 2 | |
| 3 # This file is part of the Adblock Plus web scripts, | |
| 4 # Copyright (C) 2006-2016 Eyeo GmbH | |
| 5 # | |
| 6 # Adblock Plus is free software: you can redistribute it and/or modify | |
| 7 # it under the terms of the GNU General Public License version 3 as | |
| 8 # published by the Free Software Foundation. | |
| 9 # | |
| 10 # Adblock Plus is distributed in the hope that it will be useful, | |
| 11 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 13 # GNU General Public License for more details. | |
| 14 # | |
| 15 # You should have received a copy of the GNU General Public License | |
| 16 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | |
| 17 | |
| 18 import unittest | |
| 19 | |
| 20 import MySQLdb | |
| 21 | |
| 22 from sitescripts.filterhits.test import test_helpers | |
| 23 from sitescripts.filterhits import db | |
| 24 | |
| 25 | |
| 26 class DbTestCase(test_helpers.FilterhitsTestCase): | |
| 27 longMessage = True | |
| 28 maxDiff = None | |
| 29 | |
| 30 def test_query_and_write(self): | |
| 31 insert_sql = """INSERT INTO `filters` (filter, sha1) | |
| 32 VALUES (%s, UNHEX(SHA1(filter)))""" | |
| 33 select_sql = "SELECT filter FROM filters ORDER BY filter ASC" | |
| 34 | |
| 35 # Table should be empty to start with | |
| 36 self.assertEqual(db.query(self.db, select_sql), ()) | |
| 37 # Write some data and query it back | |
| 38 db.write(self.db, ((insert_sql, "something"),)) | |
| 39 self.assertEqual(db.query(self.db, select_sql), ((u"something",),)) | |
| 40 # Write an array of SQL strings | |
| 41 db.write(self.db, ((insert_sql, "a"), (insert_sql, "b"), (insert_sql, "c
"))) | |
| 42 self.assertEqual(db.query(self.db, select_sql), ((u"a",), (u"b",), (u"c"
,), (u"something",))) | |
| 43 # Write a sequence of SQL but roll back when a problem arrises | |
| 44 with self.assertRaises(MySQLdb.ProgrammingError): | |
| 45 db.write(self.db, ((insert_sql, "f"), (insert_sql, "g"), (insert_sql
, "h"), | |
| 46 ("GFDGks",))) | |
| 47 self.assertEqual(db.query(self.db, select_sql), ((u"a",), (u"b",), (u"c"
,), (u"something",))) | |
| 48 | |
| 49 if __name__ == "__main__": | |
| 50 unittest.main() | |
| OLD | NEW |