Skip to content
Snippets Groups Projects
Verified Commit 3b964505 authored by Mark's avatar Mark
Browse files

Refactor tearDown function

parent 7837c3a8
No related branches found
No related tags found
No related merge requests found
...@@ -18,9 +18,18 @@ class GraphQlTests(LiveServerTestCase): ...@@ -18,9 +18,18 @@ class GraphQlTests(LiveServerTestCase):
def tearDown(self): def tearDown(self):
# clean up tables # clean up tables
users = User.query.filter_by(username="graphqluser").delete() for username in ["graphqluser"]:
roles = Role.query.filter_by(name="testusers").delete() user = db_session.query(User).filter_by(username=username).first()
db_session.commit() if user is not None:
user.roles = []
user.applications = []
db_session.delete(user)
db_session.commit()
role = db_session.query(Role).filter_by(name="testusers").first()
if role is not None:
role.users = []
db_session.delete(role)
db_session.commit()
def test_query_create_and_get_user(self): def test_query_create_and_get_user(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment