@@ -37,7 +37,6 @@ func TestCodereview(t *testing.T) {
3737 name string
3838 commiterr error
3939 commits []clients.Commit
40- prs []clients.PullRequest
4140 expected checker.CheckResult
4241 }{
4342 {
@@ -70,150 +69,115 @@ func TestCodereview(t *testing.T) {
7069 },
7170 {
7271 name : "Valid GitHub PR" ,
73- prs : []clients.PullRequest {
74- {
75- Number : 1 ,
76- MergedAt : time .Now (),
77- Reviews : []clients.Review {
78- {
79- State : "APPROVED" ,
80- },
81- },
82- MergeCommit : clients.Commit {
83- SHA : "sha" ,
84- },
85- },
86- },
8772 commits : []clients.Commit {
8873 {
8974 SHA : "sha" ,
9075 Committer : clients.User {
9176 Login : "user" ,
9277 },
78+ AssociatedMergeRequest : clients.PullRequest {
79+ Number : 1 ,
80+ MergedAt : time .Now (),
81+ Reviews : []clients.Review {
82+ {
83+ State : "APPROVED" ,
84+ },
85+ },
86+ },
9387 },
9488 },
95-
9689 expected : checker.CheckResult {
9790 Score : 10 ,
9891 Pass : true ,
9992 },
10093 },
10194 {
10295 name : "Valid Prow PR as not a bot" ,
103- prs : []clients.PullRequest {
104- {
105- Number : 1 ,
106- MergedAt : time .Now (),
107- Labels : []clients.Label {
108- {
109- Name : "lgtm" ,
110- },
111- },
112- MergeCommit : clients.Commit {
113- SHA : "sha" ,
114- },
115- },
116- },
11796 commits : []clients.Commit {
11897 {
11998 SHA : "sha" ,
12099 Committer : clients.User {
121100 Login : "user" ,
122101 },
102+ AssociatedMergeRequest : clients.PullRequest {
103+ Number : 1 ,
104+ MergedAt : time .Now (),
105+ Labels : []clients.Label {
106+ {
107+ Name : "lgtm" ,
108+ },
109+ },
110+ },
123111 },
124112 },
125-
126113 expected : checker.CheckResult {
127114 Score : 10 ,
128115 Pass : true ,
129116 },
130117 },
131118 {
132119 name : "Valid Prow PR and commits as bot" ,
133- prs : []clients.PullRequest {
134- {
135- Number : 1 ,
136- MergedAt : time .Now (),
137- Labels : []clients.Label {
138- {
139- Name : "lgtm" ,
140- },
141- },
142- MergeCommit : clients.Commit {
143- SHA : "sha" ,
144- },
145- },
146- },
147120 commits : []clients.Commit {
148121 {
149122 SHA : "sha" ,
150123 Committer : clients.User {
151124 Login : "bot" ,
152125 },
126+ AssociatedMergeRequest : clients.PullRequest {
127+ Number : 1 ,
128+ MergedAt : time .Now (),
129+ Labels : []clients.Label {
130+ {
131+ Name : "lgtm" ,
132+ },
133+ },
134+ },
153135 },
154136 },
155-
156137 expected : checker.CheckResult {
157138 Score : 10 ,
158139 Pass : true ,
159140 },
160141 },
161142 {
162143 name : "Valid PR's and commits with merged by someone else" ,
163- prs : []clients.PullRequest {
164- {
165- Number : 1 ,
166- Labels : []clients.Label {
167- {
168- Name : "lgtm" ,
169- },
170- },
171- MergeCommit : clients.Commit {
172- SHA : "sha" ,
173- Committer : clients.User {
174- Login : "bot" ,
175- },
176- },
177- },
178- },
179144 commits : []clients.Commit {
180145 {
181146 SHA : "sha" ,
182147 Committer : clients.User {
183148 Login : "bob" ,
184149 },
150+ AssociatedMergeRequest : clients.PullRequest {
151+ Number : 1 ,
152+ Labels : []clients.Label {
153+ {
154+ Name : "lgtm" ,
155+ },
156+ },
157+ },
185158 },
186159 },
187-
188160 expected : checker.CheckResult {
189161 Score : 0 ,
190162 },
191163 },
192164 {
193165 name : "2 PRs 2 review on GitHub" ,
194- prs : []clients.PullRequest {
195- {
196- Number : 1 ,
197- MergedAt : time .Now (),
198- Reviews : []clients.Review {
199- {
200- State : "APPROVED" ,
201- },
202- },
203- MergeCommit : clients.Commit {
204- SHA : "sha" ,
205- Committer : clients.User {
206- Login : "bot" ,
207- },
208- },
209- },
210- },
211166 commits : []clients.Commit {
212167 {
213168 SHA : "sha" ,
214169 Committer : clients.User {
215170 Login : "bob" ,
216171 },
172+ AssociatedMergeRequest : clients.PullRequest {
173+ Number : 1 ,
174+ MergedAt : time .Now (),
175+ Reviews : []clients.Review {
176+ {
177+ State : "APPROVED" ,
178+ },
179+ },
180+ },
217181 },
218182 {
219183 SHA : "sha2" ,
@@ -222,7 +186,6 @@ func TestCodereview(t *testing.T) {
222186 },
223187 },
224188 },
225-
226189 expected : checker.CheckResult {
227190 Score : 5 ,
228191 },
@@ -235,18 +198,7 @@ func TestCodereview(t *testing.T) {
235198 t .Parallel ()
236199 ctrl := gomock .NewController (t )
237200 mockRepo := mockrepo .NewMockRepoClient (ctrl )
238- mockRepo .EXPECT ().ListMergedPRs ().DoAndReturn (func () ([]clients.PullRequest , error ) {
239- if tt .err != nil {
240- return tt .prs , tt .err
241- }
242- return tt .prs , tt .err
243- }).AnyTimes ()
244- mockRepo .EXPECT ().ListCommits ().DoAndReturn (func () ([]clients.Commit , error ) {
245- if tt .commiterr != nil {
246- return tt .commits , tt .commiterr
247- }
248- return tt .commits , tt .err
249- }).AnyTimes ()
201+ mockRepo .EXPECT ().ListCommits ().Return (tt .commits , tt .err ).AnyTimes ()
250202
251203 req := checker.CheckRequest {
252204 RepoClient : mockRepo ,
0 commit comments