From d2c99ea00eae3a3ad6addc1e34f1f795950990f9 Mon Sep 17 00:00:00 2001
From: Son NK <>
Date: Thu, 18 Mar 2021 14:45:31 +0100
Subject: [PATCH] add more fake data

---
 server.py | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/server.py b/server.py
index 81353a2a..dcc4e1d2 100644
--- a/server.py
+++ b/server.py
@@ -330,12 +330,28 @@ def fake_data():
             a.enabled = False
             db.session.commit()
 
-    CustomDomain.create(user_id=user.id, domain="ab.cd", verified=True)
-    CustomDomain.create(
+    custom_domain1 = CustomDomain.create(user_id=user.id, domain="ab.cd", verified=True)
+    custom_domain2 = CustomDomain.create(
         user_id=user.id, domain="very-long-domain.com.net.org", verified=True
     )
     db.session.commit()
 
+    Alias.create(
+        user_id=user.id,
+        email="first@ab.cd",
+        mailbox_id=user.default_mailbox_id,
+        custom_domain_id=custom_domain1.id,
+        commit=True,
+    )
+
+    Alias.create(
+        user_id=user.id,
+        email="second@ab.cd",
+        mailbox_id=user.default_mailbox_id,
+        custom_domain_id=custom_domain1.id,
+        commit=True,
+    )
+
     Directory.create(user_id=user.id, name="abcd")
     Directory.create(user_id=user.id, name="xyzt")
     db.session.commit()