blob: 48ac435b92e63de0d85622f9d65f27a174c04378 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
package repository
import (
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/v14/client"
"gitlab.com/gitlab-org/gitaly/internal/gitaly/storage"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"google.golang.org/grpc/metadata"
)
func TestGetConnectionByStorage(t *testing.T) {
t.Parallel()
connPool := client.NewPool()
defer connPool.Close()
s := server{conns: connPool}
ctx := testhelper.Context(t)
storageName, address := "default", "unix:///fake/address/wont/work"
injectedCtx, err := storage.InjectGitalyServers(ctx, storageName, address, "token")
require.NoError(t, err)
md, ok := metadata.FromOutgoingContext(injectedCtx)
require.True(t, ok)
incomingCtx := metadata.NewIncomingContext(ctx, md)
_, err = s.newRepoClient(incomingCtx, storageName)
require.NoError(t, err)
}
|