mirror of
https://github.com/golang/go
synced 2024-11-18 18:54:42 -07:00
38a97e00a8
In preparation for later changes, add support for tracking outstanding work in the lsp regtests. This simply threads through progress notifications and tracks their state in regtest.Env.state. A new Expectation is added to assert that there is no outstanding work, but this is as-yet unused. A unit test is added for Env to check that we're handling work progress reports correctly after Marshaling/Unmarshaling, since we're not yet exercising this code path in actual regtests. Change-Id: I104caf25cfd49340f13d086314f5aef2b8f3bd3b Reviewed-on: https://go-review.googlesource.com/c/tools/+/229320 Run-TryBot: Robert Findley <rfindley@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Rebecca Stambler <rstambler@golang.org>
67 lines
1.7 KiB
Go
67 lines
1.7 KiB
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 (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"golang.org/x/tools/internal/lsp/protocol"
|
|
)
|
|
|
|
func TestProgressUpdating(t *testing.T) {
|
|
e := &Env{
|
|
state: State{
|
|
outstandingWork: make(map[string]*workProgress),
|
|
},
|
|
}
|
|
ctx := context.Background()
|
|
if err := e.onWorkDoneProgressCreate(ctx, &protocol.WorkDoneProgressCreateParams{
|
|
Token: "foo",
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := e.onWorkDoneProgressCreate(ctx, &protocol.WorkDoneProgressCreateParams{
|
|
Token: "bar",
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
updates := []struct {
|
|
token string
|
|
value interface{}
|
|
}{
|
|
{"foo", protocol.WorkDoneProgressBegin{Kind: "begin", Title: "foo work"}},
|
|
{"bar", protocol.WorkDoneProgressBegin{Kind: "begin", Title: "bar work"}},
|
|
{"foo", protocol.WorkDoneProgressEnd{Kind: "end"}},
|
|
{"bar", protocol.WorkDoneProgressReport{Kind: "report", Percentage: 42}},
|
|
}
|
|
for _, update := range updates {
|
|
params := &protocol.ProgressParams{
|
|
Token: update.token,
|
|
Value: update.value,
|
|
}
|
|
data, err := json.Marshal(params)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var unmarshaled protocol.ProgressParams
|
|
if err := json.Unmarshal(data, &unmarshaled); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := e.onProgress(ctx, &unmarshaled); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
if _, ok := e.state.outstandingWork["foo"]; ok {
|
|
t.Error("got work entry for \"foo\", want none")
|
|
}
|
|
got := *e.state.outstandingWork["bar"]
|
|
want := workProgress{title: "bar work", percent: 42}
|
|
if got != want {
|
|
t.Errorf("work progress for \"bar\": %v, want %v", got, want)
|
|
}
|
|
}
|