consul/lib/eof_test.go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0

package lib

import (
	"fmt"
	"io"
	"testing"

	"github.com/hashicorp/consul-net-rpc/net/rpc"
	"github.com/hashicorp/yamux"

	"github.com/stretchr/testify/require"
)

func TestErrIsEOF(t *testing.T) {
	var tests = []struct {
		name string
		err  error
	}{
		{name: "EOF", err: io.EOF},
		{name: "Wrapped EOF", err: fmt.Errorf("test: %w", io.EOF)},
		{name: "yamuxStreamClosed", err: yamux.ErrStreamClosed},
		{name: "yamuxSessionShutdown", err: yamux.ErrSessionShutdown},
		{name: "ServerError(___: EOF)", err: rpc.ServerError(fmt.Sprintf("rpc error: %s", io.EOF.Error()))},
		{name: "Wrapped ServerError(___: EOF)", err: fmt.Errorf("rpc error: %w", rpc.ServerError(fmt.Sprintf("rpc error: %s", io.EOF.Error())))},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			require.True(t, IsErrEOF(tt.err))
		})
	}
}