Properly unregister alarm manager callbacks.

Change-Id: I2d474bc1cee168a5b76b8710b382498cd6c2fd59

2 files changed, 18 insertions, 0 deletions

diff --git a/provider_src/com/android/email/mail/store/ImapFolder.java b/provider_src/com/android/email/mail/store/ImapFolder.java
index 05f44c477..bb7a6fe4b 100644
--- a/provider_src/com/android/email/mail/store/ImapFolder.java
+++ b/provider_src/com/android/email/mail/store/ImapFolder.java
@@ -85,6 +85,10 @@ public class ImapFolder extends Folder {
          */
         public void onIdled();
         /**
+         * Invoked when idle state is left
+         */
+        public void onIdlingDone();
+        /**
          * Invoked when a new change is communicated by the server.
          *
          * @param needSync whether a sync is required
@@ -313,6 +317,9 @@ public class ImapFolder extends Folder {
                         mIdling = false;
                     }
 
+                    if (callback != null) {
+                        callback.onIdlingDone();
+                    }
                 } catch (MessagingException me) {
                     cleanupConnection(connection, false);
                     synchronized (mIdleSync) {
diff --git a/provider_src/com/android/email/service/ImapService.java b/provider_src/com/android/email/service/ImapService.java
index 292a64218..52ba306d3 100644
--- a/provider_src/com/android/email/service/ImapService.java
+++ b/provider_src/com/android/email/service/ImapService.java
@@ -198,6 +198,12 @@ public class ImapService extends Service {
         }
 
         @Override
+        public void onIdlingDone() {
+            cancelKickIdleConnection();
+            resetPingDelay();
+        }
+
+        @Override
         public void onNewServerChange(final boolean needSync, final List<String> fetchMessages) {
             // Instead of checking every received change, request a sync of the mailbox
             if (Logging.LOGD) {
@@ -551,6 +557,10 @@ public class ImapService extends Service {
             mContext = context;
         }
 
+        public void destroy() {
+            cancelIdleConnectionRestart();
+        }
+
         @Override
         public void onConnectivityRestored(int networkType) {
             if (Logging.LOGD) {
@@ -778,6 +788,7 @@ public class ImapService extends Service {
         ImapIdleFolderHolder.getInstance().unregisterAllIdledMailboxes(true);
         mConnectivityManager.unregister();
         getContentResolver().unregisterContentObserver(mLocalChangesObserver);
+        mConnectivityManager.destroy();
 
         super.onDestroy();
     }