from rest_framework.test import APITestCase, APIClient
from django.urls import reverse
from smartpicasso.app.user.models import User
from rest_framework import status


class UserApiTest(APITestCase):
    client = APIClient()

    def test_login_when_user_non_exist(self):
        url = reverse('authenticate')
        response = self.client.post(url, {'email': 'non-exist', 'password': '123'}, format='json')
        assert response.status_code == 400

    def test_login_when_user_exist(self):
        User.objects.create_user(email='test@test.com', password='test')
        url = reverse('authenticate')
        response = self.client.post(url, {'email': 'test@test.com', 'password': 'test'}, format='json')
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(response.data['success'], 'True')
        self.assertIn('token', response.data)