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
33
|
package repository
import (
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/client"
"gitlab.com/gitlab-org/gitaly/internal/helper"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"google.golang.org/grpc/metadata"
)
func TestGetConnectionByStorage(t *testing.T) {
connPool := client.NewPool()
defer connPool.Close()
s := server{conns: connPool}
ctx, cancel := testhelper.Context()
defer cancel()
storageName, address := "default", "unix://fake/address/wont/work"
injectedCtx, err := helper.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)
}
|