david/jmc
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Send presence when cancelling an account error

Browse Source 
darcs-hash:20080514200208-86b55-93374592f277f9b84bcad45ecf2754368f057f22.gz
This commit is contained in:
David Rousselie
2008-05-14 22:02:08 +02:00
parent 53cc7b91ca
commit af9ae34bf8
2 changed files with 62 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/jmc/jabber/feeder.py
View File

@@ -64,7 +64,7 @@ class MailFeeder(Feeder):
_account.mark_all_as_read() _account.mark_all_as_read()
_account.disconnect() _account.disconnect()
_account.first_check = False _account.first_check = False
_account.error = None self.component.account_manager.cancel_account_error(_account)
return True return True
except Exception, e: except Exception, e:
if _account.connected: if _account.connected:
@@ -129,7 +129,7 @@ class MailFeeder(Feeder):
raise Exception("Unkown action: " + str(action) \ raise Exception("Unkown action: " + str(action) \
+ "\nPlease reconfigure account.") + "\nPlease reconfigure account.")
_account.disconnect() _account.disconnect()
_account.error = None self.component.account_manager.cancel_account_error(_account)
self.__logger.debug("\nCHECK_MAIL ends " + _account.jid) self.__logger.debug("\nCHECK_MAIL ends " + _account.jid)
except Exception, e: except Exception, e:
if _account.connected: if _account.connected:

64
src/jmc/jabber/tests/component.py
View File

