From cb8b20fc9a4dc1a6d599eb1823bf84747a73433c Mon Sep 17 00:00:00 2001
From: Son <nguyenkims@users.noreply.github.com>
Date: Fri, 15 Apr 2022 17:37:19 +0200
Subject: [PATCH] fix test

---
 tests/dashboard/test_unsubscribe.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/dashboard/test_unsubscribe.py b/tests/dashboard/test_unsubscribe.py
index af755276..d5bae207 100644
--- a/tests/dashboard/test_unsubscribe.py
+++ b/tests/dashboard/test_unsubscribe.py
@@ -1,3 +1,4 @@
+from app.db import Session
 from app.models import (
     Alias,
     Contact,
@@ -6,8 +7,9 @@ from tests.utils import login
 
 
 def test_disable_alias(flask_client):
-    login(flask_client)
-    alias = Alias.first()
+    user = login(flask_client)
+    alias = Alias.create_new_random(user)
+    Session.commit()
 
     assert alias.enabled
     flask_client.post(f"/dashboard/unsubscribe/{alias.id}")