from django.contrib.auth.models import User
from django.core.urlresolvers import reverse
from rest_framework import status
from rest_framework.test import APITestCase

from models import BlogPost
from serializers import UserSerializer, BlogPostSerializer


# User tests
class CreateUserTest(APITestCase):
    def setUp(self):
        self.superuser = User.objects.create_superuser('admin', 'admin@admin.com', 'admin1234')
        self.client.login(username='admin', password='admin1234')
        self.data = {'username': 'korisnik'}

    def test_can_create_user(self):
        response = self.client.post(reverse('api:user-list'), self.data)
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)


class GetUserTest(APITestCase):
    def setUp(self):
        self.superuser = User.objects.create_superuser('admin', 'admin@admin.com', 'admin1234')
        self.client.login(username='admin', password='admin1234')
        self.user = User.objects.create(username="korisnik")

    def test_can_read_user_list(self):
        response = self.client.get(reverse('api:user-list'))
        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_can_read_user_detail(self):
        response = self.client.get(reverse('api:user-detail', args=[self.user.id]))
        self.assertEqual(response.status_code, status.HTTP_200_OK)


class UpdateUserTest(APITestCase):
    def setUp(self):
        self.superuser = User.objects.create_superuser('admin', 'admin@admin.com', 'admin1234')
        self.client.login(username='admin', password='admin1234')
        self.user = User.objects.create(username="korisnik")
        self.data = UserSerializer(self.user).data
        self.data.update({'username': 'noviusername'})

    def test_can_update_user(self):
        response = self.client.put(reverse('api:user-detail', args=[self.user.id]), self.data)
        self.assertEqual(response.status_code, status.HTTP_200_OK)


class DeleteUserTest(APITestCase):
    def setUp(self):
        self.superuser = User.objects.create_superuser('admin', 'admin@admin.com', 'admin1234')
        self.client.login(username='admin', password='admin1234')
        self.user = User.objects.create(username="korisnik")

    def test_can_delete_user(self):
        response = self.client.delete(reverse('api:user-detail', args=[self.user.id]))
        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)


# BlogPost
class CreateBlogPostTest(APITestCase):
    def setUp(self):
        self.superuser = User.objects.create_superuser('admin', 'admin@admin.com', 'admin1234')
        self.client.login(username='admin', password='admin1234')
        self.data = {
            'title': 'Title',
            'text': 'Text'
        }

    def test_can_create_post(self):
        response = self.client.post(reverse('api:post-list'), self.data)
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)


class GetBlogPostTest(APITestCase):
    def setUp(self):
        self.superuser = User.objects.create_superuser('admin', 'admin@admin.com', 'admin1234')
        self.client.login(username='admin', password='admin1234')
        self.post = BlogPost.objects.create(owner=self.superuser, title='title', text='text')

    def test_can_read_post_list(self):
        response = self.client.get(reverse('api:post-list'))
        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_can_read_post_detail(self):
        response = self.client.get(reverse('api:post-detail', args=[self.post.id]))
        self.assertEqual(response.status_code, status.HTTP_200_OK)


class UpdateBlogPostTest(APITestCase):
    def setUp(self):
        self.superuser = User.objects.create_superuser('admin', 'admin@admin.com', 'admin1234')
        self.client.login(username='admin', password='admin1234')
        self.post = BlogPost.objects.create(owner=self.superuser, title='title', text='text')
        self.data = BlogPostSerializer(self.post).data
        self.data.update({'title': 'novititle'})

    def test_can_update_user(self):
        response = self.client.put(reverse('api:post-detail', args=[self.post.id]), self.data)
        self.assertEqual(response.status_code, status.HTTP_200_OK)


class DeleteBlogPostTest(APITestCase):
    def setUp(self):
        self.superuser = User.objects.create_superuser('admin', 'admin@admin.com', 'admin1234')
        self.client.login(username='admin', password='admin1234')
        self.post = BlogPost.objects.create(owner=self.superuser, title='title', text='text')

    def test_can_delete_user(self):
        response = self.client.delete(reverse('api:post-detail', args=[self.post.id]))
        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)