redigoのPoolとredigomockを利用する方法

// service.go

package service

import (...) // 省略

type RedigoPool interface {
    Get() redis.Conn
    Close() error
}

type ExampleService struct {
    pool RedigoPool // *redis.Poolを直接定義せず, RedigoPoolというインタフェースを利用する
}

func (s *ExampleService) HmSetWithExpire(key string, a, b int64, expireSeconds uint) (interface{}, error) {
    redisConn := s.pool.Get()
    defer redisConn.Close()

    redisConn.Send("MULTI")
    redisConn.Send("HMSET", key, "a", a, "b", b)
    redisConn.Send("EXPIRE", key, expireSeconds)

    reply, err := redisConn.Do("EXEC")
    if err != nil {
        return nil, err
    }
    return reply, nil
}

// service_test.go

package service

import (...) // 省略

// redigoMockPool redigomock用RedisPool
type redigoMockPool struct {
    conn redis.Conn
}

func (m redigoMockPool) Get() redis.Conn {
    return m.conn
}

func (m redigoMockPool) Close() error {
    return m.conn.Close()
}

func TestExampleService_HmSetWithExpire(t *testing.T) {
    mockConn := redigomock.NewConn()
    mockPool := &redigoMockPool{
        conn: mockConn,
    }

    svc := ExampleService{
        pool: mockPool,
    }

    key := "test-uhyouhyo!"
    a := int64(1)
    b := int64(2)
    expireSeconds := uint(3200)

    cmd1 := mockConn.Command("MULTI")
    cmd2 := mockConn.Command("HMSET", key, "a", a, "b", b)
    cmd3 := mockConn.Command("EXPIRE", key, expireSeconds)
    cmd4 := mockConn.Command("EXEC").Expect([]interface{}{"OK", "OK"})

    _, err := svc.HmSetWithExpire(key, a, b, expireSeconds)
    if err != nil {
        t.Error(err)
    }

    if counter := mockConn.Stats(cmd1); counter != 1 {
        t.Errorf("Expected cmd1 to be called once but was called %d times", counter)
    }

    if counter := mockConn.Stats(cmd2); counter != 1 {
        t.Errorf("Expected cmd2 to be called once but was called %d times", counter)
    }

    if counter := mockConn.Stats(cmd3); counter != 1 {
        t.Errorf("Expected cmd3 to be called once but was called %d times", counter)
    }

    if counter := mockConn.Stats(cmd4); counter != 1 {
        t.Errorf("Expected cmd4 to be called once but was called %d times", counter)
    }
}