Move custom assertion method.

Defining the assert_correct_state within a shoulda context caused
meant that in some cases the method was not found. Moving it to a
wider context (the test class) fixes this.
Merge pull request #207 from alphagov/move_assertion_method

Move custom assertion method.