32 lines
795 B
Go
32 lines
795 B
Go
package lib
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/rpc"
|
|
"testing"
|
|
|
|
"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))
|
|
})
|
|
}
|
|
}
|