diff --git a/src/calibre/library/test.py b/src/calibre/library/test.py index 1a81755971..86363b439c 100644 --- a/src/calibre/library/test.py +++ b/src/calibre/library/test.py @@ -7,7 +7,7 @@ __docformat__ = 'restructuredtext en' Unit tests for database layer. ''' -import sys, unittest, os +import sys, unittest, os, cStringIO from itertools import repeat from calibre.ptempfile import PersistentTemporaryDirectory @@ -60,6 +60,11 @@ class DBTest(unittest.TestCase): self.assertEqual(ga(self.m2, p), ga(m2, p)) self.assertEqual(self.db.format(1, 'txt', index_is_id=True), 'test') + self.assertEqual(self.db.formats(1, index_is_id=True), 'TXT') + self.db.add_format(1, 'html', cStringIO.StringIO('
'), index_is_id=True) + self.assertEqual(self.db.formats(1, index_is_id=True), 'HTML,TXT') + self.db.remove_format(1, 'html', index_is_id=True) + self.assertEqual(self.db.formats(1, index_is_id=True), 'TXT') self.assertNotEqual(self.db.cover(1, index_is_id=True), None) self.assertEqual(self.db.cover(2, index_is_id=True), None) @@ -76,6 +81,12 @@ class DBTest(unittest.TestCase): self.assertEqual('new auth', self.db.authors(2)) self.assertEqual(self.db.format(3, 'txt', index_is_id=True), 'test') + def testSorting(self): + self.db.sort('authors', True) + self.assertEqual(self.db.authors(0), 'Test Author 1') + self.db.sort('rating', False) + self.assertEqual(self.db.rating(0), 3) + def suite(): return unittest.TestLoader().loadTestsFromTestCase(DBTest) diff --git a/src/calibre/web/feeds/recipes/nytimes.py b/src/calibre/web/feeds/recipes/nytimes.py index 17fe1b9b1b..de431d6532 100644 --- a/src/calibre/web/feeds/recipes/nytimes.py +++ b/src/calibre/web/feeds/recipes/nytimes.py @@ -5,9 +5,10 @@ __copyright__ = '2008, Kovid Goyal