diff --git a/backend/cache/storage_persistent.go b/backend/cache/storage_persistent.go index 0c36a4649..054857c24 100644 --- a/backend/cache/storage_persistent.go +++ b/backend/cache/storage_persistent.go @@ -980,15 +980,6 @@ func (b *Persistent) updatePendingUpload(remote string, fn func(item *tempUpload }) } -// SetPendingUploadToStarted is a way to mark an entry as started (even if it's not already) -// TO BE USED IN TESTING ONLY -func (b *Persistent) SetPendingUploadToStarted(remote string) error { - return b.updatePendingUpload(remote, func(item *tempUploadInfo) error { - item.Started = true - return nil - }) -} - // ReconcileTempUploads will recursively look for all the files in the temp directory and add them to the queue func (b *Persistent) ReconcileTempUploads(ctx context.Context, cacheFs *Fs) error { return b.db.Update(func(tx *bolt.Tx) error { @@ -1036,19 +1027,6 @@ func (b *Persistent) ReconcileTempUploads(ctx context.Context, cacheFs *Fs) erro }) } -// PurgeTempUploads will remove all the pending uploads from the queue -// TO BE USED IN TESTING ONLY -func (b *Persistent) PurgeTempUploads() { - b.tempQueueMux.Lock() - defer b.tempQueueMux.Unlock() - - _ = b.db.Update(func(tx *bolt.Tx) error { - _ = tx.DeleteBucket([]byte(tempBucket)) - _, _ = tx.CreateBucketIfNotExists([]byte(tempBucket)) - return nil - }) -} - // Close should be called when the program ends gracefully func (b *Persistent) Close() { b.cleanupMux.Lock() diff --git a/backend/cache/utils_test.go b/backend/cache/utils_test.go new file mode 100644 index 000000000..c02d86ce9 --- /dev/null +++ b/backend/cache/utils_test.go @@ -0,0 +1,23 @@ +package cache + +import bolt "go.etcd.io/bbolt" + +// PurgeTempUploads will remove all the pending uploads from the queue +func (b *Persistent) PurgeTempUploads() { + b.tempQueueMux.Lock() + defer b.tempQueueMux.Unlock() + + _ = b.db.Update(func(tx *bolt.Tx) error { + _ = tx.DeleteBucket([]byte(tempBucket)) + _, _ = tx.CreateBucketIfNotExists([]byte(tempBucket)) + return nil + }) +} + +// SetPendingUploadToStarted is a way to mark an entry as started (even if it's not already) +func (b *Persistent) SetPendingUploadToStarted(remote string) error { + return b.updatePendingUpload(remote, func(item *tempUploadInfo) error { + item.Started = true + return nil + }) +}