From 59ecff87011c2b1209bc32b4ef4d33455827e321 Mon Sep 17 00:00:00 2001
From: jonaswinkler <17569239+jonaswinkler@users.noreply.github.com>
Date: Thu, 4 Mar 2021 00:16:24 +0100
Subject: [PATCH] test case for the migration

---
 .../tests/test_migration_tag_colors.py        | 37 +++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 src/documents/tests/test_migration_tag_colors.py

diff --git a/src/documents/tests/test_migration_tag_colors.py b/src/documents/tests/test_migration_tag_colors.py
new file mode 100644
index 000000000..2c4b35925
--- /dev/null
+++ b/src/documents/tests/test_migration_tag_colors.py
@@ -0,0 +1,37 @@
+from documents.tests.utils import DirectoriesMixin, TestMigrations
+
+
+class TestMigrateTagColor(DirectoriesMixin, TestMigrations):
+
+    migrate_from = '1012_fix_archive_files'
+    migrate_to = '1013_migrate_tag_colour'
+
+    def setUpBeforeMigration(self, apps):
+        Tag = apps.get_model("documents", "Tag")
+        self.t1_id = Tag.objects.create(name="tag1").id
+        self.t2_id = Tag.objects.create(name="tag2", colour=1).id
+        self.t3_id = Tag.objects.create(name="tag3", colour=5).id
+
+    def testMimeTypesMigrated(self):
+        Tag = self.apps.get_model('documents', 'Tag')
+        self.assertEqual(Tag.objects.get(id=self.t1_id).color, "#a6cee3")
+        self.assertEqual(Tag.objects.get(id=self.t2_id).color, "#a6cee3")
+        self.assertEqual(Tag.objects.get(id=self.t3_id).color, "#fb9a99")
+
+
+class TestMigrateTagColorBackwards(DirectoriesMixin, TestMigrations):
+
+    migrate_from = '1013_migrate_tag_colour'
+    migrate_to = '1012_fix_archive_files'
+
+    def setUpBeforeMigration(self, apps):
+        Tag = apps.get_model("documents", "Tag")
+        self.t1_id = Tag.objects.create(name="tag1").id
+        self.t2_id = Tag.objects.create(name="tag2", color="#cab2d6").id
+        self.t3_id = Tag.objects.create(name="tag3", color="#123456").id
+
+    def testMimeTypesReverted(self):
+        Tag = self.apps.get_model('documents', 'Tag')
+        self.assertEqual(Tag.objects.get(id=self.t1_id).colour, 1)
+        self.assertEqual(Tag.objects.get(id=self.t2_id).colour, 9)
+        self.assertEqual(Tag.objects.get(id=self.t3_id).colour, 1)