1 files changed, 66 insertions, 0 deletions
diff --git a/gopls/internal/lsp/regtest/env_test.go b/gopls/internal/lsp/regtest/env_test.go
new file mode 100644
index 000000000..e334faa90
--- /dev/null
+++ b/gopls/internal/lsp/regtest/env_test.go
@@ -0,0 +1,66 @@
+// 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/gopls/internal/lsp/protocol"
+)
+
+func TestProgressUpdating(t *testing.T) {
+	a := &Awaiter{
+		state: State{
+			work: make(map[protocol.ProgressToken]*workProgress),
+		},
+	}
+	ctx := context.Background()
+	if err := a.onWorkDoneProgressCreate(ctx, &protocol.WorkDoneProgressCreateParams{
+		Token: "foo",
+	}); err != nil {
+		t.Fatal(err)
+	}
+	if err := a.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 := a.onProgress(ctx, &unmarshaled); err != nil {
+			t.Fatal(err)
+		}
+	}
+	if !a.state.work["foo"].complete {
+		t.Error("work entry \"foo\" is incomplete, want complete")
+	}
+	got := *a.state.work["bar"]
+	want := workProgress{title: "bar work", percent: 42}
+	if got != want {
+		t.Errorf("work progress for \"bar\": %v, want %v", got, want)
+	}
+}