3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-05-18 03:27:16 +02:00
Code Issues Projects Releases Packages Wiki Activity

databroker: add support for putting multiple records (#3291)

Browse source 
* databroker: add support for putting multiple records

* add OptimumPutRequestsFromRecords function

* replace GetAll with SyncLatest

* fix stream when there are no records
This commit is contained in:
Caleb Doxsey 2022-04-26 22:41:38 +00:00 • committed by GitHub
parent 343fa43ed4
commit f73c5c615f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
28 changed files with 790 additions and 660 deletions
Download patch file Download diff file Expand all files Collapse all files

32
internal/databroker/server_test.go
View file

@ -58,19 +58,19 @@ func TestServer_Get(t *testing.T) {
s.Id = "1"
any := protoutil.NewAny(s)
_, err := srv.Put(context.Background(), &databroker.PutRequest{
Record: &databroker.Record{
Records: []*databroker.Record{{
Type: any.TypeUrl,
Id: s.Id,
Data: any,
},
}},
})
assert.NoError(t, err)
_, err = srv.Put(context.Background(), &databroker.PutRequest{
Record: &databroker.Record{
Records: []*databroker.Record{{
Type: any.TypeUrl,
Id: s.Id,
DeletedAt: timestamppb.Now(),
},
}},
})
assert.NoError(t, err)
_, err = srv.Get(context.Background(), &databroker.GetRequest{
@ -90,11 +90,11 @@ func TestServer_Options(t *testing.T) {
s.Id = "1"
any := protoutil.NewAny(s)
_, err := srv.Put(context.Background(), &databroker.PutRequest{
Record: &databroker.Record{
Records: []*databroker.Record{{
Type: any.TypeUrl,
Id: s.Id,
Data: any,
},
}},
})
assert.NoError(t, err)
_, err = srv.SetOptions(context.Background(), &databroker.SetOptionsRequest{
@ -139,11 +139,11 @@ func TestServer_Query(t *testing.T) {
s.Id = "1"
any := protoutil.NewAny(s)
_, err := srv.Put(context.Background(), &databroker.PutRequest{
Record: &databroker.Record{
Records: []*databroker.Record{{
Type: any.TypeUrl,
Id: s.Id,
Data: any,
},
}},
})
assert.NoError(t, err)
_, err = srv.Query(context.Background(), &databroker.QueryRequest{
@ -160,11 +160,11 @@ func TestServer_Sync(t *testing.T) {
s.Id = "1"
any := protoutil.NewAny(s)
_, err := srv.Put(context.Background(), &databroker.PutRequest{
Record: &databroker.Record{
Records: []*databroker.Record{{
Type: any.TypeUrl,
Id: s.Id,
Data: any,
},
}},
})
assert.NoError(t, err)
@ -216,11 +216,11 @@ func TestServer_Sync(t *testing.T) {
}
_, err = srv.Put(context.Background(), &databroker.PutRequest{
Record: &databroker.Record{
Records: []*databroker.Record{{
Type: any.TypeUrl,
Id: s.Id,
Data: any,
},
}},
})
assert.NoError(t, err)
@ -244,11 +244,11 @@ func TestServerInvalidStorage(t *testing.T) {
s.Id = "1"
any := protoutil.NewAny(s)
_, err := srv.Put(context.Background(), &databroker.PutRequest{
Record: &databroker.Record{
Records: []*databroker.Record{{
Type: any.TypeUrl,
Id: s.Id,
Data: any,
},
}},
})
_ = assert.Error(t, err) && assert.Contains(t, err.Error(), "unsupported storage type")
}
@ -265,11 +265,11 @@ func TestServerRedis(t *testing.T) {
s.Id = "1"
any := protoutil.NewAny(s)
_, err := srv.Put(context.Background(), &databroker.PutRequest{
Record: &databroker.Record{
Records: []*databroker.Record{{
Type: any.TypeUrl,
Id: s.Id,
Data: any,
},
}},
})
assert.NoError(t, err)