1 files changed, 95 insertions, 2 deletions

diff --git a/t/app/controller/reports.t b/t/app/controller/reports.t
index 801dbeddb..5fd1a9962 100644
--- a/t/app/controller/reports.t
+++ b/t/app/controller/reports.t
@@ -1,10 +1,21 @@
 use strict;
 use warnings;
 use Test::More;
-use Test::WWW::Mechanize::Catalyst 'FixMyStreet::App';
+use FixMyStreet::TestMech;
 use mySociety::MaPit;
+use FixMyStreet::App;
+use DateTime;
 
-ok( my $mech = Test::WWW::Mechanize::Catalyst->new, 'Created mech object' );
+ok( my $mech = FixMyStreet::TestMech->new, 'Created mech object' );
+
+delete_problems( 2504 );
+delete_problems( 2651 );
+
+my @westminster_problems = create_problems(5, 2504);
+my @edinburgh_problems = create_problems(3, 2651);
+
+is scalar @westminster_problems, 5, 'correct number of westminster problems created';
+is scalar @edinburgh_problems, 3, 'correct number of edinburgh problems created';
 
 # Run the cron script that makes the data for /reports so we don't get an error.
 system( "bin/cron-wrapper update-all-reports" );
@@ -13,8 +24,35 @@ system( "bin/cron-wrapper update-all-reports" );
 $mech->get_ok('/reports');
 $mech->title_like(qr{Summary reports});
 $mech->content_contains('Birmingham');
+
+my $stats = $mech->extract_report_stats;
+
+is $stats->{'City of Edinburgh Council'}->[1], 3, 'correct number of reports for Edinburgh';
+is $stats->{'Westminster City Council'}->[1], 5, 'correct number of reports for Westminster';
+
 $mech->follow_link_ok( { text_regex => qr/Birmingham/ } );
 
+$mech->get_ok('/reports/Westminster');
+$mech->title_like(qr/Westminster City Council/);
+$mech->content_contains('Westminster City Council');
+$mech->content_contains('All reports Test 3 for 2504', 'problem to be marked non public visible');
+
+my $problems = $mech->extract_problem_list;
+is scalar @$problems, 5, 'correct number of problems displayed';
+
+my $private = $westminster_problems[2];
+ok $private->update( { non_public => 1 } ), 'problem marked non public';
+
+$mech->get_ok('/reports/Westminster');
+$problems = $mech->extract_problem_list;
+is scalar @$problems, 4, 'only public problems are displayed';
+
+$mech->content_lacks('All reports Test 3 for 2504', 'non public problem is not visible');
+
+$mech->get_ok('/reports');
+$stats = $mech->extract_report_stats;
+is $stats->{'Westminster City Council'}->[1], 5, 'non public reports included in stats';
+
 SKIP: {
     skip( "Need 'emptyhomes' in ALLOWED_COBRANDS config", 8 )
         unless FixMyStreet::Cobrand->exists('emptyhomes');
@@ -33,5 +71,60 @@ SKIP: {
     $mech->content_unlike(qr{Oslo">Oslo.*Oslo}s);
 }
 
+sub create_problems {
+    my $count = shift;
+    my $area_id = shift;
+
+    my @problems;
+
+    my $dt = DateTime->now;
+
+    my $user =
+      FixMyStreet::App->model('DB::User')
+      ->find_or_create( { email => 'test@example.com', name => 'Test User' } );
+
+    while ( $count ) {
+        my $problem = FixMyStreet::App->model('DB::Problem')->create( {
+            postcode           => 'SW1A 1AA',
+            council            => $area_id,
+            areas              => ',105255,11806,11828,2247,2504,',
+            category           => 'Other',
+            title              => "All reports Test $count for $area_id",
+            detail             => "All reports Test $count for $area_id Detail",
+            used_map           => 't',
+            name               => 'Test User',
+            anonymous          => 'f',
+            state              => 'confirmed',
+            confirmed          => $dt->ymd . ' ' . $dt->hms,
+            lang               => 'en-gb',
+            service            => '',
+            cobrand            => 'default',
+            cobrand_data       => '',
+            send_questionnaire => 't',
+            latitude           => '51.5016605453401',
+            longitude          => '-0.142497580865087',
+            user_id            => $user->id,
+        } );
+
+        push @problems, $problem;
+        $count--;
+    }
+
+
+    return @problems;
+}
+
+sub delete_problems {
+    my $area_id = shift;
+
+    my $reports = FixMyStreet::App->model('DB::Problem')->search( { council => $area_id } );
+    if ( $reports ) {
+        for my $r ( $reports->all ) {
+            $r->comments->delete;
+        }
+        $reports->delete;
+    }
+}
+
 done_testing();