Function TestRunUpdateForOrgStopsOnCancellation

pkg/cli/update_org_test.go:421–461

(t *testing.T)

}

func TestRunUpdateForOrgStopsOnCancellation(t *testing.T) {
	origSearch := searchOrgWorkflowReposFn
	origPreview := previewOrgRepoUpdatesFn
	origUpdate := runUpdateForTargetRepoFn
	origWait := waitForOrgRateLimitFn
	searchOrgWorkflowReposFn = func(ctx context.Context, org string, workflowNames []string, verbose bool) ([]string, error) {
		return []string{"octo/a", "octo/b"}, nil
	}
	ctx, cancel := context.WithCancel(context.Background())
	previewOrgRepoUpdatesFn = func(c context.Context, repo string, opts UpdateWorkflowsOptions, verbose bool) (orgRepoPreview, error) {
		// Cancel after the first repo is previewed.
		cancel()
		return orgRepoPreview{
			Repo:           repo,
			TotalWorkflows: 1,
			Workflows: []orgWorkflowPreview{{
				Name:       "repo-assist",
				CurrentRef: "v1.0.0",
				LatestRef:  "v1.1.0",
			}},
		}, nil
	}
	runUpdateForTargetRepoFn = func(c context.Context, targetRepo string, opts UpdateWorkflowsOptions, createPR bool, verbose bool) error {
		return nil
	}
	waitForOrgRateLimitFn = func(c context.Context, resource string, verbose bool) error { return nil }
	defer func() {
		searchOrgWorkflowReposFn = origSearch
		previewOrgRepoUpdatesFn = origPreview
		runUpdateForTargetRepoFn = origUpdate
		waitForOrgRateLimitFn = origWait
	}()

	output := captureUpdateOrgStderr(t, func() {
		err := runUpdateForOrg(ctx, "octo", nil, UpdateWorkflowsOptions{}, false, false, false)
		require.NoError(t, err)
	})

	assert.Contains(t, output, "Cancellation requested")
	assert.Contains(t, output, "- octo/a")
}

func TestRunUpdateForOrgCreateIssueReturnsErrorWhenAllIssueCreatesFail(t *testing.T) {
	origSearch := searchOrgWorkflowReposFn