Refresh ops center in realtime

2026-04-07 16:04:18 +08:00
parent a42e5b75dc
commit 0b0bc5152f
7 changed files with 320 additions and 5 deletions
--- a/android/app/src/test/java/com/hyzq/boss/OpsCenterActivityTest.java
+++ b/android/app/src/test/java/com/hyzq/boss/OpsCenterActivityTest.java
@@ -0,0 +1,101 @@
+package com.hyzq.boss;
+
+import static org.junit.Assert.assertEquals;
+
+import android.content.Intent;
+
+import org.json.JSONObject;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.robolectric.Robolectric;
+import org.robolectric.RobolectricTestRunner;
+import org.robolectric.Shadows;
+import org.robolectric.annotation.Config;
+import org.robolectric.util.ReflectionHelpers;
+
+@RunWith(RobolectricTestRunner.class)
+@Config(sdk = 34)
+public class OpsCenterActivityTest {
+    @Test
+    public void appLogsUpdatedEventTriggersReload() throws Exception {
+        TestOpsCenterActivity activity = Robolectric
+                .buildActivity(TestOpsCenterActivity.class, new Intent())
+                .setup()
+                .resume()
+                .get();
+        activity.reloadEnabled = true;
+        activity.reloadCount = 0;
+
+        ReflectionHelpers.callInstanceMethod(
+                activity,
+                "handleRealtimeEvent",
+                ReflectionHelpers.ClassParameter.from(
+                        BossRealtimeEvent.class,
+                        new BossRealtimeEvent("app.logs.updated", new JSONObject().put("deviceId", "mac-studio"))
+                )
+        );
+        Shadows.shadowOf(activity.getMainLooper()).idle();
+
+        assertEquals(1, activity.reloadCount);
+    }
+
+    @Test
+    public void projectContextRiskUpdatedEventTriggersReload() throws Exception {
+        TestOpsCenterActivity activity = Robolectric
+                .buildActivity(TestOpsCenterActivity.class, new Intent())
+                .setup()
+                .resume()
+                .get();
+        activity.reloadEnabled = true;
+        activity.reloadCount = 0;
+
+        ReflectionHelpers.callInstanceMethod(
+                activity,
+                "handleRealtimeEvent",
+                ReflectionHelpers.ClassParameter.from(
+                        BossRealtimeEvent.class,
+                        new BossRealtimeEvent("project.context_risk.updated", new JSONObject().put("status", "verified"))
+                )
+        );
+        Shadows.shadowOf(activity.getMainLooper()).idle();
+
+        assertEquals(1, activity.reloadCount);
+    }
+
+    @Test
+    public void unrelatedConversationEventDoesNotTriggerReload() throws Exception {
+        TestOpsCenterActivity activity = Robolectric
+                .buildActivity(TestOpsCenterActivity.class, new Intent())
+                .setup()
+                .resume()
+                .get();
+        activity.reloadEnabled = true;
+        activity.reloadCount = 0;
+
+        ReflectionHelpers.callInstanceMethod(
+                activity,
+                "handleRealtimeEvent",
+                ReflectionHelpers.ClassParameter.from(
+                        BossRealtimeEvent.class,
+                        new BossRealtimeEvent("conversation.updated", new JSONObject().put("projectId", "project-1"))
+                )
+        );
+        Shadows.shadowOf(activity.getMainLooper()).idle();
+
+        assertEquals(0, activity.reloadCount);
+    }
+
+    public static class TestOpsCenterActivity extends OpsCenterActivity {
+        private boolean reloadEnabled;
+        private int reloadCount;
+
+        @Override
+        protected void reload() {
+            if (!reloadEnabled) {
+                return;
+            }
+            reloadCount += 1;
+            setRefreshing(false);
+        }
+    }
+}