From 4fe5e04ea47b187b5578d0c9da23ecbe6c382544 Mon Sep 17 00:00:00 2001
From: Matt Jankowski <mjankowski@thoughtbot.com>
Date: Thu, 13 Apr 2017 09:29:30 -0400
Subject: [PATCH] Fix csv export coverage in export spec (#1691)

---
 spec/models/export_spec.rb | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/spec/models/export_spec.rb b/spec/models/export_spec.rb
index 5cc62c2662..3ee042fb63 100644
--- a/spec/models/export_spec.rb
+++ b/spec/models/export_spec.rb
@@ -2,12 +2,32 @@ require 'rails_helper'
 
 describe Export do
   describe 'to_csv' do
-    it 'returns a csv of the accounts' do
+    before do
       one = Account.new(username: 'one', domain: 'local.host')
       two = Account.new(username: 'two', domain: 'local.host')
       accounts = [one, two]
 
-      export = Export.new(accounts).to_csv
+      @account = double(blocking: accounts, muting: accounts, following: accounts)
+    end
+
+    it 'returns a csv of the blocked accounts' do
+      export = Export.new(@account).to_blocked_accounts_csv
+      results = export.strip.split
+
+      expect(results.size).to eq 2
+      expect(results.first).to eq 'one@local.host'
+    end
+
+    it 'returns a csv of the muted accounts' do
+      export = Export.new(@account).to_muted_accounts_csv
+      results = export.strip.split
+
+      expect(results.size).to eq 2
+      expect(results.first).to eq 'one@local.host'
+    end
+
+    it 'returns a csv of the following accounts' do
+      export = Export.new(@account).to_following_accounts_csv
       results = export.strip.split
 
       expect(results.size).to eq 2
-- 
GitLab