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