Improve trigger tests.

Author: dnwpark

tests/test_model_sync.py

@@ -5184,6 +5184,18 @@ class TestModelSyncTrigger(tb.ModelTestCase):
                 }
             );
         };
+        type TriggerInsertComputed {
+            n: int64;
+            trigger update_target after insert
+            for each do (
+                update TestInsert
+                set {
+                    val := .val + __new__.n
+                }
+            );
+
+            test_vals := TestInsert.val;
+        };
         type TriggerInsertWithLink {
             n: int64;
             test: TestInsert;
@@ -5209,6 +5221,18 @@ class TestModelSyncTrigger(tb.ModelTestCase):
                 }
             );
         };
+        type TriggerUpdateComputed {
+            n: int64;
+            trigger update_target after update
+            for each do (
+                update TestUpdate
+                set {
+                    val := .val + __new__.n
+                }
+            );
+
+            test_vals := TestUpdate.val;
+        };
     """
 
     def test_model_sync_trigger_insert_01(self):
@@ -5231,6 +5255,19 @@ def test_model_sync_trigger_insert_01(self):
         self.assertEqual(test_obj.val, 6)
 
     def test_model_sync_trigger_insert_02(self):
+        # Insert trigger, computed modified by trigger
+        from models.TestModelSyncTrigger import default
+
+        test_obj = default.TestInsert(val=0)
+        trigger = default.TriggerInsertComputed(n=1)
+        self.client.sync(test_obj, trigger)
+        self.assertEqual(trigger.test_vals, (1,))
+
+        # Also check that the newly inserted object refetched its
+        # trigger modified value
+        self.assertEqual(test_obj.val, 1)
+
+    def test_model_sync_trigger_insert_03(self):
         # Insert trigger
         # Link will cause test objs to be batched before trigger objs
         from models.TestModelSyncTrigger import default
@@ -5268,3 +5305,15 @@ def test_model_sync_trigger_update_01(self):
         trigger.n = 3
         self.client.sync(test_obj, trigger)
         self.assertEqual(test_obj.val, 6)
+
+    def test_model_sync_trigger_update_02(self):
+        # Update trigger, computed modified by trigger
+        from models.TestModelSyncTrigger import default
+
+        test_obj = default.TestUpdate(val=0)
+        trigger = default.TriggerUpdateComputed(n=0)
+        self.client.sync(test_obj, trigger)
+
+        trigger.n = 1
+        self.client.sync(test_obj, trigger)
+        self.assertEqual(trigger.test_vals, (1,))