about summary refs log tree commit diff
path: root/spec/models
diff options
context:
space:
mode:
authorMatt Jankowski <mjankowski@thoughtbot.com>2017-04-13 09:29:30 -0400
committerEugen <eugen@zeonfederated.com>2017-04-13 15:29:30 +0200
commit4fe5e04ea47b187b5578d0c9da23ecbe6c382544 (patch)
treec74f399d96112bb9651cad978f33a44f920ca037 /spec/models
parenta57d30c6800eb3ebe15b0398d5f542527faa47f9 (diff)
Fix csv export coverage in export spec (#1691)
Diffstat (limited to 'spec/models')
-rw-r--r--spec/models/export_spec.rb24
1 files changed, 22 insertions, 2 deletions
diff --git a/spec/models/export_spec.rb b/spec/models/export_spec.rb
index 5cc62c266..3ee042fb6 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