Use stub method [skip ci]

ankane · Dec 28, 2021 · e77dea5 · e77dea5
1 parent b1b6857
commit e77dea5
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/test/explain_test.rb b/test/explain_test.rb
@@ -32,12 +32,8 @@ def test_explain_multiple_statements
   end
 
   def with_explain_timeout(value)
-    previous_value = PgHero.explain_timeout_sec
-    begin
-      PgHero.explain_timeout_sec = value
+    PgHero.stub(:explain_timeout_sec, value) do
       yield
-    ensure
-      PgHero.explain_timeout_sec = previous_value
     end
   end
 end