t/storage/stats.t

   1 #!/usr/bin/perl
   2
   3 use strict;
   4 use warnings;
   5 use Test::More;
   6
   7 plan tests => 12;
   8
   9 use lib qw(t/lib);
  10
  11 use_ok('DBICTest');
  12 my $schema = DBICTest->init_schema();
  13
  14 my $cbworks = 0;
  15
  16 $schema->storage->debugcb(sub { $cbworks = 1; });
  17 $schema->storage->debug(0);
  18 my $rs = $schema->resultset('CD')->search({});
  19 $rs->count();
  20 ok(!$cbworks, 'Callback not called with debug disabled');
  21
  22 $schema->storage->debug(1);
  23
  24 $rs->count();
  25 ok($cbworks, 'Debug callback worked.');
  26
  27 my $prof = new DBIx::Test::Profiler();
  28 $schema->storage->debugobj($prof);
  29
  30 # Test non-transaction calls.
  31 $rs->count();
  32 ok($prof->{'query_start'}, 'query_start called');
  33 ok($prof->{'query_end'}, 'query_end called');
  34 ok(!$prof->{'txn_begin'}, 'txn_begin not called');
  35 ok(!$prof->{'txn_commit'}, 'txn_commit not called');
  36
  37 $prof->reset();
  38
  39 # Test transaction calls
  40 $schema->txn_begin();
  41 ok($prof->{'txn_begin'}, 'txn_begin called');
  42
  43 $rs = $schema->resultset('CD')->search({});
  44 $rs->count();
  45 ok($prof->{'query_start'}, 'query_start called');
  46 ok($prof->{'query_end'}, 'query_end called');
  47
  48 $schema->txn_commit();
  49 ok($prof->{'txn_commit'}, 'txn_commit called');
  50
  51 $prof->reset();
  52
  53 # Test a rollback
  54 $schema->txn_begin();
  55 $rs = $schema->resultset('CD')->search({});
  56 $rs->count();
  57 $schema->txn_rollback();
  58 ok($prof->{'txn_rollback'}, 'txn_rollback called');
  59
  60 $schema->storage->debug(0);
  61
  62 package DBIx::Test::Profiler;
  63 use strict;
  64
  65 sub new {
  66     my $self = bless({});
  67 }
  68
  69 sub query_start {
  70     my $self = shift();
  71     $self->{'query_start'} = 1;
  72 }
  73
  74 sub query_end {
  75     my $self = shift();
  76     $self->{'query_end'} = 1;
  77 }
  78
  79 sub txn_begin {
  80     my $self = shift();
  81     $self->{'txn_begin'} = 1;
  82 }
  83
  84 sub txn_rollback {
  85     my $self = shift();
  86     $self->{'txn_rollback'} = 1;
  87 }
  88
  89 sub txn_commit {
  90     my $self = shift();
  91     $self->{'txn_commit'} = 1;
  92 }
  93
  94 sub reset {
  95     my $self = shift();
  96
  97     $self->{'query_start'} = 0;
  98     $self->{'query_end'} = 0;
  99     $self->{'txn_begin'} = 0;
 100     $self->{'txn_rollback'} = 0;
 101     $self->{'txn_end'} = 0;
 102 }
 103
 104 1;