greddis/commands_test.go at master · mikn/greddis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package greddis

import (
	"context"
	"testing"

	gomock "github.com/golang/mock/gomock"
	"github.com/mikn/greddis/mocks/mock_net"
	"github.com/stretchr/testify/require"
)

func TestPing(t *testing.T) {
	t.Run("fail on Flush", func(t *testing.T) {
		var ctx, cancel = context.WithCancel(context.Background())
		var ctrl = gomock.NewController(t)
		defer ctrl.Finish()
		var mockConn = mock_net.NewMockConn(ctrl)
		conn := newConn(mockConn, 4096)
		mockConn.EXPECT().Write(gomock.Any()).Return(0, ErrConnWrite)
		err := ping(ctx, conn)
		require.Error(t, err)
		cancel()
	})
	t.Run("fail on Scan", func(t *testing.T) {
		var ctx, cancel = context.WithCancel(context.Background())
		var ctrl = gomock.NewController(t)
		defer ctrl.Finish()
		var mockConn = mock_net.NewMockConn(ctrl)
		conn := newConn(mockConn, 4096)
		mockConn.EXPECT().Write(gomock.Any()).Return(14, nil)
		mockConn.EXPECT().Read(gomock.Any()).DoAndReturn(func(b []byte) (int, error) {
			copy(b, []byte("*1\r\n$4\r\nPONG\r\n"))
			return 14, nil
		})
		err := ping(ctx, conn)
		require.Error(t, err)
		cancel()
	})
	t.Run("fail on reply", func(t *testing.T) {
		var ctx, cancel = context.WithCancel(context.Background())
		var ctrl = gomock.NewController(t)
		defer ctrl.Finish()
		var mockConn = mock_net.NewMockConn(ctrl)
		conn := newConn(mockConn, 4096)
		mockConn.EXPECT().Write(gomock.Any()).Return(14, nil)
		mockConn.EXPECT().Read(gomock.Any()).DoAndReturn(func(b []byte) (int, error) {
			copy(b, []byte("+PON\r\n"))
			return 10, nil
		})
		err := ping(ctx, conn)
		require.Error(t, err)
		cancel()
	})
	t.Run("success", func(t *testing.T) {
		var ctx, cancel = context.WithCancel(context.Background())
		var ctrl = gomock.NewController(t)
		defer ctrl.Finish()
		var mockConn = mock_net.NewMockConn(ctrl)
		conn := newConn(mockConn, 4096)
		mockConn.EXPECT().Write(gomock.Any()).Return(14, nil)
		mockConn.EXPECT().Read(gomock.Any()).DoAndReturn(func(b []byte) (int, error) {
			copy(b, []byte("+PONG\r\n"))
			return 7, nil
		})
		err := ping(ctx, conn)
		require.NoError(t, err)
		cancel()
	})
}