Use default model [skip ci]

Andrew Kane
1 parent ed6d1ab5
Showing 1 changed file with 8 additions and 4 deletions Show diff stats
test/multi_tenancy_test.rb
@@ -7,16 +7,20 @@ class MultiTenancyTest &lt; Minitest::Test
  
   def test_basic
     Apartment::Tenant.switch!("tenant1")
-    store_names ["Product A"], Tenant
+    store_names ["Product A"]
     Apartment::Tenant.switch!("tenant2")
-    store_names ["Product B"], Tenant
+    store_names ["Product B"]
     Apartment::Tenant.switch!("tenant1")
-    assert_search "product", ["Product A"], {load: false}, Tenant
+    assert_search "product", ["Product A"], {load: false}
     Apartment::Tenant.switch!("tenant2")
-    assert_search "product", ["Product B"], {load: false}, Tenant
+    assert_search "product", ["Product B"], {load: false}
   end
  
   def teardown
     Apartment::Tenant.reset if defined?(Apartment)
   end
+
+  def default_model
+    Tenant
+  end
 end
...	...	@@ -7,16 +7,20 @@ class MultiTenancyTest < Minitest::Test
7	7
8	8	def test_basic
9	9	Apartment::Tenant.switch!("tenant1")
10		- store_names ["Product A"], Tenant
	10	+ store_names ["Product A"]
11	11	Apartment::Tenant.switch!("tenant2")
12		- store_names ["Product B"], Tenant
	12	+ store_names ["Product B"]
13	13	Apartment::Tenant.switch!("tenant1")
14		- assert_search "product", ["Product A"], {load: false}, Tenant
	14	+ assert_search "product", ["Product A"], {load: false}
15	15	Apartment::Tenant.switch!("tenant2")
16		- assert_search "product", ["Product B"], {load: false}, Tenant
	16	+ assert_search "product", ["Product B"], {load: false}
17	17	end
18	18
19	19	def teardown
20	20	Apartment::Tenant.reset if defined?(Apartment)
21	21	end
	22	+
	23	+ def default_model
	24	+ Tenant
	25	+ end
22	26	end
...	...