matthias/mayan-edms
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update test login function to accept any argument

Browse Source 
Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2019-04-17 02:01:46 -04:00
parent f176f6ae4f
commit 97a9d21ac4
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
mayan/apps/user_management/tests/mixins.py
View File

@@ -23,12 +23,20 @@ class UserTestMixin(object):
class UserTestCaseMixin(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): def tearDown(self):
self.client.logout() self.client.logout()
super(UserTestCaseMixin, self).tearDown() super(UserTestCaseMixin, self).tearDown()
def login(self, username, password): def login(self, *args, **kwargs):
logged_in = self.client.login(username=username, password=password) logged_in = self.client.login(*args, **kwargs)
return logged_in return logged_in
def login_user(self): def login_user(self):