Fix segment test

Author: mayorova

test/unit/tasks/segment/segment_test.rb

@@ -3,36 +3,38 @@
 require 'test_helper'
 
 module Tasks
-  class Segment::SegmentTest < ActiveSupport::TestCase
-    disable_transactional_fixtures!
+  module Segment
+    class SegmentTest < ActiveSupport::TestCase
+      disable_transactional_fixtures!
 
-    def given_segment_users_config(user_id_saved_locally)
-      FakeFS do
-        config = Rails.root.join('config', 'segment_users.csv')
-        FakeFS::FileSystem.clone(config.dirname, '/tmp/config')
-        config.open('w') { |f| f.puts(segment_users_csv(user_id_saved_locally)) }
-        yield
+      def given_segment_users_config(user_id_saved_locally)
+        FakeFS do
+          config = Rails.root.join('config', 'segment_users.csv')
+          FakeFS::FileSystem.clone(config.dirname, '/tmp/config')
+          config.open('w') { |f| f.puts(segment_users_csv(user_id_saved_locally)) }
+          yield
+        end
       end
-    end
 
-    test 'save_deleted_users' do
-      user_id = FactoryBot.create(:member).id
-      given_segment_users_config(user_id) do
-        execute_rake_task 'segment/segment.rake', 'segment:save_deleted_users', 'config/segment_users.csv'
+      test 'save_deleted_users' do
+        user_id = FactoryBot.create(:member).id
+        given_segment_users_config(user_id) do
+          execute_rake_task 'segment/segment.rake', 'segment:save_deleted_users', 'config/segment_users.csv'
+        end
+        deleted_object_ids = DeletedObject.users.pluck(:object_id)
+        assert_includes deleted_object_ids, 100
+        assert_includes deleted_object_ids, 200
+        assert_not_includes deleted_object_ids, user_id
       end
-      deleted_object_ids = DeletedObject.users.pluck(:object_id)
-      assert_includes deleted_object_ids, 100
-      assert_includes deleted_object_ids, 200
-      assert_not_includes deleted_object_ids, user_id
-    end
 
-    def segment_users_csv(user_id_saved)
-      <<-CSV
-      First name,Last name,Name,User ID,account_id
-      ExampleName1,ExampleSurname1,Display Name 1,100,-11
-      ExampleName2,ExampleSurname2,Display Name 2,200,
-      ExampleName3,ExampleSurname3,Display Name 3,#{user_id_saved},
-      CSV
+      def segment_users_csv(user_id_saved)
+        <<-CSV
+        First name,Last name,Name,User ID,account_id
+        ExampleName1,ExampleSurname1,Display Name 1,100,-11
+        ExampleName2,ExampleSurname2,Display Name 2,200,
+        ExampleName3,ExampleSurname3,Display Name 3,#{user_id_saved},
+        CSV
+      end
     end
   end
 end