From ab37264ae14c0bfc8579f0c219b9b48fedf45ed0 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@digitalocean.com>
Date: Thu, 2 Aug 2018 14:23:30 -0400
Subject: [PATCH] Added tests for assigning tags during POST/PATCH

---
 netbox/extras/tests/test_tags.py | 60 ++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 netbox/extras/tests/test_tags.py

diff --git a/netbox/extras/tests/test_tags.py b/netbox/extras/tests/test_tags.py
new file mode 100644
index 000000000..d80184f16
--- /dev/null
+++ b/netbox/extras/tests/test_tags.py
@@ -0,0 +1,60 @@
+from __future__ import unicode_literals
+
+from django.contrib.auth.models import User
+from django.urls import reverse
+from rest_framework import status
+from rest_framework.test import APITestCase
+
+from dcim.models import Site
+from users.models import Token
+from utilities.testing import HttpStatusMixin
+
+
+class TaggedItemTest(HttpStatusMixin, APITestCase):
+    """
+    Test the application of Tags to and item (a Site, for example) upon creation (POST) and modification (PATCH).
+    """
+
+    def setUp(self):
+
+        user = User.objects.create(username='testuser', is_superuser=True)
+        token = Token.objects.create(user=user)
+        self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)}
+
+    def test_create_tagged_item(self):
+
+        data = {
+            'name': 'Test Site',
+            'slug': 'test-site',
+            'tags': ['Foo', 'Bar', 'Baz']
+        }
+
+        url = reverse('dcim-api:site-list')
+        response = self.client.post(url, data, format='json', **self.header)
+
+        self.assertHttpStatus(response, status.HTTP_201_CREATED)
+        self.assertEqual(sorted(response.data['tags']), sorted(data['tags']))
+        site = Site.objects.get(pk=response.data['id'])
+        tags = [tag.name for tag in site.tags.all()]
+        self.assertEqual(sorted(tags), sorted(data['tags']))
+
+    def test_update_tagged_item(self):
+
+        site = Site.objects.create(
+            name='Test Site',
+            slug='test-site',
+            tags=['Foo', 'Bar', 'Baz']
+        )
+
+        data = {
+            'tags': ['Foo', 'Bar', 'New Tag']
+        }
+
+        url = reverse('dcim-api:site-detail', kwargs={'pk': site.pk})
+        response = self.client.patch(url, data, format='json', **self.header)
+
+        self.assertHttpStatus(response, status.HTTP_200_OK)
+        self.assertEqual(sorted(response.data['tags']), sorted(data['tags']))
+        site = Site.objects.get(pk=response.data['id'])
+        tags = [tag.name for tag in site.tags.all()]
+        self.assertEqual(sorted(tags), sorted(data['tags']))