我對(duì) Go 相當(dāng)陌生，在編寫(xiě)測(cè)試時(shí)遇到了一些問(wèn)題，特別是模擬包函數(shù)的響應(yīng)。我正在為 編寫(xiě)一個(gè)包裝器庫(kù)github.com/go-redis/redis。目前它確實(shí)只有更好的失敗錯(cuò)誤，但它會(huì)隨著 statsd 進(jìn)一步跟蹤而擴(kuò)展，但我離題了......我創(chuàng)建了以下 go 包package myredisimport ( "time" "github.com/go-redis/redis" errors "github.com/pkg/errors")var newRedisClient = redis.NewClient// Options - My Redis Connection Optionstype Options struct { *redis.Options DefaultLifetime time.Duration}// MyRedis - My Redis Typetype MyRedis struct { options Options client *redis.Client}// Connect - Connects to the Redis Server. Returns an error on failurefunc (r *MyRedis) Connect() error { r.client = newRedisClient(&redis.Options{ Addr: r.options.Addr, Password: r.options.Password, DB: r.options.DB, }) _, err := r.client.Ping().Result() if err != nil { return errors.Wrap(err, "myredis") } return nil}我的問(wèn)題是我想redis.NewClient返回一個(gè)模擬。這是我寫(xiě)的測(cè)試代碼，但它不起作用：package myredisimport ( "testing" "github.com/go-redis/redis" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/mock")type redisStatusCmdMock struct { mock.Mock}func (m *redisStatusCmdMock) Result() (string, error) { args := m.Called() return args.Get(0).(string), args.Error(1)}type redisClientMock struct { mock.Mock}func (m *redisClientMock) Ping() redis.StatusCmd { args := m.Called() return args.Get(0).(redis.StatusCmd)}func TestConnect(t *testing.T) { assert := assert.New(t) old := newRedisClient defer func() { newRedisClient = old }() newRedisClient = func(options *redis.Options) *redis.Client { assert.Equal("127.0.0.1:1001", options.Addr) assert.Equal("password", options.Password) assert.Equal(1, options.DB) }}我收到以下錯(cuò)誤：cannot use clientMock (type *redisClientMock) as type *redis.Client in return argument。我想我讀到我需要模擬 的所有功能，redis.Client以便能夠在這種情況下將其用作模擬，但事實(shí)真的是這樣嗎？這似乎太過(guò)分了，我應(yīng)該能夠以某種方式做到這一點(diǎn)。我該如何讓這個(gè)測(cè)試工作，或者我是否需要重組我的代碼以便更容易編寫(xiě)測(cè)試？
從包函數(shù)返回 Mock
拉丁的傳說(shuō)
2023-07-26 15:32:31
