From f9a9a695526afea8eef4ccba8904fb215acc43dd Mon Sep 17 00:00:00 2001
From: Kali Kaneko <kali@leap.se>
Date: Wed, 8 Jan 2014 23:29:18 -0400
Subject: add a flag to be able to close the session

---
 src/leap/mail/imap/account.py      |  1 +
 src/leap/mail/imap/service/imap.py | 12 +++---------
 2 files changed, 4 insertions(+), 9 deletions(-)

(limited to 'src')

diff --git a/src/leap/mail/imap/account.py b/src/leap/mail/imap/account.py
index fd861e7..8caafef 100644
--- a/src/leap/mail/imap/account.py
+++ b/src/leap/mail/imap/account.py
@@ -46,6 +46,7 @@ class SoledadBackedAccount(WithMsgFields, IndexedDB, MBoxParser):
 
     _soledad = None
     selected = None
+    closed = False
 
     def __init__(self, account_name, soledad=None):
         """
diff --git a/src/leap/mail/imap/service/imap.py b/src/leap/mail/imap/service/imap.py
index 234996d..dfd4862 100644
--- a/src/leap/mail/imap/service/imap.py
+++ b/src/leap/mail/imap/service/imap.py
@@ -71,15 +71,6 @@ class LeapIMAPServer(imap4.IMAP4Server):
         # populate the test account properly (and only once
         # per session)
 
-        # theAccount = SoledadBackedAccount(
-        #     user, soledad=soledad)
-
-        # ---------------------------------
-        # XXX pre-populate acct for tests!!
-        # populate_test_account(theAccount)
-        # ---------------------------------
-        #self.theAccount = theAccount
-
     def lineReceived(self, line):
         """
         Attempt to parse a single line from the server.
@@ -88,6 +79,9 @@ class LeapIMAPServer(imap4.IMAP4Server):
         :type line: str
         """
         print "RECV: STATE (%s)" % self.state
+        if self.theAccount.closed is True and self.state != "unauth":
+            log.msg("Closing the session. State: unauth")
+            self.state = "unauth"
 
         if "login" in line.lower():
             # avoid to log the pass, even though we are using a dummy auth
-- 
cgit v1.2.3