services/identity/internal/auditor/mock_auditor.go

// Copyright 2015 The Vanadium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package auditor

import (
	"reflect"

	"v.io/v23/context"
	"v.io/x/ref/lib/security/audit"
)

func NewMockBlessingAuditor() (audit.Auditor, BlessingLogReader) {
	db := &mockDatabase{}
	return &blessingAuditor{db}, &blessingLogReader{db}
}

type mockDatabase struct {
	NextEntry databaseEntry
}

func (db *mockDatabase) Insert(ctx *context.T, entry databaseEntry) error {
	db.NextEntry = entry
	return nil
}

func (db *mockDatabase) Query(ctx *context.T, email string) <-chan databaseEntry {
	c := make(chan databaseEntry)
	go func() {
		var empty databaseEntry
		if !reflect.DeepEqual(db.NextEntry, empty) {
			c <- db.NextEntry
		}
		close(c)
	}()
	return c
}