def createWorkers(self):
"""There aren't many workers in an IMAP State - most of the
work is done elsewhere. We do need to load the classifier,
though, and build the status strings."""
# Load token and message databases.
if not hasattr(self, "DBName"):
self.DBName, self.useDB = storage.database_type([])
self.bayes = storage.open_storage(self.DBName, self.useDB)
if not hasattr(self, "MBDName"):
self.MDBName, self.useMDB = message.database_type()
self.mdb = message.open_storage(self.MDBName, self.useMDB)
# Load stats manager.
self.stats = Stats(options, self.mdb)
# Build status strings.
self.buildStatusStrings()
def test_set_date(self):
now = time.time()
self.s.ResetTotal(permanently=True)
self.assertEqual(now, self.s.from_date)
for stat in [
"num_ham",
"num_spam",
"num_unsure",
"num_trained_spam",
"num_trained_spam_fn",
"num_trained_ham",
"num_trained_ham_fp",
]:
self.assertEqual(self.s.totals[stat], 0)
# Check that it was stored, too.
self.messageinfo_db.close()
self.messageinfo_db = MessageInfoPickle(self.messageinfo_db_name)
self.s = Stats(options, self.messageinfo_db)
self.assertEqual(now, self.s.from_date)
def setUp(self):
self.messageinfo_db_name = "__unittest.pik"
self.messageinfo_db = MessageInfoPickle(self.messageinfo_db_name)
self.s = Stats(options, self.messageinfo_db)
Message.message_info_db = self.messageinfo_db
