Improved test code

ankane · Feb 7, 2023 · f88b266 · f88b266
1 parent b10a80d
commit f88b266
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/activerecord_generator_test.rb b/test/activerecord_generator_test.rb
@@ -7,9 +7,12 @@ class ActiverecordGeneratorTest < Rails::Generators::TestCase
   destination File.expand_path("../tmp", __dir__)
   setup :prepare_destination
 
-  def test_works
+  def setup
     skip if ENV["ADAPTER"] == "mongoid"
+    super
+  end
 
+  def test_works
     run_generator
     assert_file "config/initializers/ahoy.rb", /DatabaseStore/
     assert_file "app/models/ahoy/visit.rb", /Ahoy::Visit < ApplicationRecord/
@@ -18,8 +21,6 @@ def test_works
   end
 
   def test_primary_key_type
-    skip if ENV["ADAPTER"] == "mongoid"
-
     Rails.configuration.generators.stub(:options, {active_record: {primary_key_type: :uuid}}) do
       run_generator
     end