refactor: stub_customer test helper

1 files changed, 9 insertions, 0 deletions

diff --git a/billing/test/support/customer_test_helper.rb b/billing/test/support/customer_test_helper.rb
new file mode 100644
index 0000000..ad72edd
--- /dev/null
+++ b/billing/test/support/customer_test_helper.rb
@@ -0,0 +1,9 @@
+module CustomerTestHelper
+
+  def stub_customer(user = nil)
+    user ||= find_record :user
+    customer = stub_record :customer_with_payment_info, user: user
+    Customer.stubs(:find_by_user_id).with(user.id).returns(customer)
+    return customer
+  end
+end