mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-30 10:56:28 +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
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package databroker
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
"google.golang.org/protobuf/types/known/anypb"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
|
|
"github.com/pomerium/pomerium/internal/log"
|
|
"github.com/pomerium/pomerium/pkg/grpc/databroker"
|
|
"github.com/pomerium/pomerium/pkg/grpc/session"
|
|
)
|
|
|
|
func newServer(cfg *serverConfig) *Server {
|
|
return &Server{
|
|
version: 11,
|
|
cfg: cfg,
|
|
log: log.With().Str("service", "databroker").Logger(),
|
|
}
|
|
}
|
|
|
|
func TestServer_Get(t *testing.T) {
|
|
cfg := newServerConfig()
|
|
t.Run("ignore deleted", func(t *testing.T) {
|
|
srv := newServer(cfg)
|
|
|
|
s := new(session.Session)
|
|
s.Id = "1"
|
|
any, err := anypb.New(s)
|
|
assert.NoError(t, err)
|
|
|
|
_, err = srv.Put(context.Background(), &databroker.PutRequest{
|
|
Record: &databroker.Record{
|
|
Type: any.TypeUrl,
|
|
Id: s.Id,
|
|
Data: any,
|
|
},
|
|
})
|
|
assert.NoError(t, err)
|
|
_, err = srv.Put(context.Background(), &databroker.PutRequest{
|
|
Record: &databroker.Record{
|
|
Type: any.TypeUrl,
|
|
Id: s.Id,
|
|
DeletedAt: timestamppb.Now(),
|
|
},
|
|
})
|
|
assert.NoError(t, err)
|
|
_, err = srv.Get(context.Background(), &databroker.GetRequest{
|
|
Type: any.TypeUrl,
|
|
Id: s.Id,
|
|
})
|
|
assert.Error(t, err)
|
|
assert.Equal(t, codes.NotFound, status.Code(err))
|
|
})
|
|
}
|