authboss/internal/mocks/mocks.go

package mocks

import "gopkg.in/authboss.v0"

type MockUser struct {
	Email    string
	Password string
}

type MockStorer struct {
	Users  map[string]authboss.Attributes
	Tokens map[string][]string
}

func NewMockStorer() *MockStorer {
	return &MockStorer{
		Users:  make(map[string]authboss.Attributes),
		Tokens: make(map[string][]string),
	}
}

func (m *MockStorer) Create(key string, attr authboss.Attributes) error {
	m.Users[key] = attr
	return nil
}

func (m *MockStorer) Put(key string, attr authboss.Attributes) error {
	m.Users[key] = attr
	return nil
}

func (m *MockStorer) Get(key string, attrMeta authboss.AttributeMeta) (result interface{}, err error) {
	return &MockUser{
		m.Users[key]["email"].(string), m.Users[key]["password"].(string),
	}, nil
}

func (m *MockStorer) AddToken(key, token string) error {
	arr := m.Tokens[key]
	m.Tokens[key] = append(arr, token)
	return nil
}

func (m *MockStorer) DelTokens(key string) error {
	delete(m.Tokens, key)
	return nil
}

func (m *MockStorer) UseToken(givenKey, token string) (key string, err error) {
	if arr, ok := m.Tokens[givenKey]; ok {
		for _, tok := range arr {
			if tok == token {
				return givenKey, nil
			}
		}
	}

	return "", authboss.ErrTokenNotFound
}

type MockClientStorer map[string]string

func (m MockClientStorer) Get(key string) (string, bool) {
	v, ok := m[key]
	return v, ok
}
func (m MockClientStorer) Put(key, val string) { m[key] = val }
func (m MockClientStorer) Del(key string)      { delete(m, key) }