diff --git a/test/integration/disabled_contracts_module_test.rb b/test/integration/disabled_contracts_module_test.rb index 8210cea..54cddf0 100644 --- a/test/integration/disabled_contracts_module_test.rb +++ b/test/integration/disabled_contracts_module_test.rb @@ -15,22 +15,14 @@ class DisabledContractsModuleTest < ActionController::IntegrationTest end should "block access to list" do - visit '/' - assert_response :success - - click_link 'Projects' - assert_response :success - - click_link @project.name - assert_response :success - + visit_project(@project) + assert_select "#main-menu" do assert_select 'a', :text => /contracts/i, :count => 0 end visit "/projects/#{@project.identifier}/contracts" - assert_response :forbidden - assert_template 'common/403' + assert_forbidden end end diff --git a/test/test_helper.rb b/test/test_helper.rb index 5647d61..5d68d8e 100644 --- a/test/test_helper.rb +++ b/test/test_helper.rb @@ -20,6 +20,22 @@ module IntegrationTestHelper assert User.current.logged? end + def visit_project(project) + visit '/' + assert_response :success + + click_link 'Projects' + assert_response :success + + click_link project.name + assert_response :success + end + + def assert_forbidden + assert_response :forbidden + assert_template 'common/403' + end + end class ActionController::IntegrationTest