Function TestUploadSingleShot_RetryOnServerErrorResetsReader

cmd/put_test.go:169–207

(t *testing.T)

}

func TestUploadSingleShot_RetryOnServerErrorResetsReader(t *testing.T) {
	stubRetrySleep(t)
	data := []byte("small file contents")
	reader := bytes.NewReader(data)
	uploadArg := &files.UploadArg{CommitInfo: *files.NewCommitInfo("/test.txt")}

	calls := 0
	var bodies []string
	mock := &mockFilesClient{
		uploadFn: func(arg *files.UploadArg, content io.Reader) (*files.FileMetadata, error) {
			calls++
			body, err := io.ReadAll(content)
			if err != nil {
				return nil, err
			}
			bodies = append(bodies, string(body))
			if calls < 3 {
				return nil, dbxauth.ServerError{APIError: dropbox.APIError{ErrorSummary: "500"}}
			}
			return &files.FileMetadata{}, nil
		},
	}

	_, err := uploadSingleShot(mock, reader, uploadArg, int64(len(data)), nil)
	if err != nil {
		t.Errorf("expected nil error after retry, got %v", err)
	}
	if calls != 3 {
		t.Errorf("expected 3 calls, got %d", calls)
	}
	if len(bodies) != 3 {
		t.Fatalf("expected 3 uploaded bodies, got %d", len(bodies))
	}
	for i, body := range bodies {
		if body != string(data) {
			t.Errorf("body %d = %q, want %q", i, body, string(data))
		}
	}
}

func TestUploadSingleShot_RetriesTooManyWriteOperations(t *testing.T) {
	stubRetrySleep(t)