Update fabricator for MediaAttachment to attach a file according to type (#3862)

This fixes a random spec failures since commit d55f207274.
7 years ago · 356df7ae6b
parent 8f03fdce7f
commit 356df7ae6b
1 changed files with 13 additions and 1 deletions
--- a/spec/fabricators/media_attachment_fabricator.rb
+++ b/spec/fabricators/media_attachment_fabricator.rb
@ -1,4 +1,16 @@
 Fabricator(:media_attachment) do
  account
-  file { [attachment_fixture(['attachment.gif', 'attachment.jpg', 'attachment.webm'].sample), nil].sample }
+  file do |attrs|
+    [
+      case attrs[:type]
+      when :gifv
+        attachment_fixture ['attachment.gif', 'attachment.webm'].sample
+      when :image
+        attachment_fixture 'attachment.jpg'
+      when nil
+        attachment_fixture ['attachment.gif', 'attachment.jpg', 'attachment.webm'].sample
+      end,
+      nil
+    ].sample
+  end
 end