from django.test import TestCase
from django.core import mail
from django.conf import settings

from sendgrid.utils import SendgridEmailMessage, SendgridEmailMultiAlternatives
from sendgrid.models import Email

import json
from copy import copy


class UtilTestCase(TestCase):
    def setUp(self):
        self.email_data = {'subject': 'Test Subject',
                           'body': 'Hi, I am a test body',
                           'from_email': 'email@example.com',
                           'to': ('other_email@example.com', )}

    def test_django_email(self):
        """ Test if we break default django email sending.
        """
        message = mail.EmailMessage(**self.email_data)
        message.send()
        # this email should not have a UUID
        self.assertEqual(len(mail.outbox), 1)
        self.assertEqual(mail.outbox[0].subject, self.email_data['subject'])
        self.assertNotIn('X-SMTPAPI', mail.outbox[0].extra_headers)

    def test_django_multipart(self):
        """ Test if we break default django email sending.
        """
        message = mail.EmailMultiAlternatives(**self.email_data)
        message.send()
        # this email should not have a UUID
        self.assertEqual(len(mail.outbox), 1)
        self.assertEqual(mail.outbox[0].subject, self.email_data['subject'])
        self.assertNotIn('X-SMTPAPI', mail.outbox[0].extra_headers)

    def test_sendgrid_email(self):
        """ Test if we can send emails using our own class.
        """
        message = SendgridEmailMessage(**self.email_data)
        message.send()
        # this email should have a UUID
        self.assertEqual(len(mail.outbox), 1)
        self.assertEqual(mail.outbox[0].subject, self.email_data['subject'])
        self.assertIn('X-SMTPAPI', mail.outbox[0].extra_headers)
        content = json.loads(mail.outbox[0].extra_headers['X-SMTPAPI'])
        self.assertIn('uuid', content['unique_args'])
        self.assertEqual(len(content['unique_args']['uuid']), 36)

    def test_sendgrid_multipart(self):
        """ Test if we can send emails using our own class.
        """
        message = SendgridEmailMultiAlternatives(**self.email_data)
        message.send()
        # this email should have a UUID
        self.assertEqual(len(mail.outbox), 1)
        self.assertEqual(mail.outbox[0].subject, self.email_data['subject'])
        self.assertIn('X-SMTPAPI', mail.outbox[0].extra_headers)
        content = json.loads(mail.outbox[0].extra_headers['X-SMTPAPI'])
        self.assertIn('uuid', content['unique_args'])
        self.assertEqual(len(content['unique_args']['uuid']), 36)

    def test_sendgrid_email_object_attachment(self):
        """ Test attaching a related object to the email.
        """
        # first send a message without an object
        message = SendgridEmailMessage(**self.email_data)
        message.send()
        mail_event_1 = Email.objects.get(uuid=message.uuid)
        self.assertEqual(mail_event_1.content_object, None)

        # now attach an object
        data = dict(self.email_data)
        data['obj'] = mail_event_1  # code re-use :)
        message = SendgridEmailMessage(**data)
        message.send()
        mail_event = Email.objects.get(uuid=message.uuid)
        self.assertEqual(mail_event.content_object, mail_event_1)

    def test_sendgrid_multipart_object_attachment(self):
        """ Test attaching a related object to the email.
        """
        # first send a message without an object
        message = SendgridEmailMultiAlternatives(**self.email_data)
        message.send()
        mail_event_1 = Email.objects.get(uuid=message.uuid)
        self.assertEqual(mail_event_1.content_object, None)

        # now attach an object
        data = dict(self.email_data)
        data['obj'] = mail_event_1  # code re-use :)
        message = SendgridEmailMultiAlternatives(**data)
        message.send()
        mail_event = Email.objects.get(uuid=message.uuid)
        self.assertEqual(mail_event.content_object, mail_event_1)

    def test_email_headers(self):
        """ Test if we handle custom headers correctly.
        """
        # Add a custom extra_header field
        mail_data = copy(self.email_data)
        mail_data.update({'headers': {'X-AWESOME': 42, }})

        message = SendgridEmailMessage(**mail_data)
        message.send()
        # this email should have a UUID
        self.assertEqual(len(mail.outbox), 1)
        self.assertIn('X-SMTPAPI', mail.outbox[0].extra_headers)

        content = json.loads(mail.outbox[0].extra_headers['X-SMTPAPI'])
        self.assertIn('uuid', content['unique_args'])
        self.assertEqual(len(content['unique_args']['uuid']), 36)

        self.assertEqual(mail.outbox[0].extra_headers['X-AWESOME'], 42)

    def test_smtpapi_headers(self):
        """ Test if we handle existing SMTPAPI headers correctly.
        """
        # Add a custom extra_header field
        mail_data = copy(self.email_data)
        mail_data.update({'headers': {
            'X-AWESOME': 42,
            'X-SMTPAPI': json.dumps({'amazing': 'stuff'})}})

        message = SendgridEmailMessage(**mail_data)
        message.send()
        # this email should have a UUID
        self.assertEqual(len(mail.outbox), 1)
        self.assertIn('X-SMTPAPI', mail.outbox[0].extra_headers)

        content = json.loads(mail.outbox[0].extra_headers['X-SMTPAPI'])
        self.assertIn('uuid', content['unique_args'])
        self.assertEqual(len(content['unique_args']['uuid']), 36)
        self.assertEqual(content['amazing'], 'stuff')
        self.assertEqual(mail.outbox[0].extra_headers['X-AWESOME'], 42)


class UtilTZTestCase(UtilTestCase):
    """ Test explicitly with USE_TZ enabled
    """
    def setUp(self):
        self._old_use_tz = settings.USE_TZ
        settings.USE_TZ = True
        super(UtilTZTestCase, self).setUp()

    def tearDown(self):
        settings.USE_TZ = self._old_use_tz


class UtilNoTZTestCase(UtilTestCase):
    """ Test explicitly with USE_TZ disabled
    """
    def setUp(self):
        self._old_use_tz = settings.USE_TZ
        settings.USE_TZ = False
        super(UtilNoTZTestCase, self).setUp()

    def tearDown(self):
        settings.USE_TZ = self._old_use_tz