mirror of
https://github.com/golang/go
synced 2024-10-01 10:38:33 -06:00
741f65b509
Add a forwarder handler that alters messages before forwarding, for now, it just intercepts the "exit" message. Also, make it easier to write regression tests for a shared gopls instance, by adding a helper that instantiates two connected environments, and only runs in the shared execution modes. Updates golang/go#36879 Updates golang/go#34111 Change-Id: I7673f72ab71b5c7fd6ad65d274c15132a942e06a Reviewed-on: https://go-review.googlesource.com/c/tools/+/218778 Run-TryBot: Robert Findley <rfindley@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Heschi Kreinick <heschi@google.com>
35 lines
974 B
Go
35 lines
974 B
Go
// Copyright 2020 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package regtest
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/tools/internal/lsp"
|
|
"golang.org/x/tools/internal/lsp/lsprpc"
|
|
)
|
|
|
|
var runner *Runner
|
|
|
|
func TestMain(m *testing.M) {
|
|
// Override functions that would shut down the test process
|
|
defer func(lspExit, forwarderExit func(code int)) {
|
|
lsp.ServerExitFunc = lspExit
|
|
lsprpc.ForwarderExitFunc = forwarderExit
|
|
}(lsp.ServerExitFunc, lsprpc.ForwarderExitFunc)
|
|
// None of these regtests should be able to shut down a server process.
|
|
lsp.ServerExitFunc = func(code int) {
|
|
panic(fmt.Sprintf("LSP server exited with code %d", code))
|
|
}
|
|
// We don't want our forwarders to exit, but it's OK if they would have.
|
|
lsprpc.ForwarderExitFunc = func(code int) {}
|
|
runner = NewTestRunner(AllModes, 30*time.Second)
|
|
defer runner.Close()
|
|
os.Exit(m.Run())
|
|
}
|