recognition: Recognition model w. grant_if_new; sky_save returns stargazer on first chart save — TDD

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

src/apps/drama/tests/integrated/test_models.py

@@ -1,7 +1,8 @@
 from django.test import TestCase
 from django.db import IntegrityError
+from django.utils import timezone
 
-from apps.drama.models import GameEvent, ScrollPosition, record
+from apps.drama.models import GameEvent, Recognition, ScrollPosition, record
 from apps.epic.models import Room
 from apps.lyric.models import User
 
@@ -173,3 +174,69 @@ class ScrollPositionModelTest(TestCase):
             defaults={"position": 200},
         )
         self.assertEqual(ScrollPosition.objects.get(user=self.user, room=self.room).position, 200)
+
+
+class RecognitionModelTest(TestCase):
+    def setUp(self):
+        self.user = User.objects.create(email="earner@test.io")
+
+    def test_can_create_recognition(self):
+        recog = Recognition.objects.create(
+            user=self.user, slug="stargazer", earned_at=timezone.now(),
+        )
+        self.assertEqual(Recognition.objects.count(), 1)
+        self.assertEqual(recog.slug, "stargazer")
+        self.assertEqual(recog.user, self.user)
+
+    def test_palette_is_null_by_default(self):
+        recog = Recognition.objects.create(
+            user=self.user, slug="stargazer", earned_at=timezone.now(),
+        )
+        self.assertIsNone(recog.palette)
+
+    def test_palette_can_be_set(self):
+        recog = Recognition.objects.create(
+            user=self.user, slug="stargazer", earned_at=timezone.now(),
+            palette="palette-bardo",
+        )
+        self.assertEqual(recog.palette, "palette-bardo")
+
+    def test_unique_per_user_and_slug(self):
+        Recognition.objects.create(user=self.user, slug="stargazer", earned_at=timezone.now())
+        with self.assertRaises(IntegrityError):
+            Recognition.objects.create(user=self.user, slug="stargazer", earned_at=timezone.now())
+
+    def test_different_users_can_share_slug(self):
+        other = User.objects.create(email="other@test.io")
+        Recognition.objects.create(user=self.user, slug="stargazer", earned_at=timezone.now())
+        Recognition.objects.create(user=other, slug="stargazer", earned_at=timezone.now())
+        self.assertEqual(Recognition.objects.count(), 2)
+
+    def test_str_includes_slug_and_email(self):
+        recog = Recognition.objects.create(
+            user=self.user, slug="stargazer", earned_at=timezone.now(),
+        )
+        s = str(recog)
+        self.assertIn("stargazer", s)
+        self.assertIn("earner@test.io", s)
+
+    def test_grant_if_new_creates_on_first_call(self):
+        recog, created = Recognition.grant_if_new(self.user, "stargazer")
+        self.assertTrue(created)
+        self.assertEqual(recog.slug, "stargazer")
+        self.assertIsNotNone(recog.earned_at)
+
+    def test_grant_if_new_is_idempotent(self):
+        Recognition.grant_if_new(self.user, "stargazer")
+        recog, created = Recognition.grant_if_new(self.user, "stargazer")
+        self.assertFalse(created)
+        self.assertEqual(Recognition.objects.count(), 1)
+
+    def test_grant_if_new_does_not_overwrite_palette(self):
+        Recognition.objects.create(
+            user=self.user, slug="stargazer",
+            earned_at=timezone.now(), palette="palette-bardo",
+        )
+        recog, created = Recognition.grant_if_new(self.user, "stargazer")
+        self.assertFalse(created)
+        self.assertEqual(recog.palette, "palette-bardo")