From 97a9d21ac49f18b0d81494e388c2dc42201b6eca Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Wed, 17 Apr 2019 02:01:46 -0400 Subject: [PATCH] Update test login function to accept any argument Signed-off-by: Roberto Rosario --- mayan/apps/user_management/tests/mixins.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/mayan/apps/user_management/tests/mixins.py b/mayan/apps/user_management/tests/mixins.py index 068586ba6c..91a0087a58 100644 --- a/mayan/apps/user_management/tests/mixins.py +++ b/mayan/apps/user_management/tests/mixins.py @@ -23,12 +23,20 @@ class UserTestMixin(object): class UserTestCaseMixin(object): + auto_login_user = False + + def setUp(self): + super(UserTestCaseMixin, self).setUp() + if self.auto_login_user: + self.login_user() + def tearDown(self): self.client.logout() super(UserTestCaseMixin, self).tearDown() - def login(self, username, password): - logged_in = self.client.login(username=username, password=password) + def login(self, *args, **kwargs): + logged_in = self.client.login(*args, **kwargs) + return logged_in def login_user(self):