chore: t.Parallel() + move poll test to own file
This commit is contained in:
parent
231423cd0b
commit
0bdd3dfef8
5 changed files with 77 additions and 48 deletions
48
poll_single_test.go
Normal file
48
poll_single_test.go
Normal file
|
@ -0,0 +1,48 @@
|
|||
package k2v_test
|
||||
|
||||
import (
|
||||
k2v "code.notaphish.fyi/milas/garage-k2v-go"
|
||||
"github.com/stretchr/testify/require"
|
||||
"net/http/httptrace"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestClient_PollItem(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
f, ctx := newFixture(t)
|
||||
|
||||
pk := randomKey()
|
||||
sk := randomKey()
|
||||
|
||||
err := f.cli.InsertItem(ctx, f.bucket, pk, sk, "", []byte("hello1"))
|
||||
require.NoError(t, err)
|
||||
|
||||
_, ct, err := f.cli.ReadItemSingle(ctx, f.bucket, pk, sk)
|
||||
|
||||
updateErrCh := make(chan error, 1)
|
||||
pollReadyCh := make(chan struct{})
|
||||
go func(ct k2v.CausalityToken) {
|
||||
defer close(updateErrCh)
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
t.Errorf("Context canceled: %v", ctx.Err())
|
||||
return
|
||||
case <-pollReadyCh:
|
||||
t.Logf("PollItem connected")
|
||||
}
|
||||
updateErrCh <- f.cli.InsertItem(ctx, f.bucket, pk, sk, ct, []byte("hello2"))
|
||||
}(ct)
|
||||
|
||||
trace := &httptrace.ClientTrace{
|
||||
WroteRequest: func(_ httptrace.WroteRequestInfo) {
|
||||
pollReadyCh <- struct{}{}
|
||||
},
|
||||
}
|
||||
item, ct, err := f.cli.PollItem(httptrace.WithClientTrace(ctx, trace), f.bucket, pk, sk, ct, 5*time.Second)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, "hello2", string(item))
|
||||
require.NotEmpty(t, ct)
|
||||
require.NoError(t, <-updateErrCh)
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue