Improve rake task output

lib/tasks/domain.rake

@@ -1,8 +1,13 @@
 namespace :domain do
   desc 'Discard domains'
   task discard: :environment do
+    domain_count = 0
+
     Domain.discard_domains do |domain|
       puts "#{domain} is discarded"
-    end
+      domain_count = domain_count + 1
+    end
+
+    puts "Discarded total: #{domain_count}"
   end
 end

test/integration/tasks/discard_domain_test.rb

@@ -40,4 +40,12 @@ class DiscardDomainTaskTest < TaskTestCase
     @domain.reload
     refute @domain.discarded?
   end
+
+  def test_show_results
+    @domain.update!(delete_at: Time.zone.parse('2010-07-05 07:59'))
+    $stdout = StringIO.new
+
+    Rake::Task['domain:discard'].execute
+    assert_equal "shop.test is discarded\nDiscarded total: 1\n", $stdout.string
+  end
 end