From dafa23c5bf29ed5f20f94205de7870714e612e1b Mon Sep 17 00:00:00 2001
From: Son NK <>
Date: Sun, 3 May 2020 16:03:29 +0200
Subject: [PATCH] Add fake aliases with multiple mailboxes

---
 server.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/server.py b/server.py
index f20799fc..18d6e916 100644
--- a/server.py
+++ b/server.py
@@ -52,6 +52,7 @@ from app.models import (
     Contact,
     EmailLog,
     Referral,
+    AliasMailbox,
 )
 from app.monitor.base import monitor_bp
 from app.oauth.base import oauth_bp
@@ -164,13 +165,23 @@ def fake_data():
     m1 = Mailbox.create(user_id=user.id, email="m1@cd.ef", verified=True)
     db.session.commit()
 
-    for i in range(30):
+    for i in range(31):
         if i % 2 == 0:
             a = Alias.create_new(user, f"e{i}@", mailbox_id=m1.id)
+
         else:
             a = Alias.create_new(user, f"e{i}@")
         db.session.commit()
 
+        if i % 5 == 0:
+            if i % 2 == 0:
+                AliasMailbox.create(
+                    user_id=user.id, alias_id=a.id, mailbox_id=user.default_mailbox_id
+                )
+            else:
+                AliasMailbox.create(user_id=user.id, alias_id=a.id, mailbox_id=m1.id)
+        db.session.commit()
+
         # some aliases don't have any activity
         if i % 3 != 0:
             contact = Contact.create(