Function TestV2ErrorStreamReading

tools/remotecommand/v2_test.go:180–228

(t *testing.T)

}

func TestV2ErrorStreamReading(t *testing.T) {
	tests := []struct {
		name          string
		stream        io.Reader
		expectedError error
	}{
		{
			name:          "error reading from stream",
			stream:        &fakeReader{errors.New("foo")},
			expectedError: errors.New("error reading from error stream: foo"),
		},
		{
			name:          "stream returns an error",
			stream:        strings.NewReader("some error"),
			expectedError: errors.New("error executing remote command: some error"),
		},
	}

	for _, test := range tests {
		h := newStreamProtocolV2(StreamOptions{}).(*streamProtocolV2)
		h.errorStream = test.stream

		ch := watchErrorStream(h.errorStream, &errorDecoderV2{})
		if ch == nil {
			t.Fatalf("%s: unexpected nil channel", test.name)
		}

		var err error
		select {
		case err = <-ch:
		case <-time.After(wait.ForeverTestTimeout):
			t.Fatalf("%s: timed out", test.name)
		}

		if test.expectedError != nil {
			if err == nil {
				t.Errorf("%s: expected an error", test.name)
			} else if e, a := test.expectedError, err; e.Error() != a.Error() {
				t.Errorf("%s: expected %q, got %q", test.name, e, a)
			}
			continue
		}

		if test.expectedError == nil && err != nil {
			t.Errorf("%s: unexpected error: %v", test.name, err)
			continue
		}
	}
}