@@ -24,6 +24,7 @@
import unittest import unittest
import sys import sys
import time import time
import logging
from sqlobject import * from sqlobject import *
from sqlobject.dbconnection import TheURIOpener from sqlobject.dbconnection import TheURIOpener
@@ -322,7 +323,6 @@ class MailComponent_TestCase(JCLTestCase):
self.assertTrue(account11.has_connected) self.assertTrue(account11.has_connected)
def test_feed_retrieve_no_mail(self): def test_feed_retrieve_no_mail(self):
model.db_connect()
account11 = MockIMAPAccount(user=User(jid="test1@test.com"), account11 = MockIMAPAccount(user=User(jid="test1@test.com"),
name="account11", name="account11",
jid="account11@jmc.test.com") jid="account11@jmc.test.com")
@@ -340,7 +340,32 @@ class MailComponent_TestCase(JCLTestCase):
self.assertFalse(account11.connected) self.assertFalse(account11.connected)
self.assertTrue(account11.has_connected) self.assertTrue(account11.has_connected)
self.assertEquals(len(self.comp.stream.sent), 0) self.assertEquals(len(self.comp.stream.sent), 0)
model.db_disconnect()
def test_feed_cancel_error(self):
account11 = MockIMAPAccount(user=User(jid="test1@test.com"),
name="account11",
jid="account11@jmc.test.com")
account11._action = MailAccount.RETRIEVE
account11.status = account.DND
account11.error = "An error"
account11.first_check = False
account11.lastcheck = 1
account11.interval = 2
account11.password = "password"
account11.get_new_mail_list = lambda: []
result = self.comp.handler.feeder.feed(account11)
self.assertEquals(account11.error, None)
self.assertEquals(result, [])
self.assertEquals(account11.lastcheck, 0)
self.assertFalse(account11.connected)
self.assertTrue(account11.has_connected)
sent = self.comp.stream.sent
self.assertEquals(len(sent), 1)
self.assertEquals(sent[0].get_to(), "test1@test.com")
self.assertEquals(sent[0].get_from(), "account11@jmc.test.com")
self.assertEquals(sent[0].xmlnode.name, "presence")
self.assertEquals(sent[0].xmlnode.children.name, "show")
self.assertEquals(sent[0].xmlnode.children.content, "online")
def test_feed_retrieve_mail(self): def test_feed_retrieve_mail(self):
def mock_get_mail(index): def mock_get_mail(index):
@@ -434,7 +459,6 @@ class MailComponent_TestCase(JCLTestCase):
# 'initialize_live_email' test methods # 'initialize_live_email' test methods
########################################################################### ###########################################################################
def test_initialize_live_email(self): def test_initialize_live_email(self):
model.db_connect()
account11 = MockIMAPAccount(user=User(jid="test1@test.com"), account11 = MockIMAPAccount(user=User(jid="test1@test.com"),
name="account11", name="account11",
jid="account11@jmc.test.com") jid="account11@jmc.test.com")
@@ -451,7 +475,6 @@ class MailComponent_TestCase(JCLTestCase):
self.assertFalse(account11.connected) self.assertFalse(account11.connected)
self.assertTrue(account11.has_connected) self.assertTrue(account11.has_connected)
self.assertTrue(account11.marked_all_as_read) self.assertTrue(account11.marked_all_as_read)
model.db_disconnect()
def test_initialize_live_email_connection_error(self): def test_initialize_live_email_connection_error(self):
def raiser(): def raiser():
@@ -486,7 +509,6 @@ class MailComponent_TestCase(JCLTestCase):
def test_initialize_live_email_mark_as_read_error(self): def test_initialize_live_email_mark_as_read_error(self):
def raiser(): def raiser():
raise Exception raise Exception
model.db_connect()
account11 = MockIMAPAccount(user=User(jid="test1@test.com"), account11 = MockIMAPAccount(user=User(jid="test1@test.com"),
name="account11", name="account11",
jid="account11@jmc.test.com") jid="account11@jmc.test.com")
@@ -513,12 +535,10 @@ class MailComponent_TestCase(JCLTestCase):
self.assertFalse(account11.connected) self.assertFalse(account11.connected)
self.assertTrue(account11.has_connected) self.assertTrue(account11.has_connected)
self.assertFalse(account11.marked_all_as_read) self.assertFalse(account11.marked_all_as_read)
model.db_disconnect()
def test_initialize_live_email_disconnection_error(self): def test_initialize_live_email_disconnection_error(self):
def raiser(): def raiser():
raise Exception raise Exception
model.db_connect()
account11 = MockIMAPAccount(user=User(jid="test1@test.com"), account11 = MockIMAPAccount(user=User(jid="test1@test.com"),
name="account11", name="account11",
jid="account11@jmc.test.com") jid="account11@jmc.test.com")
@@ -546,7 +566,31 @@ class MailComponent_TestCase(JCLTestCase):
self.assertFalse(account11.connected) self.assertFalse(account11.connected)
self.assertTrue(account11.has_connected) self.assertTrue(account11.has_connected)
self.assertTrue(account11.marked_all_as_read) self.assertTrue(account11.marked_all_as_read)
model.db_disconnect()
def test_initialize_live_email_cancel_error(self):
account11 = MockIMAPAccount(user=User(jid="test1@test.com"),
name="account11",
jid="account11@jmc.test.com")
account11.status = account.DND
account11.error = "An error"
self.assertTrue(account11.first_check)
account11.live_email_only = True
account11.password = "password"
continue_checking = self.comp.handler.feeder.initialize_live_email(account11)
self.assertEquals(continue_checking, True)
self.assertFalse(account11.first_check)
self.assertFalse(account11.waiting_password_reply)
self.assertEquals(account11.error, None)
self.assertFalse(account11.connected)
self.assertTrue(account11.has_connected)
self.assertTrue(account11.marked_all_as_read)
sent = self.comp.stream.sent
self.assertEquals(len(sent), 1)
self.assertEquals(sent[0].get_to(), "test1@test.com")
self.assertEquals(sent[0].get_from(), "account11@jmc.test.com")
self.assertEquals(sent[0].xmlnode.name, "presence")
self.assertEquals(sent[0].xmlnode.children.name, "show")
self.assertEquals(sent[0].xmlnode.children.content, "online")
def test_disco_get_info_imap_node(self): def test_disco_get_info_imap_node(self):
self.comp.stream = MockStream() self.comp.stream = MockStream()
@@ -1463,4 +1507,8 @@ def suite():
return test_suite return test_suite
if __name__ == '__main__': if __name__ == '__main__':
logger = logging.getLogger()
logger.addHandler(logging.StreamHandler())
if '-v' in sys.argv:
logger.setLevel(logging.INFO)
unittest.main(defaultTest='suite') unittest.main(defaultTest='suite')