diff --git a/test/models/provider/stripe_test.rb b/test/models/provider/stripe_test.rb index 763025033..e8f1fe076 100644 --- a/test/models/provider/stripe_test.rb +++ b/test/models/provider/stripe_test.rb @@ -67,24 +67,18 @@ class Provider::StripeTest < ActiveSupport::TestCase .with(payment_link_id: "plink_test123") .returns("https://buy.stripe.com/test_payment_link") - original_payment_link_id = ENV["STRIPE_PAYMENT_LINK_ID"] - ENV["STRIPE_PAYMENT_LINK_ID"] = "plink_test123" - - assert_equal "https://buy.stripe.com/test_payment_link", stripe.one_time_contribution_url - ensure - ENV["STRIPE_PAYMENT_LINK_ID"] = original_payment_link_id + with_env_overrides("STRIPE_PAYMENT_LINK_ID" => "plink_test123") do + assert_equal "https://buy.stripe.com/test_payment_link", stripe.one_time_contribution_url + end end test "returns nil for one-time contribution url when payment link id is missing" do stripe = Provider::Stripe.new(secret_key: "foo", webhook_secret: "bar") stripe.expects(:payment_link_url).with(payment_link_id: nil).returns(nil) - original_payment_link_id = ENV["STRIPE_PAYMENT_LINK_ID"] - ENV.delete("STRIPE_PAYMENT_LINK_ID") - - assert_nil stripe.one_time_contribution_url - ensure - ENV["STRIPE_PAYMENT_LINK_ID"] = original_payment_link_id + with_env_overrides("STRIPE_PAYMENT_LINK_ID" => nil) do + assert_nil stripe.one_time_contribution_url + end end test "returns nil when payment link retrieval fails" do