fix moving attachments specs, fix gov_document moving attachments

Author: tsubik

app/models/concerns/moveable_attachment.rb

@@ -12,8 +12,10 @@ module MoveableAttachment
         return if uploader.nil?
         return if uploader.file.nil?
 
-        from = File.dirname(uploader.file.file.gsub("/public/", "/private/"))
-        to = File.dirname(from.gsub("/private/", "/public/"))
+        public_uploads = uploader.public_root.join("uploads").to_s
+        private_uploads = uploader.private_root.join("uploads").to_s
+        from = File.dirname(uploader.file.file.gsub(public_uploads, private_uploads))
+        to = File.dirname(from.gsub(private_uploads, public_uploads))
         move_attachment(from, to)
       end
 
@@ -22,8 +24,10 @@ module MoveableAttachment
         return if uploader.nil?
         return if uploader.file.nil?
 
-        from = File.dirname(uploader.file.file.gsub("/private/", "/public/"))
-        to = File.dirname(from.gsub("/public/", "/private/"))
+        public_uploads = uploader.public_root.join("uploads").to_s
+        private_uploads = uploader.private_root.join("uploads").to_s
+        from = File.dirname(uploader.file.file.gsub(private_uploads, public_uploads))
+        to = File.dirname(from.gsub(public_uploads, private_uploads))
         move_attachment(from, to)
       end
     end

app/models/gov_document.rb

@@ -80,9 +80,8 @@ def has_data?
   end
 
   def reset_to_not_provided!
-    remove_attachment!
     update!(
-      status: OperatorDocument.statuses[:doc_not_provided],
+      status: OperatorDocument.statuses[:doc_not_provided], attachment: nil,
       expire_date: nil, start_date: nil, uploaded_by: nil, user_id: nil,
       value: nil, link: nil
     )
@@ -102,8 +101,8 @@ def move_previous_attachment_to_private_directory
     previous_attachment_filename = previous_changes[:attachment][0]
     return if previous_attachment_filename.blank?
 
-    from = File.join(attachment.root, attachment.store_dir, previous_attachment_filename)
-    to = from.gsub("/public/", "/private/")
+    from = File.join(attachment.public_root, attachment.store_dir, previous_attachment_filename)
+    to = File.join(attachment.private_root, attachment.store_dir, previous_attachment_filename)
     FileUtils.makedirs(File.dirname(to))
     system "mv #{Shellwords.escape(from)} #{Shellwords.escape(to)}"
   end
@@ -113,9 +112,9 @@ def move_attachment_to_public_directory
     attachment_attr = self[:attachment]
     return if attachment_attr.nil?
 
-    to = File.join(attachment.root, attachment.store_dir, attachment_attr)
-    from = to.gsub("/public/", "/private/")
-    FileUtils.makedirs(File.dirname(from))
+    from = File.join(attachment.private_root, attachment.store_dir, attachment_attr)
+    to = File.join(attachment.public_root, attachment.store_dir, attachment_attr)
+    FileUtils.makedirs(File.dirname(to))
     system "mv #{Shellwords.escape(from)} #{Shellwords.escape(to)}"
   end
 

app/uploaders/application_uploader.rb

@@ -4,10 +4,9 @@ class ApplicationUploader < CarrierWave::Uploader::Base
   storage :file
 
   def root
-    result = Rails.root
-    result = Rails.root.join("private") if private_upload?
-    result = result.to_s.gsub(Rails.root.to_s, Rails.root.join("tmp").to_s) if Rails.env.test?
-    result
+    return private_root if private_upload?
+
+    public_root
   end
 
   def cache_dir
@@ -34,6 +33,18 @@ def private_upload?
     false
   end
 
+  def public_root
+    return Rails.root.join("tmp") if Rails.env.test?
+
+    Rails.root
+  end
+
+  def private_root
+    return Rails.root.join("tmp", "private") if Rails.env.test?
+
+    Rails.root.join("private")
+  end
+
   def original_filename
     if file.present?
       file.filename

spec/models/gov_document_spec.rb

@@ -29,7 +29,7 @@
         it "moves previous attachment to private folder" do
           expect(document.attachment.file.file).to match(Rails.root.join("tmp/uploads").to_s)
           expect(document.attachment.file.exists?).to be(true)
-          document.remove_attachment!
+          document.attachment = nil
           document.save!
           document.reload
           expect(document.read_attribute(:attachment)).to be_nil
@@ -117,6 +117,7 @@
 
         it "moves attachment back to public directory" do
           expect(document.attachment.file.file).to match("/private/uploads")
+          expect(document.attachment.file.exists?).to be(true)
           document.restore
           reloaded_doc = GovDocument.find(document.id) # as reload does not reload paper_trail.live? weird
           expect(reloaded_doc.attachment.file.file).to match(Rails.root.join("tmp/uploads").to_s)