2 files changed, 48 insertions, 1 deletions
diff --git a/billing/test/integration/admin_customer_test.rb b/billing/test/integration/admin_customer_test.rb
new file mode 100644
index 0000000..16f2931
--- /dev/null
+++ b/billing/test/integration/admin_customer_test.rb
@@ -0,0 +1,46 @@
+require 'test_helper'
+require 'fake_braintree'
+require 'capybara/rails'
+
+class CustomerCreationTest < ActionDispatch::IntegrationTest
+  include Warden::Test::Helpers
+  include Capybara::DSL
+
+  setup do
+    Warden.test_mode!
+    @admin = User.find_by_login('admin') || FactoryGirl.create(:user, login: 'admin')
+    @user = FactoryGirl.create(:user)
+  end
+
+  teardown do
+    Warden.test_reset!
+    @user.destroy
+    @admin.destroy
+  end
+
+  test "check non customer as admin" do
+    login_as @admin
+    visit '/'
+    click_link 'Users'
+    click_link @user.login
+    click_link 'Billing Settings'
+    assert page.has_content? @user.email_address
+    assert page.has_content? 'No Saved Customer'
+  end
+
+  test "check customer as admin" do
+    login_as @user
+    visit '/'
+    click_link 'Billing Settings'
+    click_button 'Save Payment Info'
+
+    login_as @admin
+    visit '/'
+    click_link 'Users'
+    click_link @user.login
+    click_link 'Billing Settings'
+    assert page.has_content? @user.email_address
+    assert !page.has_content?('No Saved Customer')
+    save_and_open_page
+  end
+end
diff --git a/billing/test/integration/customer_creation_test.rb b/billing/test/integration/customer_creation_test.rb
index 3ab2e4f..532a5b5 100644
--- a/billing/test/integration/customer_creation_test.rb
+++ b/billing/test/integration/customer_creation_test.rb
@@ -22,7 +22,8 @@ class CustomerCreationTest < ActionDispatch::IntegrationTest
   # RackTest assumes all requests to be local. So we need
   # BraintreeTestApp for the braintree transparent redirect to work.
   test "create customer with braintree" do
-    visit '/customer/new'
+    visit '/'
+    click_link 'Billing Settings'
     assert_difference("Customer.count") do
       click_button 'Save Payment Info'
     end