mirror of
https://github.com/multica-ai/multica.git
synced 2026-07-05 21:39:54 +02:00
1ac3a03e5d
* MUL-3618: dispatch daemon feature flag snapshots Co-authored-by: multica-agent <github@multica.ai> * MUL-3618: narrow daemon flag snapshots to process scope Co-authored-by: multica-agent <github@multica.ai> --------- Co-authored-by: Eve <eve@multica-ai.local> Co-authored-by: multica-agent <github@multica.ai>
49 lines
1.6 KiB
Go
49 lines
1.6 KiB
Go
package handler
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/multica-ai/multica/server/internal/featureflagdispatch"
|
|
"github.com/multica-ai/multica/server/internal/util"
|
|
db "github.com/multica-ai/multica/server/pkg/db/generated"
|
|
"github.com/multica-ai/multica/server/pkg/featureflag"
|
|
)
|
|
|
|
func TestProcessHeartbeatIncludesDaemonFeatureFlagSnapshot(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
provider := featureflag.NewStaticProvider()
|
|
provider.Set(featureflagdispatch.RuntimeBriefSlimFlag, featureflag.Rule{Default: true})
|
|
h := &Handler{
|
|
UpdateStore: NewInMemoryUpdateStore(),
|
|
ModelListStore: NewInMemoryModelListStore(),
|
|
LocalSkillListStore: NewInMemoryLocalSkillListStore(),
|
|
LocalSkillImportStore: NewInMemoryLocalSkillImportStore(),
|
|
LivenessStore: NewNoopLivenessStore(),
|
|
HeartbeatScheduler: noopHeartbeatScheduler{},
|
|
DaemonFeatureFlags: featureflagdispatch.NewEvaluator(featureflag.NewService(provider)),
|
|
}
|
|
rt := db.AgentRuntime{
|
|
ID: util.MustParseUUID("30000000-0000-0000-0000-000000000001"),
|
|
WorkspaceID: util.MustParseUUID("30000000-0000-0000-0000-000000000002"),
|
|
}
|
|
|
|
ack, _, err := h.processHeartbeat(context.Background(), rt, false)
|
|
if err != nil {
|
|
t.Fatalf("processHeartbeat: %v", err)
|
|
}
|
|
if ack.FeatureFlags == nil {
|
|
t.Fatal("FeatureFlags snapshot is nil")
|
|
}
|
|
if got := ack.FeatureFlags.Flags[featureflagdispatch.RuntimeBriefSlimFlag]; got != "on" {
|
|
t.Fatalf("runtime_brief_slim = %q, want on", got)
|
|
}
|
|
}
|
|
|
|
type noopHeartbeatScheduler struct{}
|
|
|
|
func (noopHeartbeatScheduler) Schedule(context.Context, db.AgentRuntime) error {
|
|
return nil
|
|
}
|