Release beta 1.0.5 primogem economy

tests/test_release_update.py

@@ -20,7 +20,7 @@ SPEC.loader.exec_module(MODULE)
 
 
 class ReleaseUpdateTests(unittest.TestCase):
-    release = "beta 1.0.4.1016"
+    release = "beta 1.0.5.1016"
 
     def setUp(self):
         self.temporary = tempfile.TemporaryDirectory()
@@ -111,9 +111,16 @@ class ReleaseUpdateTests(unittest.TestCase):
         self.assertEqual(summary, "第一行更新摘要， 同一条摘要可以换行。")
         announcement = MODULE.build_announcement(self.release, summary)
         self.assertIn(f"{self.release} 更新现已完成", announcement)
-        self.assertIn("更新补偿：摩拉 x200", announcement)
+        self.assertIn("更新补偿：摩拉 x200，原石 x1", announcement)
         self.assertNotIn("第二条", announcement)
 
+    def test_patch_release_without_policy_does_not_announce_primogem(self):
+        release = "beta 1.0.6.1016"
+        self.assertEqual(MODULE.primogem_compensation_amount(release), 0)
+        announcement = MODULE.build_announcement(release, "普通小版本更新。")
+        self.assertIn("更新补偿：摩拉 x200", announcement)
+        self.assertNotIn("原石 x1", announcement)
+
     def test_release_update_grants_once_and_announces_once(self):
         delivered = []
 
@@ -148,6 +155,7 @@ class ReleaseUpdateTests(unittest.TestCase):
         self.assertIn("更新摩拉账户并修复重复回复问题", delivered[0][2])
         self.assertEqual([state["balance"] for state in self.account_states()], [1200, 250])
         for state in self.account_states():
+            self.assertEqual(state["primogems"], 1)
             matches = [
                 event
                 for event in state["coin_events"]
@@ -156,6 +164,17 @@ class ReleaseUpdateTests(unittest.TestCase):
             ]
             self.assertEqual(len(matches), 1)
             self.assertEqual(matches[0]["created_at"], applied_at.timestamp())
+            primogem_matches = [
+                event
+                for event in state["primogem_events"]
+                if event["delta"] == 1
+                and event["reason"] == f"{self.release} 版本更新补偿"
+            ]
+            self.assertEqual(len(primogem_matches), 1)
+            self.assertEqual(
+                primogem_matches[0]["created_at"],
+                applied_at.timestamp(),
+            )
         self.assertEqual(self.announcement_marker()["state"], "delivered")
         self.assertEqual(len(list(self.backups.glob("*.sqlite3"))), 1)
 
@@ -173,6 +192,7 @@ class ReleaseUpdateTests(unittest.TestCase):
                 deliverer=uncertain,
             )
         self.assertEqual([state["balance"] for state in self.account_states()], [1200, 250])
+        self.assertEqual([state["primogems"] for state in self.account_states()], [1, 1])
         self.assertEqual(self.announcement_marker()["state"], "uncertain")
 
         called = []
@@ -203,6 +223,7 @@ class ReleaseUpdateTests(unittest.TestCase):
         self.assertEqual(result["compensation"], "already_applied")
         self.assertEqual(len(called), 1)
         self.assertEqual([state["balance"] for state in self.account_states()], [1200, 250])
+        self.assertEqual([state["primogems"] for state in self.account_states()], [1, 1])
         self.assertEqual(self.announcement_marker()["state"], "delivered")