import json

from django.core.urlresolvers import reverse
from django.test import RequestFactory, TestCase

from data_explorer_example import data_explorer
from data_explorer_example.models import Author


class EndpointTest(TestCase):

    def setUp(self):
        self.view = data_explorer.Endpoint.as_view()
        self.factory = RequestFactory()

    def test_no_filter(self):
        Author.objects.create(name='name', birth_date='2016-05-23')
        request = self.factory.post('myurl', {"widget_id": "author-table"})
        response = self.view(request)
        content = json.loads(response.content.decode())
        self.assertEqual(content['status'], 'OK')
        self.assertEqual(len(content['data']['aaData']), 1)
        row = content['data']['aaData'][0]
        self.assertEqual(row['0'], 'name')
        self.assertEqual(row['1'], '2016-05-23')

    def test_no_widget(self):
        request = self.factory.post('myurl',
                                    {"widget_id": "non-existing-table"})
        response = self.view(request)
        content = json.loads(response.content.decode())
        self.assertEqual(response.status_code, 404)
        self.assertEqual(content['status'], 'WIDGET_NOT_FOUND')

    def test_endpoint_forbidden(self):
        class ForbiddenTable(data_explorer.Table):
            name = "forbidden"

            def is_accessible(self, request):
                return False

        data_explorer.Endpoint.register(ForbiddenTable)

        response = self.client.post(reverse("example-endpoint"),
                                    data={"widget_id": "forbidden"})
        self.assertEqual(response.status_code, 403)

    def test_data_missing(self):
        request = self.factory.post('myurl', {})
        response = self.view(request)
        content = json.loads(response.content.decode())
        self.assertEqual(response.status_code, 404)
        self.assertEqual(content['status'], 'WIDGET_NOT_FOUND')

    def test_invalid_widget_params(self):
        request = self.factory.post('myurl', {"widget_params": "invalid-json"})
        response = self.view(request)
        content = json.loads(response.content.decode())
        self.assertEqual(response.status_code, 400)
        self.assertEqual(content['status'], 'INVALID_WIDGET_PARAMS')

    def test_invalid_client_params(self):
        request = self.factory.post('myurl', {"widget_id": "author-table",
                                              "client_params": "invalid-json"})
        response = self.view(request)
        content = json.loads(response.content.decode())
        self.assertEqual(response.status_code, 400)
        self.assertEqual(content['status'], 'INVALID_CLIENT_PARAMS')