Mainflux.mainflux/authn/mocks/keys.go

// Copyright (c) Mainflux
// SPDX-License-Identifier: Apache-2.0

package mocks

import (
	"context"
	"sync"

	"github.com/mainflux/mainflux/authn"
)

var _ authn.KeyRepository = (*keyRepositoryMock)(nil)

type keyRepositoryMock struct {
	mu   sync.Mutex
	keys map[string]authn.Key
}

// NewKeyRepository creates in-memory user repository
func NewKeyRepository() authn.KeyRepository {
	return &keyRepositoryMock{
		keys: make(map[string]authn.Key),
	}
}

func (krm *keyRepositoryMock) Save(ctx context.Context, key authn.Key) (string, error) {
	krm.mu.Lock()
	defer krm.mu.Unlock()

	if _, ok := krm.keys[key.ID]; ok {
		return "", authn.ErrConflict
	}

	krm.keys[key.ID] = key
	return key.ID, nil
}
func (krm *keyRepositoryMock) Retrieve(ctx context.Context, issuer, id string) (authn.Key, error) {
	krm.mu.Lock()
	defer krm.mu.Unlock()

	if key, ok := krm.keys[id]; ok && key.Issuer == issuer {
		return key, nil
	}

	return authn.Key{}, authn.ErrNotFound
}
func (krm *keyRepositoryMock) Remove(ctx context.Context, issuer, id string) error {
	krm.mu.Lock()
	defer krm.mu.Unlock()
	if key, ok := krm.keys[id]; ok && key.Issuer == issuer {
		delete(krm.keys, id)
	}
	return nil
}