Function TestRunUpdateForOrgStopsOnCriticalRateLimit

pkg/cli/update_org_test.go:371–419

(t *testing.T)

}

func TestRunUpdateForOrgStopsOnCriticalRateLimit(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", "octo/c"}, nil
	}
	var previewed []string
	previewOrgRepoUpdatesFn = func(ctx context.Context, repo string, opts UpdateWorkflowsOptions, verbose bool) (orgRepoPreview, error) {
		previewed = append(previewed, repo)
		return orgRepoPreview{
			Repo:           repo,
			TotalWorkflows: 1,
			Workflows: []orgWorkflowPreview{{
				Name:       "repo-assist",
				CurrentRef: "v1.0.0",
				LatestRef:  "v1.1.0",
			}},
		}, nil
	}
	runUpdateForTargetRepoFn = func(ctx context.Context, targetRepo string, opts UpdateWorkflowsOptions, createPR bool, verbose bool) error {
		return nil
	}
	calls := 0
	waitForOrgRateLimitFn = func(ctx context.Context, resource string, verbose bool) error {
		calls++
		if calls >= 2 {
			return errOrgRateLimitCritical
		}
		return nil
	}
	defer func() {
		searchOrgWorkflowReposFn = origSearch
		previewOrgRepoUpdatesFn = origPreview
		runUpdateForTargetRepoFn = origUpdate
		waitForOrgRateLimitFn = origWait
	}()

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

	// Critical budget on the second repo stops the scan but still shows the report.
	assert.Equal(t, []string{"octo/a"}, previewed)
	assert.Contains(t, output, "budget critical")
	assert.Contains(t, output, "- octo/a")
}

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