Function TestGetDownloadWithRetry

cmd/get_test.go:171–207

(t *testing.T)

}

func TestGetDownloadWithRetry(t *testing.T) {
	stubRetrySleep(t)
	tmpDir := t.TempDir()
	dst := filepath.Join(tmpDir, "downloaded.txt")
	content := "file content here"

	calls := 0
	mock := &mockFilesClient{
		downloadFn: func(arg *files.DownloadArg) (*files.FileMetadata, io.ReadCloser, error) {
			calls++
			if calls < 3 {
				return nil, nil, dbxauth.ServerError{APIError: dropbox.APIError{ErrorSummary: "500"}}
			}
			meta := &files.FileMetadata{
				Metadata: files.Metadata{PathDisplay: "/test.txt"},
				Size:     uint64(len(content)),
			}
			return meta, io.NopCloser(strings.NewReader(content)), nil
		},
	}

	err := downloadFile(mock, "/test.txt", dst)
	if err != nil {
		t.Errorf("expected nil error, got %v", err)
	}
	if calls != 3 {
		t.Errorf("expected 3 calls, got %d", calls)
	}

	got, err := os.ReadFile(dst)
	if err != nil {
		t.Fatalf("failed to read downloaded file: %v", err)
	}
	if string(got) != content {
		t.Errorf("got %q, want %q", string(got), content)
	}
}

func TestGetDownloadRetriesBodyReadError(t *testing.T) {
	delays := stubRetrySleep(t)