mirror of
https://github.com/haiwen/seahub.git
synced 2025-06-28 07:57:50 +00:00
24 lines
802 B
Python
24 lines
802 B
Python
|
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
|