Factor out contains_or_lacks test function.

1 files changed, 2 insertions, 13 deletions

diff --git a/t/cobrand/zurich.t b/t/cobrand/zurich.t
index 53d8dba0f..330cb3389 100644
--- a/t/cobrand/zurich.t
+++ b/t/cobrand/zurich.t
@@ -936,8 +936,8 @@ subtest 'Status update shown as appropriate' => sub {
             my ($state, $update, $public, $user_response) = @$_;
             $report->update({ state => $state });
             $mech->get_ok( '/admin/report_edit/' . $report->id );
-            contains_or_lacks($mech, $update, "<textarea name='status_update'");
-            contains_or_lacks($mech, $public || $user_response, '<div class="admin-official-answer">');
+            $mech->contains_or_lacks($update, "<textarea name='status_update'");
+            $mech->contains_or_lacks($public || $user_response, '<div class="admin-official-answer">');
 
             if ($public) {
                 $mech->get_ok( '/report/' . $report->id );
@@ -947,17 +947,6 @@ subtest 'Status update shown as appropriate' => sub {
     };
 };
 
-# TODO refactor into FixMyStreet::TestMech;
-sub contains_or_lacks {
-    my ($mech, $contains, $text) = @_;
-    if ($contains) {
-        $mech->content_contains($text);
-    }
-    else {
-        $mech->content_lacks($text);
-    }
-}
-
 subtest 'time_spent' => sub {
     FixMyStreet::override_config {
         ALLOWED_COBRANDS => [ 'zurich' ],