test_redis_client_test.go

package rmq

import (
	"testing"
	"time"

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

func TestTestRedisClient_Set(t *testing.T) {
	type args struct {
		key        string
		value      string
		expiration time.Duration
	}
	tests := []struct {
		name   string
		client *TestRedisClient
		args   args
	}{
		{
			"successfull add",
			NewTestRedisClient(),
			args{
				"somekey",
				"somevalue",
				time.Duration(0),
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {

			//add
			err := tt.client.Set(tt.args.key, tt.args.value, tt.args.expiration)
			assert.NoError(t, err)

			//get
			v, err := tt.client.Get(tt.args.key)
			assert.Equal(t, tt.args.value, v)
			assert.NoError(t, err)

			//delete
			affected, err := tt.client.Del(tt.args.key)
			assert.Equal(t, int64(1), affected)
			assert.NoError(t, err)

			//delete it again
			affected, err = tt.client.Del(tt.args.key)
			assert.Equal(t, int64(0), affected)
			assert.NoError(t, err)
		})
	}
}

func TestTestRedisClient_SAdd(t *testing.T) {
	type args struct {
		key   string
		value string
	}
	tests := []struct {
		name   string
		client *TestRedisClient
		args   args
	}{
		{
			"adding member",
			NewTestRedisClient(),
			args{
				"somekey",
				"somevalue",
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			total, err := tt.client.SAdd(tt.args.key, tt.args.value)
			assert.NoError(t, err)
			assert.Equal(t, int64(1), total)

			total, err = tt.client.SAdd(tt.args.key, tt.args.value)
			assert.NoError(t, err)
			assert.Equal(t, int64(1), total)

			members, err := tt.client.SMembers(tt.args.key)
			assert.Equal(t, []string{tt.args.value}, members)
			assert.NoError(t, err)

			count, err := tt.client.SRem(tt.args.key, tt.args.value)
			assert.Equal(t, int64(1), count)
			assert.NoError(t, err)
		})
	}
}

func TestTestRedisClient_LPush(t *testing.T) {
	type args struct {
		key   string
		value string
	}
	tests := []struct {
		name   string
		client *TestRedisClient
		args   args
		total  int64
	}{
		{
			"adding to list",
			NewTestRedisClient(),
			args{
				"somekey",
				"somevalue",
			},
			1,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {

			//Push
			total, err := tt.client.LPush(tt.args.key, tt.args.value)
			assert.NoError(t, err)
			assert.Equal(t, tt.total, total)

			//Len
			count, err := tt.client.LLen(tt.args.key)
			assert.Equal(t, int64(1), count)
			assert.NoError(t, err)

			//Len of non-existing
			count, err = tt.client.LLen(tt.args.key + "nonsense")
			assert.Equal(t, int64(0), count)
			assert.NoError(t, err)

			//Lrem
			count, err = tt.client.LRem(tt.args.key, 100, tt.args.value)
			assert.Equal(t, int64(1), count)
			assert.NoError(t, err)

			//Len again
			count, err = tt.client.LLen(tt.args.key)
			assert.Equal(t, int64(0), count)
			assert.NoError(t, err)
		})
	}
}