Rebuild SCUM plugin-owned data flow

2026-08-18 07:01:17 +08:00
parent 302f1f64b7
commit 98bf944f4c
39 changed files with 1832 additions and 223 deletions
@@ -167,6 +167,92 @@ func TestDeclaredSQLiteQueryProjectionRejectsInvalidBatchAtomically(t *testing.T
 	}
 }

+func TestDeclaredSQLiteQueryProjectionMergesPresenceAndReplacesCompleteSnapshots(t *testing.T) {
+	svc, plugin, _, session, instance := createSQLiteQueryBridgeFixture(t)
+	template := &plugin.GameClientBridge.QueryTemplates[0]
+	template.RowTarget = &domain.PluginDataRowTargetDeclaration{
+		Collection: "users", UpsertKeys: []string{"steamId"}, WriteMode: domain.PluginDataRowWriteModeMerge,
+		ColumnMappings: map[string]string{"steamId": "steam_id", "displayName": "display_name", "x": "x"},
+	}
+	if err := svc.store.GamePlugins().Update(plugin); err != nil {
+		t.Fatalf("update merge target: %v", err)
+	}
+	if _, err := svc.PutPluginDataForSession(session, domain.PluginDataRecord{PluginID: plugin.ID, ServerInstanceID: instance.ID, Collection: "users", Key: "steam-1", Value: map[string]any{"steamId": "steam-1", "online": true, "lastLoginAt": "2026-07-03T12:00:00Z"}}); err != nil {
+		t.Fatalf("seed stdout user: %v", err)
+	}
+	job := domain.Job{ServerInstanceID: instance.ID, Capability: domain.JobCapabilityRemoteRunDBSQLiteQuery, State: domain.JobStateSucceeded, ExecutionInput: domain.JobExecutionInput{Inputs: map[string]string{"templateKey": template.Key}}, ExecutionResult: domain.JobExecutionResult{Content: `{"rows":[{"steam_id":"steam-1","display_name":"Ada","x":12.5}]}`}}
+	if err := svc.projectPluginDataJobResult(job); err != nil {
+		t.Fatalf("merge query projection: %v", err)
+	}
+	users, err := svc.ListPluginDataForSession(session, domain.PluginDataFilter{PluginID: plugin.ID, ServerInstanceID: instance.ID, Collection: "users"})
+	if err != nil || len(users) != 1 || users[0].Value["online"] != true || users[0].Value["displayName"] != "Ada" {
+		t.Fatalf("merged users=%+v err=%v", users, err)
+	}
+
+	plugin.GameClientBridge.QueryTemplates[0].RowTarget = &domain.PluginDataRowTargetDeclaration{
+		Collection: "vehicles", UpsertKeys: []string{"vehicleId"}, WriteMode: domain.PluginDataRowWriteModeReplace,
+		ColumnMappings: map[string]string{"vehicleId": "vehicle_id", "x": "x"},
+	}
+	if err := svc.store.GamePlugins().Update(plugin); err != nil {
+		t.Fatalf("update replace target: %v", err)
+	}
+	for _, id := range []string{"keep", "gone"} {
+		if _, err := svc.PutPluginDataForSession(session, domain.PluginDataRecord{PluginID: plugin.ID, ServerInstanceID: instance.ID, Collection: "vehicles", Key: id, Value: map[string]any{"vehicleId": id}}); err != nil {
+			t.Fatalf("seed vehicle %s: %v", id, err)
+		}
+	}
+	job.ExecutionResult.Content = `{"rows":[{"vehicle_id":"keep","x":7}]}`
+	if err := svc.projectPluginDataJobResult(job); err != nil {
+		t.Fatalf("replace query projection: %v", err)
+	}
+	vehicles, err := svc.ListPluginDataForSession(session, domain.PluginDataFilter{PluginID: plugin.ID, ServerInstanceID: instance.ID, Collection: "vehicles"})
+	if err != nil || len(vehicles) != 1 || vehicles[0].Key != "keep" {
+		t.Fatalf("replaced vehicles=%+v err=%v", vehicles, err)
+	}
+	job.ExecutionResult.Content = `{"rows":[]}`
+	if err := svc.projectPluginDataJobResult(job); err != nil {
+		t.Fatalf("empty replace query projection: %v", err)
+	}
+	vehicles, err = svc.ListPluginDataForSession(session, domain.PluginDataFilter{PluginID: plugin.ID, ServerInstanceID: instance.ID, Collection: "vehicles"})
+	if err != nil || len(vehicles) != 0 {
+		t.Fatalf("empty replace did not clear vehicles=%+v err=%v", vehicles, err)
+	}
+}
+
+func TestRunPollSchedulesDueDeclaredPluginQueryWithoutBrowserSession(t *testing.T) {
+	svc, plugin, endpoint, _, instance := createSQLiteQueryBridgeFixture(t)
+	plugin.GameClientBridge.QueryTemplates[0].PollIntervalSeconds = 3
+	if err := svc.store.GamePlugins().Update(plugin); err != nil {
+		t.Fatalf("enable automatic query: %v", err)
+	}
+	helloRequest := validRunControlHello()
+	helloRequest.CapabilityReport.Capabilities = append(helloRequest.CapabilityReport.Capabilities, domain.JobCapabilityRemoteRunDBSQLiteQuery)
+	helloRequest.CapabilityReport.Fingerprint = "cap-plugin-query-scheduler"
+	hello, err := svc.RegisterRunHello(helloRequest)
+	if err != nil {
+		t.Fatalf("register Run: %v", err)
+	}
+	claim, err := svc.ClaimRunJob(domain.RunJobClaim{RunEndpointID: endpoint.ID, SessionToken: hello.SessionToken, Capabilities: []string{domain.JobCapabilityRemoteRunDBSQLiteQuery}, Capacity: domain.RunCapacity{MaxJobs: 1}})
+	if err != nil || !claim.HasJob || claim.Job == nil {
+		t.Fatalf("automatic query claim=%+v err=%v", claim, err)
+	}
+	if claim.Job.ServerInstanceID != instance.ID || claim.Job.Capability != domain.JobCapabilityRemoteRunDBSQLiteQuery || claim.Job.ExecutionInput.Inputs["templateKey"] != "players.by-id" || claim.Job.ExecutionInput.Inputs["limit"] != "25" {
+		t.Fatalf("unexpected automatic query assignment: %+v", claim.Job)
+	}
+	jobs, err := svc.store.Jobs().List(domain.JobFilter{RunEndpointID: endpoint.ID})
+	if err != nil || len(jobs) != 1 {
+		t.Fatalf("automatic query jobs=%+v err=%v", jobs, err)
+	}
+	second, err := svc.ClaimRunJob(domain.RunJobClaim{RunEndpointID: endpoint.ID, SessionToken: hello.SessionToken, Capabilities: []string{domain.JobCapabilityRemoteRunDBSQLiteQuery}, Capacity: domain.RunCapacity{MaxJobs: 1, RunningJobs: 1}})
+	if err != nil || second.HasJob {
+		t.Fatalf("overlapping automatic query was not suppressed: %+v err=%v", second, err)
+	}
+	jobs, err = svc.store.Jobs().List(domain.JobFilter{RunEndpointID: endpoint.ID})
+	if err != nil || len(jobs) != 1 {
+		t.Fatalf("overlap created duplicate jobs=%+v err=%v", jobs, err)
+	}
+}
+
 func TestDeclaredSQLiteQueryProjectionFailureKeepsJobRetryable(t *testing.T) {
 	svc, plugin, endpoint, session, instance := createSQLiteQueryBridgeFixture(t)
 	plugin.GameClientBridge.QueryTemplates[0].RowTarget = &domain.PluginDataRowTargetDeclaration{