test.PixivDBManager.py

#!/c/Python27/python.exe
# -*- coding: UTF-8 -*-

from PixivDBManager import PixivDBManager
from PixivModel import PixivListItem
from PixivConfig import PixivConfig

import unittest
LIST_SIZE = 4
config = PixivConfig()
config.loadConfig()

class TestPixivDBManager(unittest.TestCase):
    def testImportListTxt(self):
        DB = PixivDBManager(target = "test.db.sqlite")
        DB.createDatabase()
        l = PixivListItem.parseList("test.list.txt", config.rootDirectory)
        result = DB.importList(l)
        self.assertEqual(result, 0)

    def testSelectMembersByLastDownloadDate(self):
        DB = PixivDBManager(target = "test.db.sqlite")
        DB.createDatabase()
        l = PixivListItem.parseList("test.list.txt", config.rootDirectory)
        result = DB.selectMembersByLastDownloadDate(7)
        self.assertEqual(len(result), LIST_SIZE)
        for item in result:
            print item.memberId, item.path

    def testSelectAllMember(self):
        DB = PixivDBManager(target = "test.db.sqlite")
        DB.createDatabase()
        l = PixivListItem.parseList("test.list.txt", config.rootDirectory)
        result = DB.selectAllMember()
        self.assertEqual(len(result), LIST_SIZE)
        for item in result:
            print item.memberId, item.path

if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(TestPixivDBManager)
    unittest.TextTestRunner(verbosity=5).run(suite)
    print "================================================================"