import datetime

from django.core.urlresolvers import reverse
from django.test import TestCase
from service.models import Service

from ..models import Conference, Speaker
from city.models import City
from country.models import Country
from events.models import TargetAudience


class ConferenceTest(TestCase):
    def setUp(self):
        audience = TargetAudience.objects.create(title='All')

        self.conference = Conference.objects.create(
            name='New Conference',
            data_begin=datetime.datetime.now(),
            data_end=datetime.timedelta(24)+datetime.datetime.now(),
            country=Country.objects.all()[0],
            city=City.objects.all()[0],

            url='new-conference',
            services=b'0001000'
        )
        self.conference.audience.add(audience)

    def test_a_conference_create(self):
        conference_count = Conference.objects.all().count()
        Conference.objects.create(
            name='New Conference',
            data_begin=datetime.datetime.now(),
            data_end=datetime.timedelta(24)+datetime.datetime.now(),
            country=Country.objects.all()[0],
            city=City.objects.all()[0]
        )
        new_conference_count = Conference.objects.all().count()

        self.assertEqual(conference_count + 1, new_conference_count)

    def test_save_conference_info(self):
        name = 'New Conference'
        datetime_now = datetime.datetime.now()
        data_begin = datetime_now
        data_end = datetime.timedelta(24) + datetime_now
        country = Country.objects.all()[0]
        city = City.objects.all()[0]

        new_conference = Conference.objects.create(
            name=name,
            data_begin=data_begin,
            data_end=data_end,
            country=country,
            city=city

        )

        self.assertEqual(name, new_conference.name)
        self.assertEqual(data_begin, new_conference.data_begin)
        self.assertEqual(data_end, new_conference.data_end)
        self.assertEqual(country, new_conference.country)
        self.assertEqual(city, new_conference.city)

    def test_method_get_services(self):
        pass

    def test_method_get_news_url(self):
        self.assertEqual(
            self.conference.get_news_url(),
            reverse('news_conference', kwargs={'slug': self.conference.url})
        )

    def test_method_get_audience(self):
        audience = TargetAudience.objects.all()[0]
        self.assertIn(audience, self.conference.get_audience())

    def test_method_get_services(self):
        pass

    def test_method_get_calendar_url(self):
        self.assertEqual(
            self.conference.get_calendar_url(),
            reverse('conference_add_calendar', args=[self.conference.id])
        )

    def test_method_get_visit_url(self):
        self.assertEqual(
            self.conference.get_visit_url(),
            reverse(
                'events:event_visit',
                args=['conference', self.conference.id]
            )
        )

    def test_method_get_note_url(self):
        self.assertEqual(
            self.conference.get_note_url(),
            reverse('conference_add_note', args=[self.conference.url])
        )

class SpeakerTest(TestCase):
    pass