From e767b9fc4beb8fa3cb33e40a2dfe779e9ccef1d4 Mon Sep 17 00:00:00 2001
From: Struan Donald <struan@exo.org.uk>
Date: Wed, 20 Jul 2011 11:29:58 +0100
Subject: add remove user flag button to report and update edit pages

---
 t/app/controller/admin.t | 45 +++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 41 insertions(+), 4 deletions(-)

(limited to 't/app/controller/admin.t')

diff --git a/t/app/controller/admin.t b/t/app/controller/admin.t
index 2ccb2ea73..8d55bbc18 100644
--- a/t/app/controller/admin.t
+++ b/t/app/controller/admin.t
@@ -550,13 +550,32 @@ subtest 'flagging user from report page' => sub {
     $mech->click_ok('flaguser');
 
     $mech->content_contains('User flagged');
-    $mech->content_contains('<small>(Flagged)</small>');
+    $mech->content_contains('Remove flag');
 
     $report->discard_changes;
     ok $report->user->flagged, 'user flagged';
 
     $mech->get_ok( '/admin/report_edit/' . $report->id );
-    $mech->content_contains('<small>(Flagged)</small>');
+    $mech->content_contains('Remove flag');
+};
+
+subtest 'unflagging user from report page' => sub {
+    $report->user->flagged(1);
+    $report->user->update;
+
+    $mech->get_ok( '/admin/report_edit/' . $report->id );
+    $mech->content_contains('Remove flag');
+
+    $mech->click_ok('removeuserflag');
+
+    $mech->content_contains('User flag removed');
+    $mech->content_contains('Flag user');
+
+    $report->discard_changes;
+    ok !$report->user->flagged, 'user not flagged';
+
+    $mech->get_ok( '/admin/report_edit/' . $report->id );
+    $mech->content_contains('Flag user');
 };
 
 $log_entries->delete;
@@ -770,15 +789,33 @@ subtest 'flagging user from update page' => sub {
     $mech->click_ok('flaguser');
 
     $mech->content_contains('User flagged');
-    $mech->content_contains('<small>(Flagged)</small>');
+    $mech->content_contains('Remove flag');
 
     $update->discard_changes;
     ok $update->user->flagged, 'user flagged';
 
     $mech->get_ok( '/admin/update_edit/' . $update->id );
-    $mech->content_contains('<small>(Flagged)</small>');
+    $mech->content_contains('Remove flag');
 };
 
+subtest 'unflagging user from update page' => sub {
+    $update->user->flagged(1);
+    $update->user->update;
+
+    $mech->get_ok( '/admin/update_edit/' . $update->id );
+    $mech->content_contains('Remove flag');
+
+    $mech->click_ok('removeuserflag');
+
+    $mech->content_contains('User flag removed');
+    $mech->content_contains('Flag user');
+
+    $update->discard_changes;
+    ok !$update->user->flagged, 'user not flagged';
+
+    $mech->get_ok( '/admin/update_edit/' . $update->id );
+    $mech->content_contains('Flag user');
+};
 
 subtest 'hiding comment marked as fixed reopens report' => sub {
     $update->mark_fixed( 1 );
-- 
cgit v1.2.3