From 3beb0c0bef7f95ee17e2417d4191da66969e94b4 Mon Sep 17 00:00:00 2001 From: Matt Jankowski 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