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):