Function TestUploadChunked_RetriesSessionStart

cmd/put_test.go:1479–1519

(t *testing.T)

}

func TestUploadChunked_RetriesSessionStart(t *testing.T) {
	stubRetrySleep(t)
	startCalls := 0
	var appendCalls int
	var finishCalled bool
	mock := &mockFilesClient{
		uploadSessionStartFn: func(arg *files.UploadSessionStartArg, content io.Reader) (*files.UploadSessionStartResult, error) {
			startCalls++
			if startCalls < 3 {
				return nil, dbxauth.ServerError{APIError: dropbox.APIError{ErrorSummary: "500"}}
			}
			return &files.UploadSessionStartResult{SessionId: "test-session"}, nil
		},
		uploadSessionAppendV2Fn: func(arg *files.UploadSessionAppendArg, content io.Reader) error {
			appendCalls++
			return nil
		},
		uploadSessionFinishFn: func(arg *files.UploadSessionFinishArg, content io.Reader) (*files.FileMetadata, error) {
			finishCalled = true
			return &files.FileMetadata{}, nil
		},
	}

	data := bytes.Repeat([]byte("x"), 1024)
	reader := bytes.NewReader(data)
	commitInfo := files.NewCommitInfo("/test.txt")

	_, err := uploadChunked(mock, reader, commitInfo, int64(len(data)), 1, 512, false)
	if err != nil {
		t.Errorf("expected nil error, got %v", err)
	}
	if startCalls != 3 {
		t.Errorf("expected 3 session start calls, got %d", startCalls)
	}
	if appendCalls != 2 {
		t.Errorf("expected 2 append calls, got %d", appendCalls)
	}
	if !finishCalled {
		t.Error("expected finish to be called")
	}
}

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