58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
//
|
|
// Copyright (c) 2018
|
|
// Mainflux
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
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[string][]mainflux.Message
|
|
}
|
|
|
|
// NewMessageRepository returns mock implementation of message repository.
|
|
func NewMessageRepository(messages map[string][]mainflux.Message) readers.MessageRepository {
|
|
return &messageRepositoryMock{
|
|
mutex: sync.Mutex{},
|
|
messages: messages,
|
|
}
|
|
}
|
|
|
|
func (repo *messageRepositoryMock) ReadAll(chanID string, offset, limit uint64, query map[string]string) readers.MessagesPage {
|
|
repo.mutex.Lock()
|
|
defer repo.mutex.Unlock()
|
|
|
|
end := offset + limit
|
|
|
|
numOfMessages := uint64(len(repo.messages[chanID]))
|
|
if offset < 0 || offset >= numOfMessages {
|
|
return readers.MessagesPage{}
|
|
}
|
|
|
|
if limit < 1 {
|
|
return readers.MessagesPage{}
|
|
}
|
|
|
|
if offset+limit > numOfMessages {
|
|
end = numOfMessages
|
|
}
|
|
|
|
return readers.MessagesPage{
|
|
Total: numOfMessages,
|
|
Limit: limit,
|
|
Offset: offset,
|
|
Messages: repo.messages[chanID][offset:end],
|
|
}
|
|
}
|