Function TestServer_RateLimit_HandleMessage

pkg/lsp/server_ratelimit_test.go:164–199

TestServer_RateLimit_HandleMessage tests rate limiting in message handling

(t *testing.T)

// TestServer_RateLimit_HandleMessage tests rate limiting in message handling
func TestServer_RateLimit_HandleMessage(t *testing.T) {
	mock := newMockReadWriter()
	logger := log.New(io.Discard, "", 0)
	server := NewServer(mock.input, mock.output, logger)

	// Create a valid request
	req := Request{
		JSONRPC: "2.0",
		ID:      1,
		Method:  "shutdown",
	}
	msgBytes, err := json.Marshal(req)
	if err != nil {
		t.Fatalf("failed to marshal request: %v", err)
	}

	// Send requests up to the limit
	for i := 0; i < RateLimitRequests; i++ {
		server.handleMessage(msgBytes)
	}

	// Clear output buffer to check next response
	mock.output.Reset()

	// Next request should be rate limited
	server.handleMessage(msgBytes)

	// Check that a rate limit error response was sent
	output := mock.output.String()
	if output != "" {
		// Rate limited requests should get an error response
		if !contains(output, "RequestCancelled") && !contains(output, "rate limit") {
			t.Logf("Output after rate limit: %s", output)
		}
	}
}

// TestServer_RateLimit_MultipleWindows tests behavior across multiple windows
func TestServer_RateLimit_MultipleWindows(t *testing.T) {