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