46 lines
979 B
Go
46 lines
979 B
Go
package mocks
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/mainflux/mainflux"
|
|
"github.com/mainflux/mainflux/readers"
|
|
)
|
|
|
|
var _ readers.MessageRepository = (*messageRepositoryMock)(nil)
|
|
|
|
type messageRepositoryMock struct {
|
|
mutex sync.Mutex
|
|
messages map[uint64][]mainflux.Message
|
|
}
|
|
|
|
// NewMessageRepository returns mock implementation of message repository.
|
|
func NewMessageRepository(messages map[uint64][]mainflux.Message) readers.MessageRepository {
|
|
return &messageRepositoryMock{
|
|
mutex: sync.Mutex{},
|
|
messages: messages,
|
|
}
|
|
}
|
|
|
|
func (repo *messageRepositoryMock) ReadAll(chanID, offset, limit uint64) []mainflux.Message {
|
|
repo.mutex.Lock()
|
|
defer repo.mutex.Unlock()
|
|
|
|
end := offset + limit
|
|
|
|
numOfMessages := uint64(len(repo.messages[chanID]))
|
|
if offset < 0 || offset >= numOfMessages {
|
|
return []mainflux.Message{}
|
|
}
|
|
|
|
if limit < 1 {
|
|
return []mainflux.Message{}
|
|
}
|
|
|
|
if offset+limit > numOfMessages {
|
|
end = numOfMessages
|
|
}
|
|
|
|
return repo.messages[chanID][offset:end]
|
|
}
|