浏览代码

create custom domain in fake_data

Son NK 5 年之前
父节点
当前提交
a22a635508
共有 1 个文件被更改,包括 5 次插入1 次删除
  1. 5 1
      server.py

+ 5 - 1
server.py

@@ -38,7 +38,7 @@ from app.models import (
     Subscription,
     PlanEnum,
     ApiKey,
-)
+    CustomDomain)
 from app.monitor.base import monitor_bp
 from app.oauth.base import oauth_bp
 
@@ -121,6 +121,10 @@ def fake_data():
     GenEmail.create_custom_alias(user.id, "e2@")
     GenEmail.create_custom_alias(user.id, "e3@")
 
+    CustomDomain.create(user_id=user.id, domain="ab.cd", verified=True)
+    CustomDomain.create(user_id=user.id, domain="very-long-domain.com.net.org", verified=True)
+    db.session.commit()
+
     # Create a client
     client1 = Client.create_new(name="Demo", user_id=user.id)
     client1.oauth_client_id = "client-id"