103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
package gitaly
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc/metadata"
|
|
|
|
"gitlab.com/gitlab-org/gitlab/workhorse/internal/api"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
InitializeSidechannelRegistry(logrus.StandardLogger())
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
func TestNewSmartHTTPClient(t *testing.T) {
|
|
ctx, client, err := NewSmartHTTPClient(
|
|
context.Background(),
|
|
serverFixture(),
|
|
)
|
|
require.NoError(t, err)
|
|
testOutgoingMetadata(t, ctx)
|
|
require.NotNil(t, client.sidechannelRegistry)
|
|
}
|
|
|
|
func TestNewBlobClient(t *testing.T) {
|
|
ctx, _, err := NewBlobClient(
|
|
context.Background(),
|
|
serverFixture(),
|
|
)
|
|
require.NoError(t, err)
|
|
testOutgoingMetadata(t, ctx)
|
|
}
|
|
|
|
func TestNewRepositoryClient(t *testing.T) {
|
|
ctx, _, err := NewRepositoryClient(
|
|
context.Background(),
|
|
serverFixture(),
|
|
)
|
|
|
|
require.NoError(t, err)
|
|
testOutgoingMetadata(t, ctx)
|
|
}
|
|
|
|
func TestNewNamespaceClient(t *testing.T) {
|
|
ctx, _, err := NewNamespaceClient(
|
|
context.Background(),
|
|
serverFixture(),
|
|
)
|
|
require.NoError(t, err)
|
|
testOutgoingMetadata(t, ctx)
|
|
}
|
|
|
|
func TestNewDiffClient(t *testing.T) {
|
|
ctx, _, err := NewDiffClient(
|
|
context.Background(),
|
|
serverFixture(),
|
|
)
|
|
require.NoError(t, err)
|
|
testOutgoingMetadata(t, ctx)
|
|
}
|
|
|
|
func testOutgoingMetadata(t *testing.T, ctx context.Context) {
|
|
t.Helper()
|
|
md, ok := metadata.FromOutgoingContext(ctx)
|
|
require.True(t, ok, "get metadata from context")
|
|
|
|
require.Equal(t, metadata.MD{"username": {"janedoe"}}, md)
|
|
}
|
|
|
|
func serverFixture() api.GitalyServer {
|
|
return api.GitalyServer{
|
|
Address: "tcp://localhost:123",
|
|
CallMetadata: map[string]string{"username": "janedoe"},
|
|
}
|
|
}
|
|
|
|
func TestWithOutgoingMetadata(t *testing.T) {
|
|
ctx := withOutgoingMetadata(context.Background(), api.GitalyServer{
|
|
CallMetadata: map[string]string{
|
|
"gitaly-feature-abc": "true",
|
|
"gitaly-featuregarbage": "blocked",
|
|
"bad-header": "blocked",
|
|
"user_id": "234",
|
|
"username": "janedoe",
|
|
"remote_ip": "1.2.3.4",
|
|
},
|
|
})
|
|
|
|
md, ok := metadata.FromOutgoingContext(ctx)
|
|
require.True(t, ok)
|
|
|
|
require.Equal(t, metadata.MD{
|
|
"gitaly-feature-abc": {"true"},
|
|
"user_id": {"234"},
|
|
"username": {"janedoe"},
|
|
"remote_ip": {"1.2.3.4"},
|
|
}, md)
|
|
}
|