From 356df7ae6b6edfc600fbc532f32fbc46314e00a0 Mon Sep 17 00:00:00 2001 From: "Akihiko Odaki (@fn_aki@pawoo.net)" Date: Wed, 21 Jun 2017 03:40:45 +0900 Subject: [PATCH] Update fabricator for MediaAttachment to attach a file according to type (#3862) This fixes a random spec failures since commit d55f207274648369cba30ff001aa3e354fa30dca. --- spec/fabricators/media_attachment_fabricator.rb | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/spec/fabricators/media_attachment_fabricator.rb b/spec/fabricators/media_attachment_fabricator.rb index c5dfe12e5..bb938e36d 100644 --- 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