mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-29 18:36:30 +02:00
f73c5c615f
* databroker: add support for putting multiple records * add OptimumPutRequestsFromRecords function * replace GetAll with SyncLatest * fix stream when there are no records
40 lines
1 KiB
Go
40 lines
1 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/pomerium/pomerium/pkg/grpc/databroker"
|
|
"github.com/pomerium/pomerium/pkg/grpc/user"
|
|
"github.com/pomerium/pomerium/pkg/protoutil"
|
|
)
|
|
|
|
type mockBackend struct {
|
|
Backend
|
|
put func(ctx context.Context, records []*databroker.Record) (uint64, error)
|
|
get func(ctx context.Context, recordType, id string) (*databroker.Record, error)
|
|
}
|
|
|
|
func (m *mockBackend) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (m *mockBackend) Put(ctx context.Context, records []*databroker.Record) (uint64, error) {
|
|
return m.put(ctx, records)
|
|
}
|
|
|
|
func (m *mockBackend) Get(ctx context.Context, recordType, id string) (*databroker.Record, error) {
|
|
return m.get(ctx, recordType, id)
|
|
}
|
|
|
|
func TestMatchAny(t *testing.T) {
|
|
u := &user.User{Id: "id", Name: "name", Email: "email"}
|
|
data := protoutil.NewAny(u)
|
|
assert.True(t, MatchAny(data, ""))
|
|
assert.True(t, MatchAny(data, "id"))
|
|
assert.True(t, MatchAny(data, "name"))
|
|
assert.True(t, MatchAny(data, "email"))
|
|
assert.False(t, MatchAny(data, "nope"))
|
|
}
|