diff options
Diffstat (limited to 'workhorse/internal/gitaly/gitaly_test.go')
-rw-r--r-- | workhorse/internal/gitaly/gitaly_test.go | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/workhorse/internal/gitaly/gitaly_test.go b/workhorse/internal/gitaly/gitaly_test.go index b17fb5c1d7b..9c54caae8c6 100644 --- a/workhorse/internal/gitaly/gitaly_test.go +++ b/workhorse/internal/gitaly/gitaly_test.go @@ -4,14 +4,32 @@ import ( "context" "testing" + "github.com/sirupsen/logrus" "github.com/stretchr/testify/require" "google.golang.org/grpc/metadata" ) func TestNewSmartHTTPClient(t *testing.T) { - ctx, _, err := NewSmartHTTPClient(context.Background(), serverFixture()) + ctx, client, err := NewSmartHTTPClient(context.Background(), serverFixture()) require.NoError(t, err) testOutgoingMetadata(t, ctx) + + require.False(t, client.useSidechannel) + require.Nil(t, client.sidechannelRegistry) +} + +func TestNewSmartHTTPClientWithSidechannel(t *testing.T) { + InitializeSidechannelRegistry(logrus.StandardLogger()) + + fixture := serverFixture() + fixture.Sidechannel = true + + ctx, client, err := NewSmartHTTPClient(context.Background(), fixture) + require.NoError(t, err) + testOutgoingMetadata(t, ctx) + + require.True(t, client.useSidechannel) + require.NotNil(t, client.sidechannelRegistry) } func TestNewBlobClient(t *testing.T) { |