Improved geocoding tests

ankane · Oct 2, 2023 · 830d9b1 · 830d9b1
1 parent 1b8c2cf
commit 830d9b1
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/test/geocode_test.rb b/test/geocode_test.rb
@@ -20,8 +20,9 @@ def test_geocode_true
       assert_enqueued_with(job: Ahoy::GeocodeV2Job, queue: "ahoy") do
         get products_url
       end
-      perform_enqueued_jobs
       visit = Ahoy::Visit.last
+      perform_enqueued_jobs
+      visit.reload
       assert_equal "Country", visit.country
       assert_equal "Region", visit.region
       assert_equal "City", visit.city
@@ -35,8 +36,9 @@ def test_geocode_true_cookies_none
       assert_enqueued_with(job: Ahoy::GeocodeV2Job, queue: "ahoy") do
         get products_url
       end
-      perform_enqueued_jobs
       visit = Ahoy::Visit.last
+      perform_enqueued_jobs
+      visit.reload
       assert_equal "Country", visit.country
       assert_equal "Region", visit.region
       assert_equal "City", visit.city
@@ -50,10 +52,11 @@ def test_geocode_failed
       assert_enqueued_with(job: Ahoy::GeocodeV2Job, queue: "ahoy") do
         get products_url
       end
+      visit = Ahoy::Visit.last
       Geocoder.stub(:search, []) do
         perform_enqueued_jobs
       end
-      visit = Ahoy::Visit.last
+      visit.reload
       assert_nil visit.country
       assert_nil visit.region
       assert_nil visit.city