Function TestSimplicityChecker_Reset

pkg/middleware/simplicity_checker_test.go:294–329

(t *testing.T)

}

func TestSimplicityChecker_Reset(t *testing.T) {
	m := NewSimplicityCheckerMiddleware(&SimplicityCheckerConfig{
		Enabled:            true,
		MaxHelperFunctions: 1,
	})

	code := `func parseHelper() {}`

	req := &ToolCallRequest{
		ToolName: "Write",
		ToolInput: map[string]any{
			"file_path": "test.go",
			"content":   code,
		},
	}

	handler := func(ctx context.Context, req *ToolCallRequest) (*ToolCallResponse, error) {
		return &ToolCallResponse{Result: "ok"}, nil
	}

	_, _ = m.WrapToolCall(context.Background(), req, handler)

	if m.helperCount == 0 {
		t.Error("Expected helperCount > 0 after detection")
	}

	// 重置
	m.Reset()

	if m.helperCount != 0 {
		t.Errorf("Expected helperCount=0 after reset, got %d", m.helperCount)
	}
	if len(m.GetWarnings()) != 0 {
		t.Error("Expected empty warnings after reset")
	}
}

func TestSimplicityChecker_OnAgentStart(t *testing.T) {
	m := NewSimplicityCheckerMiddleware(&SimplicityCheckerConfig{