Function TestUploadSingleShot_RetriesTooManyWriteOperations

cmd/put_test.go:209–236

(t *testing.T)

}

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

	calls := 0
	mock := &mockFilesClient{
		uploadFn: func(arg *files.UploadArg, content io.Reader) (*files.FileMetadata, error) {
			calls++
			if _, err := io.ReadAll(content); err != nil {
				return nil, err
			}
			if calls == 1 {
				return nil, uploadWriteThrottleError()
			}
			return &files.FileMetadata{}, nil
		},
	}

	_, err := uploadSingleShot(mock, reader, uploadArg, int64(len(data)), nil)
	if err != nil {
		t.Errorf("expected nil error after write-throttle retry, got %v", err)
	}
	if calls != 2 {
		t.Errorf("expected 2 calls, got %d", calls)
	}
}

func testConfig() dropbox.Config {
	return dropbox.Config{Token: "test-token"}