-
Notifications
You must be signed in to change notification settings - Fork 3
/
gitlab_app_test.go
104 lines (90 loc) · 2.49 KB
/
gitlab_app_test.go
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package probot_test
import (
"context"
"encoding/json"
"net/http"
"testing"
"time"
"github.com/h2non/gock"
"github.com/onsi/gomega"
"github.com/spf13/pflag"
"github.com/xanzy/go-gitlab"
"github.com/airconduct/go-probot"
"github.com/airconduct/go-probot/mock"
)
func TestGitLabApp(t *testing.T) {
gomega.RegisterTestingT(t)
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
f := gitlabFixture{}
f.Start(ctx)
defer f.Stop()
app := createGitLabApp()
go func() {
if err := app.Run(ctx); err != nil {
panic(err)
}
}()
gomega.Eventually(func(g gomega.Gomega) {
issueCommentEvent := gitlab.IssueCommentEvent{
ObjectKind: "note",
ProjectID: 1,
}
issueCommentEvent.ObjectAttributes.ID = 2
issueCommentEvent.ObjectAttributes.Note = "hello"
issueCommentEvent.ObjectAttributes.NoteableType = "Issue"
issueCommentEvent.Issue.ID = 3
g.Expect(mock.Send(
app.(mock.AppMock[probot.GitHubClient]),
probot.GitLab.IssueComment,
issueCommentEvent,
)).Should(gomega.Succeed())
}, 5*time.Second, time.Second).Should(gomega.Succeed())
}
func createGitLabApp() probot.App[probot.GitLabClient] {
flags := pflag.NewFlagSet("", pflag.PanicOnError)
app := probot.NewGitLabAPP()
app.AddFlags(flags)
flags.Parse([]string{
// TODO: add gitlab flags
"--gitlab.token-file=examples/github/testdata/hmac_token",
"--address=127.0.0.1",
"--port=7771",
"--path=/hook",
})
app.On(probot.GitLab.IssueComment).
WithHandler(probot.GitLab.IssueComment.Handler(func(ctx probot.GitLabIssueCommentContext) {
payload := ctx.Payload()
ctx.Logger().Info("Get IssueComment event", "payload", payload, "content", payload.ObjectAttributes.Note)
pid := payload.ProjectID
issueID := payload.Issue.ID
ctx.Must(ctx.Client().Notes.CreateIssueNote(
pid, issueID, &gitlab.CreateIssueNoteOptions{
Body: gitlab.String("Thanks!"),
},
))
}))
return app
}
type gitlabFixture struct {
}
func (f gitlabFixture) Start(ctx context.Context) {
// TODO: Add more http fixture
// Mock get token
// gock.New("https://gitlab.com").
// Get("/api/v4/").
// Reply(404)
gock.New("https://gitlab.com").
Post("/api/v4/projects/1/issues/3/notes").
AddMatcher(func(r1 *http.Request, r2 *gock.Request) (bool, error) {
data := make(map[string]string)
if err := json.NewDecoder(r1.Body).Decode(&data); err != nil {
return false, err
}
return data["body"] == "Thanks!", nil
}).
Reply(200).JSON(gitlab.Note{})
}
func (gitlabFixture) Stop() {
gock.Off()
}