Use testify

Signed-off-by: Maksym Pavlenko <pavlenko.maksym@gmail.com>
This commit is contained in:
Maksym Pavlenko
2022-04-01 18:17:58 -07:00
parent 9766107a53
commit 871b6b6a9f
30 changed files with 356 additions and 362 deletions

View File

@@ -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)
}