From c0047fc4738d1cf5d0feaf8595a8135505851f54 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mica=C3=ABl=20Bergeron?= <mbergeron@gitlab.com>
Date: Fri, 16 Mar 2018 14:34:55 -0400
Subject: [PATCH] personal snippets will now comply with `background_upload`

---
 app/uploaders/file_mover.rb       | 6 +++++-
 app/uploaders/object_storage.rb   | 1 +
 spec/uploaders/file_mover_spec.rb | 6 ++++++
 3 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/app/uploaders/file_mover.rb b/app/uploaders/file_mover.rb
index 93913d2fcce..bd7736ad74e 100644
--- a/app/uploaders/file_mover.rb
+++ b/app/uploaders/file_mover.rb
@@ -10,7 +10,11 @@ class FileMover
 
   def execute
     move
-    uploader.record_upload if update_markdown
+
+    if update_markdown
+      uploader.record_upload
+      uploader.schedule_background_upload
+    end
   end
 
   private
diff --git a/app/uploaders/object_storage.rb b/app/uploaders/object_storage.rb
index 132d78607d6..7218cb0a0fc 100644
--- a/app/uploaders/object_storage.rb
+++ b/app/uploaders/object_storage.rb
@@ -52,6 +52,7 @@ module ObjectStorage
 
       def schedule_background_upload(*args)
         return unless schedule_background_upload?
+        return unless upload
 
         ObjectStorage::BackgroundMoveWorker.perform_async(self.class.name,
                                                 upload.class.to_s,
diff --git a/spec/uploaders/file_mover_spec.rb b/spec/uploaders/file_mover_spec.rb
index bc024cd307c..68b7e24776d 100644
--- a/spec/uploaders/file_mover_spec.rb
+++ b/spec/uploaders/file_mover_spec.rb
@@ -36,6 +36,12 @@ describe FileMover do
       it 'creates a new update record' do
         expect { subject }.to change { Upload.count }.by(1)
       end
+
+      it 'schedules a background migration' do
+        expect_any_instance_of(PersonalFileUploader).to receive(:schedule_background_upload).once
+
+        subject
+      end
     end
 
     context 'when update_markdown fails' do