@@ -26,7 +26,7 @@ import (
|
||||
"github.com/containerd/containerd/pkg/ttrpcutil"
|
||||
"github.com/containerd/ttrpc"
|
||||
"github.com/gogo/protobuf/types"
|
||||
"gotest.tools/v3/assert"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestClientTTRPC_New(t *testing.T) {
|
||||
@@ -34,10 +34,10 @@ func TestClientTTRPC_New(t *testing.T) {
|
||||
t.Skip()
|
||||
}
|
||||
client, err := ttrpcutil.NewClient(address + ".ttrpc")
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
|
||||
err = client.Close()
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestClientTTRPC_Reconnect(t *testing.T) {
|
||||
@@ -45,13 +45,13 @@ func TestClientTTRPC_Reconnect(t *testing.T) {
|
||||
t.Skip()
|
||||
}
|
||||
client, err := ttrpcutil.NewClient(address + ".ttrpc")
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
|
||||
err = client.Reconnect()
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
|
||||
service, err := client.EventsService()
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// Send test request to make sure its alive after reconnect
|
||||
_, err = service.Forward(context.Background(), &v1.ForwardRequest{
|
||||
@@ -62,10 +62,10 @@ func TestClientTTRPC_Reconnect(t *testing.T) {
|
||||
Event: &types.Any{},
|
||||
},
|
||||
})
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
|
||||
err = client.Close()
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestClientTTRPC_Close(t *testing.T) {
|
||||
@@ -73,17 +73,17 @@ func TestClientTTRPC_Close(t *testing.T) {
|
||||
t.Skip()
|
||||
}
|
||||
client, err := ttrpcutil.NewClient(address + ".ttrpc")
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
|
||||
service, err := client.EventsService()
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
|
||||
err = client.Close()
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
|
||||
_, err = service.Forward(context.Background(), &v1.ForwardRequest{Envelope: &v1.Envelope{}})
|
||||
assert.Equal(t, err, ttrpc.ErrClosed)
|
||||
|
||||
err = client.Close()
|
||||
assert.NilError(t, err)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user