upstream/icingadb
1
0
Fork 0
mirror of https://github.com/Icinga/icingadb.git synced 2026-05-28 04:35:54 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Remove Redis testing mocks

Browse source
This commit is contained in:
Noah Hilverling 2019-03-04 18:11:00 +01:00
parent debeb88de6
commit 4f09b5b5e8
1 changed files with 0 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
redis_test.go
View file

@ -3,52 +3,12 @@ package icingadb_connection
import (
"github.com/go-redis/redis"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/mock"
"sync"
"sync/atomic"
"testing"
"time"
)
type RdbMock struct {
mock.Mock
}
func (m *RdbMock) Ping() *redis.StatusCmd {
args := m.Called()
return args.Get(0).(*redis.StatusCmd)
}
func (m *RdbMock) Publish(channel string, message interface{}) *redis.IntCmd {
args := m.Called(channel, message)
return args.Get(0).(*redis.IntCmd)
}
func (m *RdbMock) XRead(a *redis.XReadArgs) *redis.XStreamSliceCmd {
args := m.Called(a)
return args.Get(0).(*redis.XStreamSliceCmd)
}
func (m *RdbMock) XDel(stream string, ids ...string) *redis.IntCmd {
args := m.Called(stream, ids)
return args.Get(0).(*redis.IntCmd)
}
func (m *RdbMock) HGetAll(key string) *redis.StringStringMapCmd {
args := m.Called(key)
return args.Get(0).(*redis.StringStringMapCmd)
}
func (m *RdbMock) TxPipelined(fn func(redis.Pipeliner) error) ([]redis.Cmder, error) {
args := m.Called(fn)
return args.Get(0).([]redis.Cmder), args.Error(1)
}
func (m *RdbMock) Subscribe(channels ...string) *redis.PubSub {
args := m.Called(channels)
return args.Get(0).(*redis.PubSub)
}
func NewTestRDBW(rdb RedisClient) RDBWrapper {
dbw := RDBWrapper{Rdb: rdb, ConnectedAtomic: new(uint32), ConnectionLostCounterAtomic: new(uint32)}
dbw.ConnectionUpCondition = sync.NewCond(&sync.Mutex{})