1
0
mirror of https://github.com/haiwen/seahub.git synced 2025-09-18 00:00:00 +00:00

Fix api error when 2fa enabled

This commit is contained in:
zhengxie
2016-08-08 15:59:19 +08:00
parent c53791f27a
commit c255735326
3 changed files with 28 additions and 5 deletions

View File

@@ -0,0 +1,23 @@
from mock import patch
from seahub.test_utils import BaseTestCase
from seahub.api2.serializers import AuthTokenSerializer
class AuthTokenSerializerTest(BaseTestCase):
def test_validate(self):
s = AuthTokenSerializer(data={
'username': self.user.username,
'password': self.user_password,
}, context={'request': self.fake_request})
assert s.is_valid() is True
@patch('seahub.api2.serializers.has_two_factor_auth')
def test_two_factor_auth(self, mock_has_two_factor_auth):
mock_has_two_factor_auth.return_value = True
s = AuthTokenSerializer(data={
'username': self.user.username,
'password': self.user_password,
}, context={'request': self.fake_request})
assert s.is_valid() is True