Fixing integration tests.

1 files changed, 7 insertions, 2 deletions

diff --git a/service/pixelated/adapter/services/mailbox.py b/service/pixelated/adapter/services/mailbox.py
index a4029d78..a083e8a6 100644
--- a/service/pixelated/adapter/services/mailbox.py
+++ b/service/pixelated/adapter/services/mailbox.py
@@ -14,6 +14,8 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
 
+from twisted.internet import defer
+
 
 class Mailbox(object):
 
@@ -30,10 +32,13 @@ class Mailbox(object):
     def mail(self, mail_id):
         return self.querier.mail(mail_id)
 
+    @defer.inlineCallbacks
     def add(self, mail):
-        added_mail = self.querier.create_mail(mail, self.mailbox_name)
+        added_mail = yield self.querier.create_mail(mail, self.mailbox_name)
+
+        import pdb; pdb.set_trace()
         self.search_engine.index_mail(added_mail)
-        return added_mail
+        defer.returnValue(added_mail)
 
     def remove(self, ident):
         mail = self.querier.mail(ident)