Allow user-specified uri to test for leaks

[catagits/Catalyst-Runtime.git] / t / optional / memleak.t

diff --git a/t/optional/memleak.t b/t/optional/memleak.t
index 648cc2c..0a9fbe1 100644 (file)
--- a/t/optional/memleak.t
+++ b/t/optional/memleak.t
@@ -8,32 +8,61 @@ use lib "$FindBin::Bin/../live/lib";
 
 use Test::More;
 use Catalyst::Test 'TestApp';
-eval "use GTop2";
+use YAML;
+eval "use GTop";
 
 plan skip_all => 'set TEST_MEMLEAK to enable this test'
     unless $ENV{TEST_MEMLEAK};
 plan skip_all => 'GTop required for this test' if $@;
 
-plan tests => 1;
+our $gtop = GTop->new;
+our ( $initial, $final ) = ( 0, 0 ); 
+our $tests = YAML::LoadFile("$FindBin::Bin/stress.yml");
 
-{
-    # make a request to set initial memory size
-    request('http://localhost');
+my $total_tests = 0;
+
+# let the user specify a single uri to test
+my $user_test = shift;
+if ( $user_test ) {
+    plan tests => 1;
+    run_test( $user_test );
+}
+# otherwise, run all tests
+else {
+    map { $total_tests += scalar @{ $tests->{$_} } } keys %{$tests};
+    plan tests => $total_tests;
+    
+    foreach my $test_group ( keys %{$tests} ) {
+        foreach my $test ( @{ $tests->{$test_group} } ) {
+            run_test( $test );
+        }
+    }
+}
+
+sub run_test {
+    my $uri = shift || die 'No URI given for test';
     
-    my $gtop = GTop->new;
-    my $initial = $gtop->proc_mem($$)->size;
-    print "Initial Size: $initial\n";
+    print "TESTING $uri\n";
     
-    for ( 1 .. 1000 ) {
-        request('http://localhost');
+    # make a few requests to set initial memory size
+    for ( 1 .. 3 ) {
+        request( $uri );
     }
     
-    my $final = $gtop->proc_mem($$)->size;
-    print "Final Size:   $final\n";
+    $initial = $gtop->proc_mem($$)->size;
+    print "Initial Size: " . GTop::size_string($initial) . "\n";
+    
+    for ( 1 .. 500 ) {
+        request( $uri );
+    }
+    
+    $final = $gtop->proc_mem($$)->size;
+    print "Final Size:   " . GTop::size_string($final) . "\n";
     
     if ( $final > $initial ) {
-        print "Leaked Bytes: " . ( $final - $initial ) . "\n";
+        print "Leaked Bytes: " . GTop::size_string($final - $initial) . "\n";
     }
     
-    is( $final, $initial, 'memory is not leaking' );
+    is( $final, $initial, "'$uri' memory is not leaking" );
 }
+