85 lines
1.7 KiB
Go
85 lines
1.7 KiB
Go
package prism
|
|
|
|
import (
|
|
"context"
|
|
"git.wisehodl.dev/jay/go-mana-component"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
// "time"
|
|
)
|
|
|
|
// Helpers
|
|
|
|
func newTestLetter(ctx context.Context, id uint64) OutboundLetter {
|
|
ctx, cancel := context.WithCancel(
|
|
component.MustExtend(ctx, "test_letter"))
|
|
return OutboundLetter{
|
|
id: id,
|
|
peerID: "wss://test",
|
|
data: []byte("[]"),
|
|
ctx: ctx,
|
|
cancel: cancel,
|
|
}
|
|
}
|
|
|
|
// Tests
|
|
|
|
func TestCourierSendsAfterConnect(t *testing.T) {
|
|
ctx := component.MustNew(context.Background(), "prism", "test")
|
|
|
|
sent := make(chan []byte, 1)
|
|
sendFunc := func(data Envelope) error {
|
|
sent <- data
|
|
return nil
|
|
}
|
|
|
|
c := NewCourier(ctx, sendFunc, nil)
|
|
called := make(chan LetterOutcome, 1)
|
|
c.Enqueue(newTestLetter(ctx, 1), func(o LetterOutcome) { called <- o })
|
|
|
|
Never(t, func() bool { return len(sent) > 0 },
|
|
"should not have sent while disconnected")
|
|
|
|
c.HandleConnect()
|
|
|
|
Eventually(t, func() bool { return len(sent) > 0 },
|
|
"should have sent after connect")
|
|
|
|
var outcome LetterOutcome
|
|
Eventually(t, func() bool {
|
|
select {
|
|
default:
|
|
return false
|
|
case outcome = <-called:
|
|
return true
|
|
}
|
|
}, "should have returned outcome")
|
|
|
|
assert.Equal(t, uint64(1), outcome.LetterID)
|
|
assert.Equal(t, "wss://test", outcome.PeerID)
|
|
assert.Equal(t, "sent", outcome.Kind.String())
|
|
assert.False(t, outcome.SentAt.IsZero())
|
|
assert.True(t, outcome.MissedAt.IsZero())
|
|
assert.Equal(t, 0, outcome.Retries)
|
|
}
|
|
|
|
func TestCourierSequentialSends(t *testing.T) {
|
|
|
|
}
|
|
|
|
func TestCourierSkipsCancelledLetter(t *testing.T) {
|
|
|
|
}
|
|
|
|
func TestCourierRetryOnFailure(t *testing.T) {
|
|
|
|
}
|
|
|
|
func TestCourierPauseOnDisconnect(t *testing.T) {
|
|
|
|
}
|
|
|
|
func TestCourierDrainOnClose(t *testing.T) {
|
|
|
|
}
|