1// SPDX-License-Identifier: MIT
2
3package api_test
4
5import (
6 "errors"
7 "testing"
8
9 "connectrpc.com/connect"
10
11 "google.golang.org/protobuf/proto"
12 "google.golang.org/protobuf/reflect/protoreflect"
13 "google.golang.org/protobuf/types/known/fieldmaskpb"
14
15 "bnema.dev/ermite-api/gen/ermite"
16 "bnema.dev/ermite-api/gen/ermite/ermiteconnect"
17)
18
19// TestProcedures pins the procedure paths: they are the wire contract.
20func TestProcedures(t *testing.T) {
21 tests := []struct{ got, want string }{
22 {ermiteconnect.MirrorServiceGetMirrorKeyProcedure, "/ermite.MirrorService/GetMirrorKey"},
23 {ermiteconnect.MirrorServiceRotateMirrorKeyProcedure, "/ermite.MirrorService/RotateMirrorKey"},
24 {ermiteconnect.MirrorServiceGetMirrorConfigProcedure, "/ermite.MirrorService/GetMirrorConfig"},
25 {ermiteconnect.MirrorServiceSetMirrorConfigProcedure, "/ermite.MirrorService/SetMirrorConfig"},
26 {ermiteconnect.MirrorServiceSetRepoMirrorProcedure, "/ermite.MirrorService/SetRepoMirror"},
27 {ermiteconnect.MirrorServiceRemoveRepoMirrorProcedure, "/ermite.MirrorService/RemoveRepoMirror"},
28 {ermiteconnect.MirrorServiceSetRepoMirrorExcludedProcedure, "/ermite.MirrorService/SetRepoMirrorExcluded"},
29 {ermiteconnect.MirrorServiceGetMirrorStatusProcedure, "/ermite.MirrorService/GetMirrorStatus"},
30 {ermiteconnect.MirrorServiceListMirrorStatusProcedure, "/ermite.MirrorService/ListMirrorStatus"},
31 {ermiteconnect.MirrorServiceSyncMirrorProcedure, "/ermite.MirrorService/SyncMirror"},
32
33 {ermiteconnect.RepoServiceListReposProcedure, "/ermite.RepoService/ListRepos"},
34 {ermiteconnect.RepoServiceGetRepoProcedure, "/ermite.RepoService/GetRepo"},
35 {ermiteconnect.RepoServiceCreateRepoProcedure, "/ermite.RepoService/CreateRepo"},
36 {ermiteconnect.RepoServiceUpdateRepoProcedure, "/ermite.RepoService/UpdateRepo"},
37 {ermiteconnect.RepoServiceDeleteRepoProcedure, "/ermite.RepoService/DeleteRepo"},
38 {ermiteconnect.GitServiceListRefsProcedure, "/ermite.GitService/ListRefs"},
39 {ermiteconnect.GitServiceGetTreeProcedure, "/ermite.GitService/GetTree"},
40 {ermiteconnect.GitServiceListPathsProcedure, "/ermite.GitService/ListPaths"},
41 {ermiteconnect.GitServiceGetBlobProcedure, "/ermite.GitService/GetBlob"},
42 {ermiteconnect.GitServiceListCommitsProcedure, "/ermite.GitService/ListCommits"},
43 {ermiteconnect.GitServiceGetCommitProcedure, "/ermite.GitService/GetCommit"},
44 {ermiteconnect.GitServiceResolveRevisionProcedure, "/ermite.GitService/ResolveRevision"},
45 {ermiteconnect.TokenServiceCreateTokenProcedure, "/ermite.TokenService/CreateToken"},
46 {ermiteconnect.TokenServiceListTokensProcedure, "/ermite.TokenService/ListTokens"},
47 {ermiteconnect.TokenServiceRevokeTokenProcedure, "/ermite.TokenService/RevokeToken"},
48 {ermiteconnect.EventServiceWatchProcedure, "/ermite.EventService/Watch"},
49 {ermiteconnect.MergeRequestServiceCreateMergeRequestProcedure, "/ermite.MergeRequestService/CreateMergeRequest"},
50 {ermiteconnect.MergeRequestServiceListMergeRequestsProcedure, "/ermite.MergeRequestService/ListMergeRequests"},
51 {ermiteconnect.MergeRequestServiceGetMergeRequestProcedure, "/ermite.MergeRequestService/GetMergeRequest"},
52 {ermiteconnect.MergeRequestServiceAddCommentProcedure, "/ermite.MergeRequestService/AddComment"},
53 {ermiteconnect.MergeRequestServiceResolveCommentProcedure, "/ermite.MergeRequestService/ResolveComment"},
54 {ermiteconnect.MergeRequestServiceCloseMergeRequestProcedure, "/ermite.MergeRequestService/CloseMergeRequest"},
55 {ermiteconnect.MergeRequestServiceMergeMergeRequestProcedure, "/ermite.MergeRequestService/MergeMergeRequest"},
56 {ermiteconnect.MergeRequestServiceMarkReadProcedure, "/ermite.MergeRequestService/MarkRead"},
57 {ermiteconnect.MergeRequestServiceEnableAutoMergeProcedure, "/ermite.MergeRequestService/EnableAutoMerge"},
58 {ermiteconnect.MergeRequestServiceCancelAutoMergeProcedure, "/ermite.MergeRequestService/CancelAutoMerge"},
59 {ermiteconnect.SessionServiceCreateLoginLinkProcedure, "/ermite.SessionService/CreateLoginLink"},
60 {ermiteconnect.SessionServiceSignOutEverywhereProcedure, "/ermite.SessionService/SignOutEverywhere"},
61 {ermiteconnect.ServerServiceGetVersionProcedure, "/ermite.ServerService/GetVersion"},
62 {ermiteconnect.ServerServiceGetSigningKeyProcedure, "/ermite.ServerService/GetSigningKey"},
63 {ermiteconnect.ServerServiceRotateOIDCKeyProcedure, "/ermite.ServerService/RotateOIDCKey"},
64 {ermiteconnect.RunnerServiceRegisterRunnerProcedure, "/ermite.RunnerService/RegisterRunner"},
65 {ermiteconnect.RunnerServiceFetchJobProcedure, "/ermite.RunnerService/FetchJob"},
66 {ermiteconnect.RunnerServiceUpdateJobProcedure, "/ermite.RunnerService/UpdateJob"},
67 {ermiteconnect.RunnerServiceAppendLogProcedure, "/ermite.RunnerService/AppendLog"},
68 {ermiteconnect.RunnerServiceHeartbeatProcedure, "/ermite.RunnerService/Heartbeat"},
69 {ermiteconnect.RunnerServiceWatchJobProcedure, "/ermite.RunnerService/WatchJob"},
70 {ermiteconnect.RunnerServiceCompleteJobProcedure, "/ermite.RunnerService/CompleteJob"},
71 {ermiteconnect.RunnerServiceResolveActionsProcedure, "/ermite.RunnerService/ResolveActions"},
72 {ermiteconnect.RunnerServiceUploadSummaryProcedure, "/ermite.RunnerService/UploadSummary"},
73 {ermiteconnect.RunServiceListRunsProcedure, "/ermite.RunService/ListRuns"},
74 {ermiteconnect.RunServiceGetRunProcedure, "/ermite.RunService/GetRun"},
75 {ermiteconnect.RunServiceGetJobProcedure, "/ermite.RunService/GetJob"},
76 {ermiteconnect.RunServiceGetJobLogProcedure, "/ermite.RunService/GetJobLog"},
77 {ermiteconnect.RunServiceListAnnotationsProcedure, "/ermite.RunService/ListAnnotations"},
78 {ermiteconnect.RunServiceGetStepSummariesProcedure, "/ermite.RunService/GetStepSummaries"},
79 {ermiteconnect.RunServiceCancelRunProcedure, "/ermite.RunService/CancelRun"},
80 {ermiteconnect.RunServiceRerunRunProcedure, "/ermite.RunService/RerunRun"},
81 {ermiteconnect.RunServiceDispatchWorkflowProcedure, "/ermite.RunService/DispatchWorkflow"},
82 {ermiteconnect.RunServiceListWorkflowsProcedure, "/ermite.RunService/ListWorkflows"},
83 {ermiteconnect.RunServiceDisableWorkflowProcedure, "/ermite.RunService/DisableWorkflow"},
84 {ermiteconnect.RunServiceEnableWorkflowProcedure, "/ermite.RunService/EnableWorkflow"},
85 {ermiteconnect.RunServiceListChecksProcedure, "/ermite.RunService/ListChecks"},
86 {ermiteconnect.RunServiceListRunnersProcedure, "/ermite.RunService/ListRunners"},
87 {ermiteconnect.RunServiceRemoveRunnerProcedure, "/ermite.RunService/RemoveRunner"},
88 {ermiteconnect.SecretServiceSetSecretProcedure, "/ermite.SecretService/SetSecret"},
89 {ermiteconnect.SecretServiceListSecretsProcedure, "/ermite.SecretService/ListSecrets"},
90 {ermiteconnect.SecretServiceDeleteSecretProcedure, "/ermite.SecretService/DeleteSecret"},
91 {ermiteconnect.VariableServiceSetVariableProcedure, "/ermite.VariableService/SetVariable"},
92 {ermiteconnect.VariableServiceGetVariableProcedure, "/ermite.VariableService/GetVariable"},
93 {ermiteconnect.VariableServiceListVariablesProcedure, "/ermite.VariableService/ListVariables"},
94 {ermiteconnect.VariableServiceDeleteVariableProcedure, "/ermite.VariableService/DeleteVariable"},
95 {ermiteconnect.EnvironmentServiceCreateEnvironmentProcedure, "/ermite.EnvironmentService/CreateEnvironment"},
96 {ermiteconnect.EnvironmentServiceListEnvironmentsProcedure, "/ermite.EnvironmentService/ListEnvironments"},
97 {ermiteconnect.EnvironmentServiceDeleteEnvironmentProcedure, "/ermite.EnvironmentService/DeleteEnvironment"},
98 {ermiteconnect.JobArtifactServiceCreateJobArtifactProcedure, "/ermite.JobArtifactService/CreateJobArtifact"},
99 {ermiteconnect.JobArtifactServiceUploadJobArtifactBlockProcedure, "/ermite.JobArtifactService/UploadJobArtifactBlock"},
100 {ermiteconnect.JobArtifactServiceCommitJobArtifactProcedure, "/ermite.JobArtifactService/CommitJobArtifact"},
101 {ermiteconnect.JobArtifactServiceFinalizeJobArtifactProcedure, "/ermite.JobArtifactService/FinalizeJobArtifact"},
102 {ermiteconnect.JobArtifactServiceListJobArtifactsProcedure, "/ermite.JobArtifactService/ListJobArtifacts"},
103 {ermiteconnect.JobArtifactServiceDownloadJobArtifactProcedure, "/ermite.JobArtifactService/DownloadJobArtifact"},
104 {ermiteconnect.JobArtifactServiceDeleteJobArtifactProcedure, "/ermite.JobArtifactService/DeleteJobArtifact"},
105 {ermiteconnect.ArtifactServiceListArtifactsProcedure, "/ermite.ArtifactService/ListArtifacts"},
106 {ermiteconnect.ArtifactServiceDownloadArtifactProcedure, "/ermite.ArtifactService/DownloadArtifact"},
107 {ermiteconnect.ArtifactServiceDeleteArtifactProcedure, "/ermite.ArtifactService/DeleteArtifact"},
108 }
109 for _, tt := range tests {
110 if tt.got != tt.want {
111 t.Errorf("procedure = %q, want %q", tt.got, tt.want)
112 }
113 }
114}
115
116// TestEnumNumbers pins enum numbers: they are part of the wire contract.
117func TestEnumNumbers(t *testing.T) {
118 if got := ermite.Scope_SCOPE_TOKEN_ADMIN; got != 9 {
119 t.Errorf("SCOPE_TOKEN_ADMIN = %d, want 9", got)
120 }
121 if got := ermite.FileStatus_FILE_STATUS_TYPE_CHANGED; got != 6 {
122 t.Errorf("FILE_STATUS_TYPE_CHANGED = %d, want 6", got)
123 }
124 if got := ermite.Visibility_VISIBILITY_PRIVATE; got != 2 {
125 t.Errorf("VISIBILITY_PRIVATE = %d, want 2", got)
126 }
127 if got := ermite.MergeMode_MERGE_MODE_SQUASH; got != 2 {
128 t.Errorf("MERGE_MODE_SQUASH = %d, want 2", got)
129 }
130 if got := ermite.MergeMode_MERGE_MODE_MERGE_COMMIT; got != 3 {
131 t.Errorf("MERGE_MODE_MERGE_COMMIT = %d, want 3", got)
132 }
133 if got := ermite.MergeabilityState_MERGEABILITY_STATE_CONFLICT; got != 4 {
134 t.Errorf("MERGEABILITY_STATE_CONFLICT = %d, want 4", got)
135 }
136 if got := ermite.MergeRequestState_MERGE_REQUEST_STATE_CLOSED; got != 3 {
137 t.Errorf("MERGE_REQUEST_STATE_CLOSED = %d, want 3", got)
138 }
139 if got := ermite.DiffSide_DIFF_SIDE_NEW; got != 2 {
140 t.Errorf("DIFF_SIDE_NEW = %d, want 2", got)
141 }
142 enums := []struct {
143 name string
144 got int32
145 want int32
146 }{
147 {"CI_STATUS_UNSPECIFIED", int32(ermite.CiStatus_CI_STATUS_UNSPECIFIED), 0},
148 {"CI_STATUS_QUEUED", int32(ermite.CiStatus_CI_STATUS_QUEUED), 1},
149 {"CI_STATUS_IN_PROGRESS", int32(ermite.CiStatus_CI_STATUS_IN_PROGRESS), 2},
150 {"CI_STATUS_COMPLETED", int32(ermite.CiStatus_CI_STATUS_COMPLETED), 3},
151 {"CI_STATUS_PENDING", int32(ermite.CiStatus_CI_STATUS_PENDING), 4},
152 {"CI_CONCLUSION_UNSPECIFIED", int32(ermite.CiConclusion_CI_CONCLUSION_UNSPECIFIED), 0},
153 {"CI_CONCLUSION_SUCCESS", int32(ermite.CiConclusion_CI_CONCLUSION_SUCCESS), 1},
154 {"CI_CONCLUSION_FAILURE", int32(ermite.CiConclusion_CI_CONCLUSION_FAILURE), 2},
155 {"CI_CONCLUSION_CANCELLED", int32(ermite.CiConclusion_CI_CONCLUSION_CANCELLED), 3},
156 {"CI_CONCLUSION_SKIPPED", int32(ermite.CiConclusion_CI_CONCLUSION_SKIPPED), 4},
157 {"CI_CONCLUSION_TIMED_OUT", int32(ermite.CiConclusion_CI_CONCLUSION_TIMED_OUT), 5},
158 {"CI_CONCLUSION_NEUTRAL", int32(ermite.CiConclusion_CI_CONCLUSION_NEUTRAL), 6},
159 {"CI_CONCLUSION_ACTION_REQUIRED", int32(ermite.CiConclusion_CI_CONCLUSION_ACTION_REQUIRED), 7},
160 {"STEP_STATE_UNSPECIFIED", int32(ermite.StepState_STEP_STATE_UNSPECIFIED), 0},
161 {"STEP_STATE_PENDING", int32(ermite.StepState_STEP_STATE_PENDING), 1},
162 {"STEP_STATE_IN_PROGRESS", int32(ermite.StepState_STEP_STATE_IN_PROGRESS), 2},
163 {"STEP_STATE_COMPLETED", int32(ermite.StepState_STEP_STATE_COMPLETED), 3},
164 {"RUNNER_ISOLATION_UNSPECIFIED", int32(ermite.RunnerIsolation_RUNNER_ISOLATION_UNSPECIFIED), 0},
165 {"RUNNER_ISOLATION_VM", int32(ermite.RunnerIsolation_RUNNER_ISOLATION_VM), 1},
166 {"RUNNER_ISOLATION_CONTAINER_SYSBOX", int32(ermite.RunnerIsolation_RUNNER_ISOLATION_CONTAINER_SYSBOX), 2},
167 {"RUNNER_ISOLATION_CONTAINER_PRIVILEGED", int32(ermite.RunnerIsolation_RUNNER_ISOLATION_CONTAINER_PRIVILEGED), 3},
168 {"CANCEL_REASON_UNSPECIFIED", int32(ermite.CancelReason_CANCEL_REASON_UNSPECIFIED), 0},
169 {"CANCEL_REASON_USER", int32(ermite.CancelReason_CANCEL_REASON_USER), 1},
170 {"CANCEL_REASON_FAIL_FAST", int32(ermite.CancelReason_CANCEL_REASON_FAIL_FAST), 2},
171 {"CANCEL_REASON_TIMEOUT", int32(ermite.CancelReason_CANCEL_REASON_TIMEOUT), 3},
172 {"ANNOTATION_LEVEL_UNSPECIFIED", int32(ermite.AnnotationLevel_ANNOTATION_LEVEL_UNSPECIFIED), 0},
173 {"ANNOTATION_LEVEL_NOTICE", int32(ermite.AnnotationLevel_ANNOTATION_LEVEL_NOTICE), 1},
174 {"ANNOTATION_LEVEL_WARNING", int32(ermite.AnnotationLevel_ANNOTATION_LEVEL_WARNING), 2},
175 {"ANNOTATION_LEVEL_FAILURE", int32(ermite.AnnotationLevel_ANNOTATION_LEVEL_FAILURE), 3},
176 {"ACTION_SOURCE_UNSPECIFIED", int32(ermite.ActionSource_ACTION_SOURCE_UNSPECIFIED), 0},
177 {"ACTION_SOURCE_GITHUB", int32(ermite.ActionSource_ACTION_SOURCE_GITHUB), 1},
178 {"ACTION_SOURCE_ERMITE", int32(ermite.ActionSource_ACTION_SOURCE_ERMITE), 2},
179 }
180 for _, e := range enums {
181 if e.got != e.want {
182 t.Errorf("%s = %d, want %d", e.name, e.got, e.want)
183 }
184 }
185}
186
187// TestEventRoundTrip checks that an event with a oneof kind survives the wire.
188func TestEventRoundTrip(t *testing.T) {
189 in := &ermite.WatchResponse{Event: &ermite.Event{
190 Id: "42",
191 Kind: &ermite.Event_Pushed{Pushed: &ermite.Pushed{
192 Repo: "ermite",
193 RefUpdates: []*ermite.RefUpdate{
194 {Ref: "refs/heads/main", OldId: "a", NewId: "b"},
195 },
196 }},
197 }}
198 b, err := proto.Marshal(in)
199 if err != nil {
200 t.Fatal(err)
201 }
202 var out ermite.WatchResponse
203 if err := proto.Unmarshal(b, &out); err != nil {
204 t.Fatal(err)
205 }
206 if !proto.Equal(in, &out) {
207 t.Errorf("round trip mismatch: %v != %v", &out, in)
208 }
209 if out.GetEvent().GetPushed().GetRefUpdates()[0].GetNewId() != "b" {
210 t.Errorf("unexpected ref update: %v", &out)
211 }
212}
213
214// TestUpdateRepoMask checks that the request carries a field mask.
215func TestUpdateRepoMask(t *testing.T) {
216 req := &ermite.UpdateRepoRequest{
217 Repo: "ermite",
218 Visibility: ermite.Visibility_VISIBILITY_PUBLIC,
219 UpdateMask: &fieldmaskpb.FieldMask{Paths: []string{"visibility"}},
220 }
221 if got := req.GetUpdateMask().GetPaths(); len(got) != 1 || got[0] != "visibility" {
222 t.Errorf("paths = %v", got)
223 }
224}
225
226// TestMergeRequestEventRoundTrip checks the merge request event kinds and the
227// repository settings survive the wire.
228func TestMergeRequestEventRoundTrip(t *testing.T) {
229 in := &ermite.WatchResponse{Event: &ermite.Event{
230 Id: "43",
231 Kind: &ermite.Event_CommentAdded{CommentAdded: &ermite.CommentAdded{
232 Repo: "ermite",
233 Number: 7,
234 Comment: &ermite.Comment{
235 Id: "c1", Path: "a.go", Line: 3, Side: ermite.DiffSide_DIFF_SIDE_OLD, Text: "hi",
236 },
237 }},
238 }}
239 b, err := proto.Marshal(in)
240 if err != nil {
241 t.Fatal(err)
242 }
243 var out ermite.WatchResponse
244 if err := proto.Unmarshal(b, &out); err != nil {
245 t.Fatal(err)
246 }
247 if !proto.Equal(in, &out) {
248 t.Errorf("round trip mismatch: %v != %v", &out, in)
249 }
250 if got := out.GetEvent().GetCommentAdded().GetNumber(); got != 7 {
251 t.Errorf("number = %d, want 7", got)
252 }
253
254 req := &ermite.UpdateRepoRequest{
255 Repo: "ermite",
256 Settings: &ermite.RepoSettings{
257 AllowedMergeModes: []ermite.MergeMode{ermite.MergeMode_MERGE_MODE_SQUASH},
258 },
259 UpdateMask: &fieldmaskpb.FieldMask{Paths: []string{"settings.allowed_merge_modes"}},
260 }
261 if got := req.GetSettings().GetAllowedMergeModes(); len(got) != 1 || got[0] != ermite.MergeMode_MERGE_MODE_SQUASH {
262 t.Errorf("modes = %v", got)
263 }
264}
265
266// TestCIEventRoundTrip checks that the CI event kinds keep their field numbers
267// and survive the wire.
268func TestCIEventRoundTrip(t *testing.T) {
269 in := &ermite.WatchResponse{Event: &ermite.Event{
270 Id: "44",
271 Kind: &ermite.Event_JobUpdated{JobUpdated: &ermite.JobUpdated{Job: &ermite.RunJob{
272 Id: "j1",
273 RunId: 9,
274 Attempt: 2,
275 Status: ermite.CiStatus_CI_STATUS_COMPLETED,
276 Conclusion: ermite.CiConclusion_CI_CONCLUSION_TIMED_OUT,
277 Isolation: ermite.RunnerIsolation_RUNNER_ISOLATION_VM,
278 Steps: []*ermite.Step{{Number: 1, Name: "Set up job", LogStart: 0, LogEnd: 3}},
279 }}},
280 }}
281 b, err := proto.Marshal(in)
282 if err != nil {
283 t.Fatal(err)
284 }
285 var out ermite.WatchResponse
286 if err := proto.Unmarshal(b, &out); err != nil {
287 t.Fatal(err)
288 }
289 if !proto.Equal(in, &out) {
290 t.Errorf("round trip mismatch: %v != %v", &out, in)
291 }
292 f := (&ermite.Event{}).ProtoReflect().Descriptor().Fields()
293 for name, num := range map[string]int{"run_updated": 12, "job_updated": 13} {
294 if fd := f.ByName(protoreflect.Name(name)); fd == nil || int(fd.Number()) != num {
295 t.Errorf("event field %s: %v, want number %d", name, fd, num)
296 }
297 }
298}
299
300// TestJobRoundTrip checks the job handed to a runner survives the wire.
301func TestJobRoundTrip(t *testing.T) {
302 in := &ermite.FetchJobResponse{Job: &ermite.Job{
303 JobId: "j1", RunId: 9, Attempt: 1, JobToken: "tok", WorkflowPath: ".github/workflows/ci.yml",
304 WorkflowYaml: "on: push", JobKey: "build", Matrix: "{}", Contexts: "{}",
305 Labels: []string{"self-hosted"}, TimeoutMinutes: 360, Repo: "ermite",
306 Sha: "0123456789012345678901234567890123456789", Ref: "refs/heads/main",
307 }}
308 b, err := proto.Marshal(in)
309 if err != nil {
310 t.Fatal(err)
311 }
312 var out ermite.FetchJobResponse
313 if err := proto.Unmarshal(b, &out); err != nil {
314 t.Fatal(err)
315 }
316 if !proto.Equal(in, &out) {
317 t.Errorf("round trip mismatch: %v != %v", &out, in)
318 }
319}
320
321// TestWatchJobRoundTrip checks the three kinds of WatchJob message and the
322// JobEnded detail.
323func TestWatchJobRoundTrip(t *testing.T) {
324 msgs := []*ermite.WatchJobResponse{
325 {Msg: &ermite.WatchJobResponse_Keepalive{Keepalive: &ermite.Keepalive{}}},
326 {Msg: &ermite.WatchJobResponse_Cancel{Cancel: &ermite.CancelRequest{
327 Reason: ermite.CancelReason_CANCEL_REASON_FAIL_FAST, Message: "m",
328 }}},
329 {Msg: &ermite.WatchJobResponse_Ended{Ended: &ermite.JobEnded{
330 Conclusion: ermite.CiConclusion_CI_CONCLUSION_CANCELLED, Error: "e",
331 }}},
332 }
333 for _, in := range msgs {
334 b, err := proto.Marshal(in)
335 if err != nil {
336 t.Fatal(err)
337 }
338 var out ermite.WatchJobResponse
339 if err := proto.Unmarshal(b, &out); err != nil {
340 t.Fatal(err)
341 }
342 if !proto.Equal(in, &out) {
343 t.Errorf("round trip mismatch: %v != %v", &out, in)
344 }
345 }
346 // The detail travels in a Connect error.
347 d, err := connect.NewErrorDetail(&ermite.JobEnded{Conclusion: ermite.CiConclusion_CI_CONCLUSION_TIMED_OUT})
348 if err != nil {
349 t.Fatal(err)
350 }
351 cerr := connect.NewError(connect.CodeFailedPrecondition, errors.New("job ended"))
352 cerr.AddDetail(d)
353 v, err := cerr.Details()[0].Value()
354 if err != nil {
355 t.Fatal(err)
356 }
357 if got := v.(*ermite.JobEnded).GetConclusion(); got != ermite.CiConclusion_CI_CONCLUSION_TIMED_OUT {
358 t.Errorf("detail conclusion = %v", got)
359 }
360}
361
362// TestCompleteJobRoundTrip checks the final report of a job and its answer.
363func TestCompleteJobRoundTrip(t *testing.T) {
364 in := &ermite.CompleteJobRequest{
365 JobId: "j1",
366 Conclusion: ermite.CiConclusion_CI_CONCLUSION_CANCELLED,
367 Outputs: map[string]string{"a": "b"},
368 RunnerShutdown: true,
369 Steps: []*ermite.StepUpdate{{Number: 2, State: ermite.StepState_STEP_STATE_COMPLETED}},
370 }
371 b, err := proto.Marshal(in)
372 if err != nil {
373 t.Fatal(err)
374 }
375 var out ermite.CompleteJobRequest
376 if err := proto.Unmarshal(b, &out); err != nil {
377 t.Fatal(err)
378 }
379 if !proto.Equal(in, &out) {
380 t.Errorf("round trip mismatch: %v != %v", &out, in)
381 }
382 r := &ermite.CompleteJobResponse{Applied: false, Conclusion: ermite.CiConclusion_CI_CONCLUSION_TIMED_OUT}
383 if r.GetApplied() || r.GetConclusion() != ermite.CiConclusion_CI_CONCLUSION_TIMED_OUT {
384 t.Errorf("unexpected response %v", r)
385 }
386}
387
388// TestRunNameField pins the number of Run.name, the GitHub name of the run
389// that sits next to workflow_name, and that it survives the wire.
390func TestRunNameField(t *testing.T) {
391 fd := (&ermite.Run{}).ProtoReflect().Descriptor().Fields().ByName("name")
392 if fd == nil || fd.Number() != 19 {
393 t.Fatalf("Run.name = %v, want field 19", fd)
394 }
395 in := &ermite.Run{WorkflowName: "Names Workflow", Name: "run push on branch by someone", Title: "run push on branch by someone"}
396 b, err := proto.Marshal(in)
397 if err != nil {
398 t.Fatal(err)
399 }
400 var out ermite.Run
401 if err := proto.Unmarshal(b, &out); err != nil {
402 t.Fatal(err)
403 }
404 if out.GetName() != in.GetName() || out.GetWorkflowName() != "Names Workflow" {
405 t.Errorf("round trip: name %q, workflow_name %q", out.GetName(), out.GetWorkflowName())
406 }
407}
408
409// TestCI2FieldNumbers pins the numbers of the fields added for CI-2.
410func TestCI2FieldNumbers(t *testing.T) {
411 tests := []struct {
412 msg proto.Message
413 fields map[string]int
414 }{
415 {&ermite.Job{}, map[string]int{"github_token": 15}},
416 {&ermite.UpdateJobRequest{}, map[string]int{"annotations": 3}},
417 {&ermite.CompleteJobRequest{}, map[string]int{"annotations": 6}},
418 {&ermite.StepUpdate{}, map[string]int{"outcome": 7}},
419 {&ermite.Step{}, map[string]int{"outcome": 9}},
420 {&ermite.Annotation{}, map[string]int{
421 "level": 1, "title": 2, "message": 3, "file": 4, "start_line": 5,
422 "end_line": 6, "start_column": 7, "end_column": 8, "step_number": 9,
423 }},
424 {&ermite.ActionRef{}, map[string]int{"uses": 1}},
425 {&ermite.ResolveActionsRequest{}, map[string]int{"job_id": 1, "refs": 2}},
426 {&ermite.ResolveActionsResponse{}, map[string]int{"actions": 1}},
427 {&ermite.ResolvedAction{}, map[string]int{
428 "ref": 1, "name_with_owner": 2, "path": 3, "resolved_sha": 4,
429 "tarball_url": 5, "token": 6, "source": 7, "error": 8,
430 }},
431 {&ermite.UploadSummaryRequest{}, map[string]int{"job_id": 1, "step_number": 2, "markdown": 3}},
432 {&ermite.ListAnnotationsRequest{}, map[string]int{"repo": 1, "run_id": 2, "attempt": 3, "job_id": 4}},
433 {&ermite.ListAnnotationsResponse{}, map[string]int{"annotations": 1, "truncated": 2}},
434 {&ermite.JobAnnotation{}, map[string]int{"job_id": 1, "annotation": 2}},
435 {&ermite.GetStepSummariesRequest{}, map[string]int{"repo": 1, "job_id": 2, "attempt": 3}},
436 {&ermite.GetStepSummariesResponse{}, map[string]int{"summaries": 1, "truncated": 2}},
437 {&ermite.StepSummary{}, map[string]int{"step_number": 1, "markdown": 2}},
438 }
439 for _, tt := range tests {
440 md := tt.msg.ProtoReflect().Descriptor()
441 for name, num := range tt.fields {
442 if fd := md.Fields().ByName(protoreflect.Name(name)); fd == nil || int(fd.Number()) != num {
443 t.Errorf("%s.%s: %v, want number %d", md.Name(), name, fd, num)
444 }
445 }
446 }
447}
448
449// TestCI2RoundTrip checks that the CI-2 messages survive the wire.
450func TestCI2RoundTrip(t *testing.T) {
451 msgs := []proto.Message{
452 &ermite.Job{JobId: "j1", GithubToken: "ermitegh_x"},
453 &ermite.UpdateJobRequest{
454 JobId: "j1",
455 Steps: []*ermite.StepUpdate{{
456 Number: 2, State: ermite.StepState_STEP_STATE_COMPLETED,
457 Conclusion: ermite.CiConclusion_CI_CONCLUSION_SUCCESS,
458 Outcome: ermite.CiConclusion_CI_CONCLUSION_FAILURE,
459 }},
460 Annotations: []*ermite.Annotation{{
461 Level: ermite.AnnotationLevel_ANNOTATION_LEVEL_WARNING, Title: "t", Message: "m",
462 File: "a.go", StartLine: 1, EndLine: 2, StartColumn: 3, EndColumn: 4, StepNumber: 2,
463 }},
464 },
465 &ermite.CompleteJobRequest{
466 JobId: "j1",
467 Conclusion: ermite.CiConclusion_CI_CONCLUSION_SUCCESS,
468 Annotations: []*ermite.Annotation{{Level: ermite.AnnotationLevel_ANNOTATION_LEVEL_FAILURE, Message: "m", StepNumber: 3}},
469 },
470 &ermite.ResolveActionsRequest{JobId: "j1", Refs: []*ermite.ActionRef{{Uses: "actions/checkout@v4"}, {Uses: "o/r/p@main"}}},
471 &ermite.ResolveActionsResponse{Actions: []*ermite.ResolvedAction{
472 {
473 Ref: "o/r/p@main", NameWithOwner: "o/r", Path: "p", ResolvedSha: "0123456789012345678901234567890123456789",
474 TarballUrl: "https://example.test/o/r.tar.gz", Token: "t", Source: ermite.ActionSource_ACTION_SOURCE_ERMITE,
475 },
476 {Ref: "x/y@z", Error: "Unable to resolve action `x/y@z`, unable to find version `z`"},
477 }},
478 &ermite.UploadSummaryRequest{JobId: "j1", StepNumber: 2, Markdown: "# hi"},
479 &ermite.ListAnnotationsResponse{Annotations: []*ermite.JobAnnotation{{JobId: "j1", Annotation: &ermite.Annotation{Message: "m"}}}, Truncated: true},
480 &ermite.GetStepSummariesResponse{Summaries: []*ermite.StepSummary{{StepNumber: 2, Markdown: "# hi"}}, Truncated: true},
481 &ermite.RunJob{Id: "j1", Steps: []*ermite.Step{{Number: 2, Outcome: ermite.CiConclusion_CI_CONCLUSION_FAILURE}}},
482 }
483 for _, in := range msgs {
484 b, err := proto.Marshal(in)
485 if err != nil {
486 t.Fatal(err)
487 }
488 out := in.ProtoReflect().New().Interface()
489 if err := proto.Unmarshal(b, out); err != nil {
490 t.Fatal(err)
491 }
492 if !proto.Equal(in, out) {
493 t.Errorf("%T: round trip mismatch: %v != %v", in, out, in)
494 }
495 }
496}
497
498// TestCI3FieldNumbers pins the numbers of the fields added for CI-3.
499func TestCI3FieldNumbers(t *testing.T) {
500 tests := []struct {
501 msg proto.Message
502 fields map[string]int
503 }{
504 {&ermite.Job{}, map[string]int{
505 "secrets": 16, "vars": 17, "environment": 18, "permissions": 19, "cache": 20, "trusted": 21,
506 }},
507 {&ermite.JobPermissions{}, map[string]int{"levels": 1}},
508 {&ermite.JobCache{}, map[string]int{"repository": 1, "write_ref": 2, "read_refs": 3}},
509 {&ermite.Run{}, map[string]int{"trusted": 20}},
510 {&ermite.RunJob{}, map[string]int{"concurrency_group": 20}},
511 {&ermite.DispatchWorkflowRequest{}, map[string]int{"repo": 1, "workflow_path": 2, "ref": 3, "inputs": 4}},
512 {&ermite.DispatchWorkflowResponse{}, map[string]int{"run": 1}},
513 {&ermite.Workflow{}, map[string]int{"path": 1, "name": 2, "state": 3}},
514 {&ermite.ListWorkflowsRequest{}, map[string]int{"repo": 1}},
515 {&ermite.ListWorkflowsResponse{}, map[string]int{"workflows": 1}},
516 {&ermite.DisableWorkflowRequest{}, map[string]int{"repo": 1, "workflow": 2}},
517 {&ermite.DisableWorkflowResponse{}, map[string]int{"workflow": 1}},
518 {&ermite.EnableWorkflowRequest{}, map[string]int{"repo": 1, "workflow": 2}},
519 {&ermite.EnableWorkflowResponse{}, map[string]int{"workflow": 1}},
520 {&ermite.SettingsScope{}, map[string]int{"instance": 1, "repo": 2, "environment": 3}},
521 {&ermite.EnvironmentRef{}, map[string]int{"repo": 1, "name": 2}},
522 {&ermite.Secret{}, map[string]int{"name": 1, "updated_at": 2}},
523 {&ermite.Variable{}, map[string]int{"name": 1, "value": 2, "updated_at": 3}},
524 {&ermite.Environment{}, map[string]int{"repo": 1, "name": 2, "created_at": 3}},
525 {&ermite.SetSecretRequest{}, map[string]int{"scope": 1, "name": 2, "value": 3}},
526 {&ermite.ListSecretsRequest{}, map[string]int{"scope": 1, "page_size": 2, "page_token": 3}},
527 {&ermite.ListSecretsResponse{}, map[string]int{"secrets": 1, "next_page_token": 2}},
528 {&ermite.ListVariablesResponse{}, map[string]int{"variables": 1, "next_page_token": 2}},
529 {&ermite.ListEnvironmentsRequest{}, map[string]int{"repo": 1, "page_size": 2, "page_token": 3}},
530 {&ermite.ListEnvironmentsResponse{}, map[string]int{"environments": 1, "next_page_token": 2}},
531 }
532 for _, tt := range tests {
533 md := tt.msg.ProtoReflect().Descriptor()
534 for name, num := range tt.fields {
535 if fd := md.Fields().ByName(protoreflect.Name(name)); fd == nil || int(fd.Number()) != num {
536 t.Errorf("%s.%s: %v, want number %d", md.Name(), name, fd, num)
537 }
538 }
539 }
540}
541
542// TestCI3RoundTrip checks that the CI-3 messages survive the wire.
543func TestCI3RoundTrip(t *testing.T) {
544 msgs := []proto.Message{
545 &ermite.Job{
546 JobId: "j1", Secrets: map[string]string{"TOKEN": "s3cret"}, Vars: map[string]string{"REGION": "eu"},
547 Environment: "staging", Trusted: true,
548 Permissions: &ermite.JobPermissions{Levels: map[string]string{"contents": "write", "packages": "none"}},
549 Cache: &ermite.JobCache{
550 Repository: "ermite", WriteRef: "refs/pull/7/merge",
551 ReadRefs: []string{"refs/pull/7/merge", "refs/heads/main"},
552 },
553 },
554 &ermite.Run{Id: 9, Event: "workflow_dispatch", Trusted: true},
555 &ermite.RunJob{Id: "j1", Status: ermite.CiStatus_CI_STATUS_PENDING, ConcurrencyGroup: "deploy"},
556 &ermite.DispatchWorkflowRequest{Repo: "ermite", WorkflowPath: "ci.yml", Ref: "main", Inputs: map[string]string{"debug": "true"}},
557 &ermite.DispatchWorkflowResponse{Run: &ermite.Run{Id: 9, Event: "workflow_dispatch"}},
558 &ermite.ListWorkflowsResponse{Workflows: []*ermite.Workflow{{Path: ".github/workflows/release.yml", Name: "Release", State: ermite.WorkflowState_WORKFLOW_STATE_DISABLED}}},
559 &ermite.DisableWorkflowRequest{Repo: "ermite", Workflow: "release.yml"},
560 &ermite.EnableWorkflowResponse{Workflow: &ermite.Workflow{Path: ".github/workflows/release.yml", State: ermite.WorkflowState_WORKFLOW_STATE_ENABLED}},
561 &ermite.SetSecretRequest{Scope: &ermite.SettingsScope{Where: &ermite.SettingsScope_Instance{Instance: true}}, Name: "A", Value: "v"},
562 &ermite.SetSecretRequest{Scope: &ermite.SettingsScope{Where: &ermite.SettingsScope_Repo{Repo: "ermite"}}, Name: "A", Value: "v"},
563 &ermite.SetVariableRequest{
564 Scope: &ermite.SettingsScope{Where: &ermite.SettingsScope_Environment{Environment: &ermite.EnvironmentRef{Repo: "ermite", Name: "prod"}}},
565 Name: "A", Value: "v",
566 },
567 &ermite.ListSecretsResponse{Secrets: []*ermite.Secret{{Name: "A"}}, NextPageToken: "t"},
568 &ermite.ListEnvironmentsResponse{Environments: []*ermite.Environment{{Repo: "ermite", Name: "prod"}}},
569 }
570 for _, in := range msgs {
571 b, err := proto.Marshal(in)
572 if err != nil {
573 t.Fatal(err)
574 }
575 out := in.ProtoReflect().New().Interface()
576 if err := proto.Unmarshal(b, out); err != nil {
577 t.Fatal(err)
578 }
579 if !proto.Equal(in, out) {
580 t.Errorf("%T: round trip mismatch: %v != %v", in, out, in)
581 }
582 }
583}
584
585// TestMirrorWire pins event numbers and verifies presence of explicit zero patches.
586func TestMirrorWire(t *testing.T) {
587 fields := (&ermite.Event{}).ProtoReflect().Descriptor().Fields()
588 for name, number := range map[protoreflect.Name]protoreflect.FieldNumber{"mirror_failed": 23, "mirror_recovered": 24} {
589 if fd := fields.ByName(name); fd == nil || fd.Number() != number || fd.ContainingOneof().Name() != "kind" {
590 t.Fatalf("Event.%s: %v, want kind field %d", name, fd, number)
591 }
592 }
593 for _, msg := range []proto.Message{&ermite.MirrorFailed{}, &ermite.MirrorRecovered{}} {
594 fields := msg.ProtoReflect().Descriptor().Fields()
595 if fields.ByName("url") != nil || fields.ByName("detail") != nil {
596 t.Fatalf("%T leaks URL or detail", msg)
597 }
598 }
599 for _, in := range []proto.Message{
600 &ermite.SetMirrorConfigRequest{Enabled: proto.Bool(false), UrlTemplate: proto.String(""), Cron: proto.String(""), AllBranches: proto.Bool(false)},
601 &ermite.SetRepoMirrorRequest{Repo: "repo", AllBranches: proto.Bool(false)},
602 &ermite.Event{Kind: &ermite.Event_MirrorFailed{MirrorFailed: &ermite.MirrorFailed{Repo: "repo", State: ermite.MirrorState_MIRROR_STATE_PAUSED, FailureClass: "diverged"}}},
603 &ermite.Event{Kind: &ermite.Event_MirrorRecovered{MirrorRecovered: &ermite.MirrorRecovered{Repo: "repo"}}},
604 } {
605 b, err := proto.Marshal(in)
606 if err != nil {
607 t.Fatal(err)
608 }
609 out := in.ProtoReflect().New().Interface()
610 if err := proto.Unmarshal(b, out); err != nil {
611 t.Fatal(err)
612 }
613 if !proto.Equal(in, out) {
614 t.Errorf("%T: round trip mismatch: %v != %v", in, out, in)
615 }
616 }
617}