import mock
from .base_accounts import BaseAccountsTestCase

# Create your tests here.


class UserRegisterTestCase(BaseAccountsTestCase):
    def test_user_create_invalid_token(self):
        data = {
            'firebase_token': '345dfg',
            'device_token': '4w35ty',
            'device_type': 'ios'
        }
        urlname = self.get_url_by_name('customuser-list')
        response = self.client.post(path=urlname, data=data)
        self.assertEqual(response.status_code, 400)

    @mock.patch('rest.serializers.UserRegisterSerializer.get_phone_number')
    def test_user_create_success(self, mock_get_phone_number):
        mock_get_phone_number.return_value = '79129129121'
        data = {
            'firebase_token': '345dfg',
            'device_token': '4w35ty',
            'device_type': 'ios'
        }
        urlname = self.get_url_by_name('customuser-list')
        response = self.client.post(path=urlname, data=data)
        self.assertEqual(response.status_code, 201)