from __future__ import unicode_literals

import json

from django.contrib.auth.models import AnonymousUser, User
from django.core.exceptions import PermissionDenied
from django.test import TestCase
from django.test.client import RequestFactory
from django.urls import reverse
from django.utils.encoding import force_text

from djadmin2 import apiviews
from djadmin2.site import djadmin2_site
from djadmin2.types import ModelAdmin2
from ..models import Post


class APITestCase(TestCase):
    def setUp(self):
        self.factory = RequestFactory()
        self.user = User(
            username='admin',
            is_staff=True)
        self.user.set_password('admin')
        self.user.save()

    def get_model_admin(self, model):
        return ModelAdmin2(model, djadmin2_site)


class IndexAPIViewTest(APITestCase):
    def test_response_ok(self):
        request = self.factory.get(reverse('admin2:api_index'))
        request.user = self.user
        view = apiviews.IndexAPIView.as_view(
            **djadmin2_site.get_api_index_kwargs())
        response = view(request)
        self.assertEqual(response.status_code, 200)

    def test_view_permission(self):
        request = self.factory.get(reverse('admin2:api_index'))
        request.user = AnonymousUser()
        view = apiviews.IndexAPIView.as_view(
            **djadmin2_site.get_api_index_kwargs())
        self.assertRaises(PermissionDenied, view, request)


class ListCreateAPIViewTest(APITestCase):
    def test_response_ok(self):
        request = self.factory.get(reverse('admin2:blog_post_api_list'))
        request.user = self.user
        model_admin = self.get_model_admin(Post)
        view = apiviews.ListCreateAPIView.as_view(
            **model_admin.get_api_list_kwargs())
        response = view(request)
        self.assertEqual(response.status_code, 200)

    def test_view_permission(self):
        request = self.factory.get(reverse('admin2:blog_post_api_list'))
        request.user = AnonymousUser()
        model_admin = self.get_model_admin(Post)
        view = apiviews.ListCreateAPIView.as_view(
            **model_admin.get_api_list_kwargs())
        self.assertRaises(PermissionDenied, view, request)

    def test_list_includes_unicode_field(self):
        Post.objects.create(title='Foo', body='Bar')
        request = self.factory.get(reverse('admin2:blog_post_api_list'))
        request.user = self.user
        model_admin = self.get_model_admin(Post)
        view = apiviews.ListCreateAPIView.as_view(
            **model_admin.get_api_list_kwargs())
        response = view(request)
        response.render()

        self.assertEqual(response.status_code, 200)
        self.assertIn('"__unicode__":"Foo"', force_text(response.content))

    def test_pagination(self):
        request = self.factory.get(reverse('admin2:blog_post_api_list'))
        request.user = self.user
        model_admin = self.get_model_admin(Post)
        view = apiviews.ListCreateAPIView.as_view(
            **model_admin.get_api_list_kwargs())
        response = view(request)
        response.render()

        self.assertEqual(response.status_code, 200)
        data = json.loads(force_text(response.content))
        self.assertEqual(data['count'], 0)
        # next and previous fields exist, but are null because we have no
        # content
        self.assertTrue('next' in data)
        self.assertEqual(data['next'], None)
        self.assertTrue('previous' in data)
        self.assertEqual(data['previous'], None)


class RetrieveUpdateDestroyAPIViewTest(APITestCase):
    def test_response_ok(self):
        post = Post.objects.create(title='Foo', body='Bar')
        request = self.factory.get(
            reverse('admin2:blog_post_api_detail',
                    kwargs={'pk': post.pk}))
        request.user = self.user
        model_admin = self.get_model_admin(Post)
        view = apiviews.RetrieveUpdateDestroyAPIView.as_view(
            **model_admin.get_api_detail_kwargs())
        response = view(request, pk=post.pk)
        self.assertEqual(response.status_code, 200)

    def test_view_permission(self):
        post = Post.objects.create(title='Foo', body='Bar')
        request = self.factory.get(
            reverse('admin2:blog_post_api_detail',
                    kwargs={'pk': post.pk}))
        request.user = AnonymousUser()
        model_admin = self.get_model_admin(Post)
        view = apiviews.RetrieveUpdateDestroyAPIView.as_view(
            **model_admin.get_api_detail_kwargs())
        self.assertRaises(PermissionDenied, view, request, pk=post.pk)
