mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-02 20:06:03 +02:00
5d60cff21e
* refactor backend, implement encrypted store * refactor in-memory store * wip * wip * wip * add syncer test * fix redis expiry * fix linting issues * fix test by skipping non-config records * fix backoff import * fix init issues * fix query * wait for initial sync before starting directory sync * add type to SyncLatest * add more log messages, fix deadlock in in-memory store, always return server version from SyncLatest * update sync types and tests * add redis tests * skip macos in github actions * add comments to proto * split getBackend into separate methods * handle errors in initVersion * return different error for not found vs other errors in get * use exponential backoff for redis transaction retry * rename raw to result * use context instead of close channel * store type urls as constants in databroker * use timestampb instead of ptypes * fix group merging not waiting * change locked names * update GetAll to return latest record version * add method to grpcutil to get the type url for a protobuf type
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package evaluator
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/golang/protobuf/ptypes"
|
|
"github.com/open-policy-agent/opa/storage"
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/pomerium/pomerium/pkg/grpc/databroker"
|
|
"github.com/pomerium/pomerium/pkg/grpc/user"
|
|
)
|
|
|
|
func TestStore(t *testing.T) {
|
|
ctx, clearTimeout := context.WithTimeout(context.Background(), time.Second*10)
|
|
defer clearTimeout()
|
|
|
|
s := NewStore()
|
|
t.Run("records", func(t *testing.T) {
|
|
u := &user.User{
|
|
Version: "v1",
|
|
Id: "u1",
|
|
Name: "name",
|
|
Email: "name@example.com",
|
|
}
|
|
any, _ := ptypes.MarshalAny(u)
|
|
s.UpdateRecord(&databroker.Record{
|
|
Version: 1,
|
|
Type: any.GetTypeUrl(),
|
|
Id: u.GetId(),
|
|
Data: any,
|
|
})
|
|
|
|
v, err := storage.ReadOne(ctx, s.opaStore, storage.MustParsePath("/databroker_data/type.googleapis.com/user.User/u1"))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, map[string]interface{}{
|
|
"version": "v1",
|
|
"id": "u1",
|
|
"name": "name",
|
|
"email": "name@example.com",
|
|
}, v)
|
|
|
|
s.UpdateRecord(&databroker.Record{
|
|
Version: 2,
|
|
Type: any.GetTypeUrl(),
|
|
Id: u.GetId(),
|
|
Data: any,
|
|
DeletedAt: ptypes.TimestampNow(),
|
|
})
|
|
|
|
v, err = storage.ReadOne(ctx, s.opaStore, storage.MustParsePath("/databroker_data/type.googleapis.com/user.User/u1"))
|
|
assert.Error(t, err)
|
|
assert.Nil(t, v)
|
|
|
|
s.UpdateRecord(&databroker.Record{
|
|
Version: 3,
|
|
Type: any.GetTypeUrl(),
|
|
Id: u.GetId(),
|
|
Data: any,
|
|
})
|
|
|
|
v, err = storage.ReadOne(ctx, s.opaStore, storage.MustParsePath("/databroker_data/type.googleapis.com/user.User/u1"))
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, v)
|
|
|
|
s.ClearRecords()
|
|
v, err = storage.ReadOne(ctx, s.opaStore, storage.MustParsePath("/databroker_data/type.googleapis.com/user.User/u1"))
|
|
assert.Error(t, err)
|
|
assert.Nil(t, v)
|
|
})
|
|
}
|