lsp: Increase test buffered channel size (#657)

TestLanguageServerMultipleFiles is sometimes locking up as the handler
can't save new messages as the channel is waiting to be read and the
part of the test that is waiting to read those particular messages is
later and waiting for another file first.

This makes the buffered channels larger to allow more messages to be
received for each file so that we can no block and check the messages we
want in order as before.

Signed-off-by: Charlie Egan <charlie@styra.com>

internal/lsp/server_test.go

@@ -18,6 +18,8 @@ const mainRegoFileName = "/main.rego"
 
 const defaultTimeout = 3 * time.Second
 
+const defaultBufferedChannelSize = 5
+
 // InMemoryReadWriteCloser is an in-memory implementation of jsonrpc2.ReadWriteCloser.
 type InMemoryReadWriteCloser struct {
 	Buffer bytes.Buffer
@@ -87,7 +89,7 @@ allow = true
 	go ls.StartDiagnosticsWorker(ctx)
 	go ls.StartConfigWorker(ctx)
 
-	receivedMessages := make(chan FileDiagnostics, 1)
+	receivedMessages := make(chan FileDiagnostics, defaultBufferedChannelSize)
 	clientHandler := func(_ context.Context, _ *jsonrpc2.Conn, req *jsonrpc2.Request) (result any, err error) {
 		if req.Method == methodTextDocumentPublishDiagnostics {
 			var requestData FileDiagnostics
@@ -331,9 +333,9 @@ ignore:
 	go ls.StartDiagnosticsWorker(ctx)
 	go ls.StartConfigWorker(ctx)
 
-	authzFileMessages := make(chan FileDiagnostics, 1)
-	adminsFileMessages := make(chan FileDiagnostics, 1)
-	ignoredFileMessages := make(chan FileDiagnostics, 1)
+	authzFileMessages := make(chan FileDiagnostics, defaultBufferedChannelSize)
+	adminsFileMessages := make(chan FileDiagnostics, defaultBufferedChannelSize)
+	ignoredFileMessages := make(chan FileDiagnostics, defaultBufferedChannelSize)
 	clientHandler := func(_ context.Context, _ *jsonrpc2.Conn, req *jsonrpc2.Request) (result any, err error) {
 		if req.Method == "textDocument/publishDiagnostics" {
 			var requestData FileDiagnostics
@@ -562,7 +564,7 @@ allow := true
 	go ls.StartDiagnosticsWorker(ctx)
 	go ls.StartConfigWorker(ctx)
 
-	receivedMessages := make(chan FileDiagnostics, 1)
+	receivedMessages := make(chan FileDiagnostics, defaultBufferedChannelSize)
 	clientHandler := func(_ context.Context, _ *jsonrpc2.Conn, req *jsonrpc2.Request) (result any, err error) {
 		if req.Method == methodTextDocumentPublishDiagnostics {
 			var requestData FileDiagnostics