Golang 新手在这里:)
我有一个基于 go-redis 的 redis 客户端包装器,我想测试它是否正常工作。我尝试使用 miniredis 来模拟我要连接的 redis 服务器,但我不断收到错误。
当我将所有内容都放在同一个函数中时,这就可以了——我可以针对 miniredis 服务运行 go-redis。但是,当我尝试使用我的客户端时,它失败了。
我尝试查看 go-redis 和 miniredis 的文档,但找不到任何答案。我也尝试过将 miniredis 服务器定义为全局变量,但这也没有帮助。
客户端实现:
package redis
import (
"errors"
"fmt"
"github.com/go-redis/redis"
"go.uber.org/zap"
"os"
"strconv"
"time"
)
var redisClient *redis.Client
var redisCert string
var redisURL = "localhost:6379"
var redisPass = ""
var redisDB = 0
type Client struct {
redisClient *redis.Client
}
func newRedisClient() *Client {
if os.Getenv("REDIS_URL") != "" {
redisURL = os.Getenv("REDIS_URL")
}
if os.Getenv("REDIS_PASS") != "" {
redisPass = os.Getenv("REDIS_PASS")
}
if os.Getenv("REDIS_DB") != "" {
redisDB, _ = strconv.Atoi(os.Getenv("REDIS_DB"))
}
client := redis.NewClient(&redis.Options{
Addr: redisURL,
Password: redisPass,
DB: redisDB,
})
redisClient = client
_, err := redisClient.Ping().Result()
if err != nil {
fmt.Println("Failed to connect", zap.Error(err))
return nil
}
fmt.Println("Redis client is ready")
return &Client{redisClient: client}
}
func (r *Client) Get(key string) (string, error) {
res, err := redisClient.Get(key).Result()
fmt.Println(res, err)
if err == redis.Nil {
return "", errors.New("Key Not Found")
} else if err != nil {
return "", err
} else {
return res, nil
}
}
测试实现:
package redis
import (
"github.com/alicebob/miniredis/v2"
"gopkg.in/go-playground/assert.v1"
"os"
"testing"
)
var mr *miniredis.Miniredis
func TestGet(t *testing.T) {
mr, _ = miniredis.Run()
mr.Set("test1", "some")
os.Setenv("REDIS_URL", mr.Addr())
c := newRedisClient()
r, err := c.Get("test1")
assert.Equal(t, r, "")
assert.Equal(t, err, "")
mr.Close()
}
我做错了什么?
通过miniredis模拟失败,可以使用
SetError
:https://github.com/alicebob/miniredis/issues/36#issuecomment-642009898
func TestGet(t *testing.T) {
mr, _ = miniredis.Run()
// normal test
// ...
// error test
mr.SetError("mock error")
_, err = c.Get("test2")
assert.EqualError(t, err, "mock error")
mr.Close()
}
抱歉这么久才回复。我已在本地运行您的代码,但没有收到错误。
普通文件
package redis
import (
"errors"
"fmt"
"os"
"strconv"
"github.com/go-redis/redis"
"go.uber.org/zap"
)
var redisClient *redis.Client
var redisCert string
var redisURL = "localhost:6379"
var redisPass = ""
var redisDB = 0
type Client struct {
redisClient *redis.Client
}
func newRedisClient() *Client {
if os.Getenv("REDIS_URL") != "" {
redisURL = os.Getenv("REDIS_URL")
}
if os.Getenv("REDIS_PASS") != "" {
redisPass = os.Getenv("REDIS_PASS")
}
if os.Getenv("REDIS_DB") != "" {
redisDB, _ = strconv.Atoi(os.Getenv("REDIS_DB"))
}
client := redis.NewClient(&redis.Options{
Addr: redisURL,
Password: redisPass,
DB: redisDB,
})
redisClient = client
_, err := redisClient.Ping().Result()
if err != nil {
fmt.Println("Failed to connect", zap.Error(err))
return nil
}
fmt.Println("Redis client is ready")
return &Client{redisClient: client}
}
func (r *Client) Get(key string) (string, error) {
res, err := redisClient.Get(key).Result()
if err == redis.Nil {
return "", errors.New("Key Not Found")
} else if err != nil {
return "", err
} else {
return res, nil
}
}
测试文件
package redis
import (
"fmt"
"os"
"testing"
"github.com/alicebob/miniredis/v2"
)
var mr *miniredis.Miniredis
func TestGet(t *testing.T) {
var err error
mr, err = miniredis.Run()
fmt.Printf("Run err: %v\n", err)
fmt.Printf("mr.Addr(): %v\n", mr.Addr())
err = mr.Set("test1", "some")
fmt.Printf("Set err: %v\n", err)
err = os.Setenv("REDIS_URL", mr.Addr())
fmt.Printf("Setenv err: %v\n", err)
c := newRedisClient()
r, err := c.Get("test1")
fmt.Printf("Get err: %v\n", err)
fmt.Printf("r: %v\n", r)
mr.Close()
}
结果
go test
Run err: <nil>
mr.Addr(): 127.0.0.1:54966
Set err: <nil>
Setenv err: <nil>
Redis client is ready
Get err: <nil>
r: some
PASS
ok test/sof 0.536s
你已经设置了 var redisURL = "localhost:6379"
但在测试结果中 先生地址(): 127.0.0.1:54966
有什么方法可以始终运行固定端口的 miniredis 实例吗?