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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
package server
import (
"context"
"errors"
"testing"
"github.com/google/go-cmp/cmp"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/modserver"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/modshared"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/notifications/rpc"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/tool/testing/mock_modshared"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/pkg/event"
"go.uber.org/mock/gomock"
"go.uber.org/zap"
"google.golang.org/protobuf/proto"
"google.golang.org/protobuf/testing/protocmp"
)
var (
_ modserver.Module = &module{}
_ modserver.Factory = &Factory{}
_ rpc.NotificationsServer = &server{}
)
func TestServer_GitPushEvent_SuccessfulPublish(t *testing.T) {
// GIVEN
// setup test fixtures
ctrl := gomock.NewController(t)
rpcAPI := mock_modshared.NewMockRPCAPI(ctrl)
ctx := modshared.InjectRPCAPI(context.Background(), rpcAPI)
var gotEvent proto.Message
// setup server under test
s := newServer(func(ctx context.Context, e proto.Message) error {
gotEvent = e
return nil
})
// WHEN
gitPushEvent := &event.GitPushEvent{
Project: &event.Project{Id: 42, FullPath: "foo/bar"},
}
_, err := s.GitPushEvent(ctx, &rpc.GitPushEventRequest{
Event: gitPushEvent,
})
// THEN
require.NoError(t, err)
assert.Empty(t, cmp.Diff(gotEvent, gitPushEvent, protocmp.Transform()))
}
func TestServer_GitPushEvent_FailedPublish(t *testing.T) {
// GIVEN
// setup test fixtures
ctrl := gomock.NewController(t)
rpcAPI := mock_modshared.NewMockRPCAPI(ctrl)
ctx := modshared.InjectRPCAPI(context.Background(), rpcAPI)
rpcAPI.EXPECT().Log().Return(zap.NewNop())
givenErr := errors.New("some error")
rpcAPI.EXPECT().
HandleProcessingError(gomock.Any(), modshared.NoAgentID, gomock.Any(), givenErr)
// setup server under test
s := newServer(func(ctx context.Context, e proto.Message) error {
return givenErr
})
// WHEN
_, err := s.GitPushEvent(ctx, &rpc.GitPushEventRequest{
Event: &event.GitPushEvent{
Project: &event.Project{Id: 42, FullPath: "foo/bar"},
},
})
// THEN
assert.EqualError(t, err, "rpc error: code = Unavailable desc = failed to publish received git push event: some error")
}
|
