diff --git a/helpers_test.go b/helpers_test.go index 2c112b7..5995a88 100644 --- a/helpers_test.go +++ b/helpers_test.go @@ -133,3 +133,16 @@ func newMockSessionHarness() *mockSessionHarness { terminate: terminate, } } + +// MockEnvoy + +func newMockEnvoy(t *testing.T) (*mockPool, *Envoy) { + t.Helper() + + p := newMockPool(t) + emb := NewEmbassy(p.ctx, p.plugin, nil) + err := emb.Dispatch(p.url) + assert.NoError(t, err) + envoy := emb.Call(p.url) + return p, envoy +} diff --git a/request_test.go b/request_test.go index 1a814e1..d22b4fb 100644 --- a/request_test.go +++ b/request_test.go @@ -208,11 +208,7 @@ func TestRequestManager_Session(t *testing.T) { func TestRequestManager_Stream(t *testing.T) { t.Run("spawns session and sends req when connected", func(t *testing.T) { - p := newMockPool(t) - emb := NewEmbassy(p.ctx, p.plugin, nil) - err := emb.Dispatch(p.url) - assert.NoError(t, err) - envoy := emb.Call(p.url) + p, envoy := newMockEnvoy(t) p.connect() Eventually(t, envoy.IsConnected, "envoy should be connected") @@ -239,11 +235,7 @@ func TestRequestManager_Stream(t *testing.T) { }) t.Run("registers but does not spawn session when disconnected", func(t *testing.T) { - p := newMockPool(t) - emb := NewEmbassy(p.ctx, p.plugin, nil) - err := emb.Dispatch(p.url) - assert.NoError(t, err) - envoy := emb.Call(p.url) + p, envoy := newMockEnvoy(t) m := NewRequestManager(envoy) filters := [][]byte{[]byte(`{}`)}