1package gitlab
2
3import (
4 "context"
5 "fmt"
6 "os"
7 "testing"
8 "time"
9
10 "github.com/stretchr/testify/require"
11
12 "github.com/MichaelMure/git-bug/bridge/core"
13 "github.com/MichaelMure/git-bug/bridge/core/auth"
14 "github.com/MichaelMure/git-bug/bug"
15 "github.com/MichaelMure/git-bug/cache"
16 "github.com/MichaelMure/git-bug/identity"
17 "github.com/MichaelMure/git-bug/repository"
18 "github.com/MichaelMure/git-bug/util/interrupt"
19)
20
21func TestGitlabImport(t *testing.T) {
22 envToken := os.Getenv("GITLAB_API_TOKEN")
23 if envToken == "" {
24 t.Skip("Env var GITLAB_API_TOKEN missing")
25 }
26
27 projectID := os.Getenv("GITLAB_PROJECT_ID")
28 if projectID == "" {
29 t.Skip("Env var GITLAB_PROJECT_ID missing")
30 }
31
32 repo := repository.CreateGoGitTestRepo(t, false)
33
34 backend, err := cache.NewRepoCache(repo)
35 require.NoError(t, err)
36
37 defer backend.Close()
38 interrupt.RegisterCleaner(backend.Close)
39
40 author, err := identity.NewIdentity(repo, "Amine Hilaly", "hilalyamine@gmail.com")
41 require.NoError(t, err)
42
43 tests := []struct {
44 name string
45 url string
46 bug *bug.Snapshot
47 }{
48 {
49 name: "simple issue",
50 url: "https://gitlab.com/git-bug/test/-/issues/1",
51 bug: &bug.Snapshot{
52 Operations: []bug.Operation{
53 bug.NewCreateOp(author, 0, "simple issue", "initial comment", nil),
54 bug.NewAddCommentOp(author, 0, "first comment", nil),
55 bug.NewAddCommentOp(author, 0, "second comment", nil),
56 },
57 },
58 },
59 {
60 name: "empty issue",
61 url: "https://gitlab.com/git-bug/test/-/issues/2",
62 bug: &bug.Snapshot{
63 Operations: []bug.Operation{
64 bug.NewCreateOp(author, 0, "empty issue", "", nil),
65 },
66 },
67 },
68 {
69 name: "complex issue",
70 url: "https://gitlab.com/git-bug/test/-/issues/3",
71 bug: &bug.Snapshot{
72 Operations: []bug.Operation{
73 bug.NewCreateOp(author, 0, "complex issue", "initial comment", nil),
74 bug.NewAddCommentOp(author, 0, "### header\n\n**bold**\n\n_italic_\n\n> with quote\n\n`inline code`\n\n```\nmultiline code\n```\n\n- bulleted\n- list\n\n1. numbered\n1. list\n\n- [ ] task\n- [x] list\n\n@MichaelMure mention\n\n#2 reference issue\n#3 auto-reference issue", nil),
75 bug.NewSetTitleOp(author, 0, "complex issue edited", "complex issue"),
76 bug.NewSetTitleOp(author, 0, "complex issue", "complex issue edited"),
77 bug.NewSetStatusOp(author, 0, bug.ClosedStatus),
78 bug.NewSetStatusOp(author, 0, bug.OpenStatus),
79 bug.NewLabelChangeOperation(author, 0, []bug.Label{"bug"}, []bug.Label{}),
80 bug.NewLabelChangeOperation(author, 0, []bug.Label{"critical"}, []bug.Label{}),
81 bug.NewLabelChangeOperation(author, 0, []bug.Label{}, []bug.Label{"critical"}),
82 },
83 },
84 },
85 {
86 name: "editions",
87 url: "https://gitlab.com/git-bug/test/-/issues/4",
88 bug: &bug.Snapshot{
89 Operations: []bug.Operation{
90 bug.NewCreateOp(author, 0, "editions", "initial comment edited", nil),
91 bug.NewAddCommentOp(author, 0, "first comment edited", nil),
92 },
93 },
94 },
95 }
96
97 login := "test-identity"
98 author.SetMetadata(metaKeyGitlabLogin, login)
99
100 token := auth.NewToken(target, envToken)
101 token.SetMetadata(auth.MetaKeyLogin, login)
102 token.SetMetadata(auth.MetaKeyBaseURL, defaultBaseURL)
103 err = auth.Store(repo, token)
104 require.NoError(t, err)
105
106 ctx := context.Background()
107
108 importer := &gitlabImporter{}
109 err = importer.Init(ctx, backend, core.Configuration{
110 confKeyProjectID: projectID,
111 confKeyGitlabBaseUrl: defaultBaseURL,
112 confKeyDefaultLogin: login,
113 })
114 require.NoError(t, err)
115
116 start := time.Now()
117
118 events, err := importer.ImportAll(ctx, backend, time.Time{})
119 require.NoError(t, err)
120
121 for result := range events {
122 require.NoError(t, result.Err)
123 }
124
125 fmt.Printf("test repository imported in %f seconds\n", time.Since(start).Seconds())
126
127 require.Len(t, backend.AllBugsIds(), len(tests))
128
129 for _, tt := range tests {
130 t.Run(tt.name, func(t *testing.T) {
131 b, err := backend.ResolveBugCreateMetadata(metaKeyGitlabUrl, tt.url)
132 require.NoError(t, err)
133
134 ops := b.Snapshot().Operations
135 require.Len(t, tt.bug.Operations, len(ops))
136
137 for i, op := range tt.bug.Operations {
138
139 require.IsType(t, ops[i], op)
140 require.Equal(t, op.Author().Name(), ops[i].Author().Name())
141
142 switch op := op.(type) {
143 case *bug.CreateOperation:
144 require.Equal(t, op.Title, ops[i].(*bug.CreateOperation).Title)
145 require.Equal(t, op.Message, ops[i].(*bug.CreateOperation).Message)
146 case *bug.SetStatusOperation:
147 require.Equal(t, op.Status, ops[i].(*bug.SetStatusOperation).Status)
148 case *bug.SetTitleOperation:
149 require.Equal(t, op.Was, ops[i].(*bug.SetTitleOperation).Was)
150 require.Equal(t, op.Title, ops[i].(*bug.SetTitleOperation).Title)
151 case *bug.LabelChangeOperation:
152 require.ElementsMatch(t, op.Added, ops[i].(*bug.LabelChangeOperation).Added)
153 require.ElementsMatch(t, op.Removed, ops[i].(*bug.LabelChangeOperation).Removed)
154 case *bug.AddCommentOperation:
155 require.Equal(t, op.Message, ops[i].(*bug.AddCommentOperation).Message)
156 case *bug.EditCommentOperation:
157 require.Equal(t, op.Message, ops[i].(*bug.EditCommentOperation).Message)
158
159 default:
160 panic("unknown operation type")
161 }
162 }
163 })
164 }
165}