Mainflux.mainflux/clients/mocks/clients.go

package mocks

import (
	"fmt"
	"strings"
	"sync"

	"github.com/mainflux/mainflux/clients"
)

var _ clients.ClientRepository = (*clientRepositoryMock)(nil)

const cliID = "123e4567-e89b-12d3-a456-"

type clientRepositoryMock struct {
	mu      sync.Mutex
	counter int
	clients map[string]clients.Client
}

// NewClientRepository creates in-memory client repository.
func NewClientRepository() clients.ClientRepository {
	return &clientRepositoryMock{
		clients: make(map[string]clients.Client),
	}
}

func (crm *clientRepositoryMock) ID() string {
	crm.mu.Lock()
	defer crm.mu.Unlock()

	crm.counter++
	return fmt.Sprintf("%s%012d", cliID, crm.counter)
}

func (crm *clientRepositoryMock) Save(client clients.Client) error {
	crm.mu.Lock()
	defer crm.mu.Unlock()

	crm.clients[key(client.Owner, client.ID)] = client

	return nil
}

func (crm *clientRepositoryMock) Update(client clients.Client) error {
	crm.mu.Lock()
	defer crm.mu.Unlock()

	dbKey := key(client.Owner, client.ID)

	if _, ok := crm.clients[dbKey]; !ok {
		return clients.ErrNotFound
	}

	crm.clients[dbKey] = client

	return nil
}

func (crm *clientRepositoryMock) One(owner, id string) (clients.Client, error) {
	if c, ok := crm.clients[key(owner, id)]; ok {
		return c, nil
	}

	return clients.Client{}, clients.ErrNotFound
}

func (crm *clientRepositoryMock) All(owner string, offset, limit int) []clients.Client {
	// This obscure way to examine map keys is enforced by the key structure
	// itself (see mocks/commons.go).
	prefix := fmt.Sprintf("%s-", owner)
	clients := make([]clients.Client, 0)

	if offset < 0 || limit <= 0 {
		return clients
	}

	// Since IDs start from 1, shift everything by one.
	first := fmt.Sprintf("%s%012d", cliID, offset+1)
	last := fmt.Sprintf("%s%012d", cliID, offset+limit+1)

	for k, v := range crm.clients {
		if strings.HasPrefix(k, prefix) && v.ID >= first && v.ID < last {
			clients = append(clients, v)
		}
	}

	return clients
}

func (crm *clientRepositoryMock) Remove(owner, id string) error {
	delete(crm.clients, key(owner, id))
	return nil
}