from django.test import TestCase, Client
from . import *
import logging
import json

logger = logging.getLogger(name='django_kepi')

def _response_to_dict(response):

    result = json.loads(response.content.decode('utf-8'))

    # @context is often huge, and is irrelevant for testing here
    if '@context' in result:
        del result['@context']

    logger.info('Response: %s', result)

    return result

class TestKepiView(TestCase):

    def setUp(self):
        settings.KEPI['LOCAL_OBJECT_HOSTNAME'] = 'testserver'
        self.maxDiff = None

    def test_single_kepi_view(self):

        alice = create_local_person('alice')

        c = Client()
        response = c.get('/users/alice')
        result = _response_to_dict(response)

        self.assertDictContainsSubset(
                {
                    'name': 'alice',
                    'id': 'https://testserver/users/alice',
                    'type': 'Person',
                    'preferredUsername': 'alice',
                    },
                result,
                )

    def test_multiple_kepi_view(self):

        alice = create_local_person('alice')
        bob = create_local_person('bob')

        c = Client()
        response = c.get('/users')
        result = _response_to_dict(response)

        self.assertDictContainsSubset(
                {
                    "first": "http://testserver/users?page=1",
                    "id": "http://testserver/users",
                    "totalItems": 2,
                    "type": "OrderedCollection"
                    },
                result,
                )

        response = c.get('/users?page=1')
        result = _response_to_dict(response)

        self.assertDictContainsSubset(
                {
                    'id': 'http://testserver/users?page=1',
                    'partOf': 'http://testserver/users',
                    'totalItems': 2,
                    'type': 'OrderedCollectionPage',
                    },
                result,
                )

        self.assertEqual(
                len(result['orderedItems']),
                2)

        for item, name in zip(
                result['orderedItems'],
                ['alice', 'bob']):
            self.assertDictContainsSubset(
                    {
                        'id': 'https://testserver/users/'+name,
                        'name': name,
                        'type': 'Person',
                        },
                    item,
                    )

class TestTombstone(TestCase):

    def setUp(self):
        settings.KEPI['LOCAL_OBJECT_HOSTNAME'] = 'testserver'
        self.maxDiff = None

    def test_tombstone(self):

        queen_anne = create_local_person('queen_anne')

        c = Client()
        response = c.get('/users/queen_anne')

        self.assertEqual(response.status_code, 200)
        self.assertDictContainsSubset(
                {
                    'name': 'queen_anne',
                    'id': 'https://testserver/users/queen_anne',
                    'type': 'Person',
                    },
                _response_to_dict(response),
                )

        queen_anne.entomb()

        response = c.get('/users/queen_anne')

        self.assertEqual(response.status_code, 410)
        self.assertDictContainsSubset(
                {
                    'id': 'https://testserver/users/queen_anne',
                    'type': 'Tombstone',
                    'former_type': 'Person',
                    'name': 'queen_anne',
                    },
                _response_to_dict(response),
                )