from django.test import TestCase

from dcim.choices import SiteStatusChoices
from dcim.models import Site
from extras.choices import CustomFieldChoiceSetBaseChoices
from extras.models import CustomFieldChoiceSet, Tag
from utilities.serialization import deserialize_object, serialize_object


class SerializationTestCase(TestCase):

    @classmethod
    def setUpTestData(cls):
        tags = (
            Tag(name='Tag 1', slug='tag-1'),
            Tag(name='Tag 2', slug='tag-2'),
            Tag(name='Tag 3', slug='tag-3'),
        )
        Tag.objects.bulk_create(tags)

    def test_serialize_object(self):
        site = Site.objects.create(
            name='Site 1',
            slug='site=1',
            description='Ignore me',
        )
        site.tags.set(Tag.objects.all())

        data = serialize_object(site, extra={'foo': 123}, exclude=['description'])
        self.assertEqual(data['name'], site.name)
        self.assertEqual(data['slug'], site.slug)
        self.assertEqual(data['tags'], [tag.name for tag in Tag.objects.all()])
        self.assertEqual(data['foo'], 123)
        self.assertNotIn('description', data)

    def test_serialize_object_empty_array_field_subclass(self):
        """An empty ArrayField subclass value serializes as a list, not a string."""
        choice_set = CustomFieldChoiceSet.objects.create(
            name='Choice Set 1',
            base_choices=CustomFieldChoiceSetBaseChoices.IATA,
            extra_choices=[],
        )
        data = serialize_object(choice_set)
        self.assertEqual(data['extra_choices'], [])

    def test_deserialize_object(self):
        data = {
            'name': 'Site 1',
            'slug': 'site-1',
            'tags': ['Tag 1', 'Tag 2', 'Tag 3'],
            'foo': 123,
        }

        instance = deserialize_object(Site, data, pk=123)
        self.assertEqual(instance.object.pk, 123)
        self.assertEqual(instance.object.name, data['name'])
        self.assertEqual(instance.object.slug, data['slug'])
        self.assertEqual(instance.object.status, SiteStatusChoices.STATUS_ACTIVE)  # Default field value
        self.assertEqual(instance.object.foo, data['foo'])  # Non-field attribute
        self.assertEqual(list(instance.m2m_data['tags']), list(Tag.objects.all()))