You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.7 KiB
53 lines
1.7 KiB
package redis
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-redis/redismock/v9"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestStandaloneGatewayRedisRegisterGateway(t *testing.T) {
|
|
rdb, mock := redismock.NewClientMock()
|
|
cache := NewStandaloneGatewayRedis(rdb, time.Second*10)
|
|
|
|
mock.Regexp().ExpectHSet(standaloneGatewayHashKey, "127.0.0.1:10001", `^[0-9]+$`).SetVal(1)
|
|
mock.ExpectExpire(standaloneGatewayHashKey, time.Second*20).SetVal(true)
|
|
|
|
err := cache.RegisterGateway(context.Background(), "127.0.0.1:10001")
|
|
require.NoError(t, err)
|
|
assert.NoError(t, mock.ExpectationsWereMet())
|
|
}
|
|
|
|
func TestStandaloneGatewayRedisUnregisterGateway(t *testing.T) {
|
|
rdb, mock := redismock.NewClientMock()
|
|
cache := NewStandaloneGatewayRedis(rdb, time.Second)
|
|
|
|
mock.ExpectHDel(standaloneGatewayHashKey, "127.0.0.1:10001").SetVal(1)
|
|
|
|
err := cache.UnregisterGateway(context.Background(), "127.0.0.1:10001")
|
|
require.NoError(t, err)
|
|
assert.NoError(t, mock.ExpectationsWereMet())
|
|
}
|
|
|
|
func TestStandaloneGatewayRedisGetGatewayAddrs(t *testing.T) {
|
|
rdb, mock := redismock.NewClientMock()
|
|
cache := NewStandaloneGatewayRedis(rdb, time.Second*10)
|
|
|
|
now := time.Now()
|
|
mock.ExpectHGetAll(standaloneGatewayHashKey).SetVal(map[string]string{
|
|
"127.0.0.1:10001": strconv.FormatInt(now.Add(-time.Second).UnixMilli(), 10),
|
|
"127.0.0.1:10002": strconv.FormatInt(now.Add(-time.Second*20).UnixMilli(), 10),
|
|
"127.0.0.1:10003": strconv.FormatInt(now.Add(time.Second).UnixMilli(), 10),
|
|
})
|
|
|
|
addrs, err := cache.GetGatewayAddrs(context.Background())
|
|
require.NoError(t, err)
|
|
assert.ElementsMatch(t, []string{"127.0.0.1:10001", "127.0.0.1:10003"}, addrs)
|
|
assert.NoError(t, mock.ExpectationsWereMet())
|
|
}
|