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
|
package mock_modserver
import (
"context"
"testing"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/modserver"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/tool/testing/testhelpers"
"go.uber.org/zap/zaptest"
)
func IncomingAgentCtx(t *testing.T, rpcApi *MockAgentRpcApi) context.Context {
rpcApi.EXPECT().
AgentToken().
Return(testhelpers.AgentkToken).
AnyTimes()
rpcApi.EXPECT().
Log().
Return(zaptest.NewLogger(t)).
AnyTimes()
ctx, cancel := context.WithCancel(context.Background())
t.Cleanup(cancel)
ctx = modserver.InjectAgentRpcApi(ctx, rpcApi)
return ctx
}
|
