feat: complete cmpp platform phases 0-5
This commit is contained in:
@@ -0,0 +1,47 @@
|
||||
package connection
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestReconnectorRetriesAfterDisconnect(t *testing.T) {
|
||||
failures := 0
|
||||
reconnector := Reconnector{
|
||||
MaxAttempts: 3,
|
||||
Dial: func(context.Context) error {
|
||||
failures++
|
||||
if failures < 2 {
|
||||
return errors.New("simulated disconnect")
|
||||
}
|
||||
return nil
|
||||
},
|
||||
}
|
||||
|
||||
attempts, err := reconnector.Connect(context.Background())
|
||||
if err != nil {
|
||||
t.Fatalf("connect after retry: %v", err)
|
||||
}
|
||||
if attempts != 2 {
|
||||
t.Fatalf("expected success on second attempt, got %d", attempts)
|
||||
}
|
||||
}
|
||||
|
||||
func TestReconnectorReturnsLastError(t *testing.T) {
|
||||
expected := errors.New("still disconnected")
|
||||
reconnector := Reconnector{
|
||||
MaxAttempts: 2,
|
||||
Dial: func(context.Context) error {
|
||||
return expected
|
||||
},
|
||||
}
|
||||
|
||||
attempts, err := reconnector.Connect(context.Background())
|
||||
if !errors.Is(err, expected) {
|
||||
t.Fatalf("expected last error, got %v", err)
|
||||
}
|
||||
if attempts != 2 {
|
||||
t.Fatalf("expected two attempts, got %d", attempts)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user