def test_pop_client(self): account_config = { "user_name": "foo@localhost", "password": "******", "protocol": "pop3", "hostname": "localhost:{0}".format(SpamCanPOPTest.server_port), "smtp_host": "localhost" } account = database.Account(account_config) mail_handler = MailUtil() protocol_handler = mail_handler.request(account) count = protocol_handler.get_stats() protocol_handler.disconnect() self.assert_(count == 1)
def test_get_stats_method(self): tmpdir = tempfile.mkdtemp() try: write_config_files(tmpdir, SpamCanPOPTest.server_port) mail_handler = MailUtil() db = database.Database(conf_dir=tmpdir) account = db.fetch_by_id(1) protocol_handler = mail_handler.request(account) if protocol_handler: account.remote_count = protocol_handler.get_stats() protocol_handler.disconnect() self.assert_(account.remote_count == 1) finally: db.session.close() if os.path.isdir(tmpdir): shutil.rmtree(tmpdir)