Function TestContextInjector_InjectSessionState

pkg/context/auto_inject_test.go:64–95

(t *testing.T)

}

func TestContextInjector_InjectSessionState(t *testing.T) {
	state := &SessionState{
		SessionID:    "session-123",
		MessageCount: 42,
		Duration:     30 * time.Minute,
		CustomState: map[string]any{
			"mode": "debug",
		},
	}

	config := InjectorConfig{
		EnableSessionState:   true,
		SessionStateProvider: NewSimpleSessionStateProvider(state),
	}

	injector := NewContextInjector(config)
	systemPrompt := "You are a helpful assistant."

	result := injector.InjectContext(context.Background(), systemPrompt)

	if !strings.Contains(result, "Session State") {
		t.Error("expected session state context to be injected")
	}

	if !strings.Contains(result, "session-123") {
		t.Error("expected session ID to be included")
	}

	if !strings.Contains(result, "42") {
		t.Error("expected message count to be included")
	}
}

func TestContextInjector_MultipleContexts(t *testing.T) {
	config := InjectorConfig{