From 12ffea333922d99ee7f7b4ab2cd46cfcec6a0d05 Mon Sep 17 00:00:00 2001
From: Kali Kaneko <kali@leap.se>
Date: Mon, 17 Feb 2014 11:31:40 -0400
Subject: fix get_size call

---
 src/leap/mail/imap/memorystore.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/leap/mail/imap/memorystore.py')

diff --git a/src/leap/mail/imap/memorystore.py b/src/leap/mail/imap/memorystore.py
index ed2b3f2..d0321ae 100644
--- a/src/leap/mail/imap/memorystore.py
+++ b/src/leap/mail/imap/memorystore.py
@@ -350,6 +350,12 @@ class MemoryStore(object):
                 continue
             self._phash_store[phash] = weakref.proxy(referenciable_cdoc)
 
+        # Update memory store size
+        # XXX this should use [mbox][uid]
+        key = mbox, uid
+        self._sizes[key] = size.get_size(self._fdoc_store[key])
+        # TODO add hdoc and cdocs sizes too
+
         def prune(seq, store):
             for key in seq:
                 if key in store and empty(store.get(key)):
-- 
cgit v1.2.3