t/storage/debug.t

   1 use strict;
   2 use warnings;
   3 no warnings 'once';
   4
   5 use Test::More;
   6 use Test::Exception;
   7 use Try::Tiny;
   8 use lib qw(t/lib);
   9 use DBICTest;
  10 use Path::Class qw/file/;
  11
  12 BEGIN { delete @ENV{qw(DBIC_TRACE DBIC_TRACE_PROFILE DBICTEST_SQLITE_USE_FILE)} }
  13
  14 my $schema = DBICTest->init_schema();
  15
  16 my $lfn = file("t/var/sql-$$.log");
  17 unlink $lfn or die $!
  18   if -e $lfn;
  19
  20 # make sure we are testing the vanilla debugger and not ::PrettyPrint
  21 require DBIx::Class::Storage::Statistics;
  22 $schema->storage->debugobj(DBIx::Class::Storage::Statistics->new);
  23
  24 ok ( $schema->storage->debug(1), 'debug' );
  25 $schema->storage->debugfh($lfn->openw);
  26 $schema->storage->debugfh->autoflush(1);
  27 $schema->resultset('CD')->count;
  28
  29 my @loglines = $lfn->slurp;
  30 is (@loglines, 1, 'one line of log');
  31 like($loglines[0], qr/^SELECT COUNT/, 'File log via debugfh success');
  32
  33 $schema->storage->debugfh(undef);
  34
  35 {
  36   local $ENV{DBIC_TRACE} = "=$lfn";
  37   unlink $lfn;
  38
  39   $schema->resultset('CD')->count;
  40
  41   my $schema2 = DBICTest->init_schema(no_deploy => 1);
  42   $schema2->storage->_do_query('SELECT 1'); # _do_query() logs via standard mechanisms
  43
  44   my @loglines = $lfn->slurp;
  45   is(@loglines, 2, '2 lines of log');
  46   like($loglines[0], qr/^SELECT COUNT/, 'Env log from schema1 success');
  47   like($loglines[1], qr/^SELECT 1:/, 'Env log from schema2 success');
  48
  49   $schema->storage->debugobj->debugfh(undef)
  50 }
  51
  52 END {
  53   unlink $lfn;
  54 }
  55
  56 open(STDERRCOPY, '>&STDERR');
  57
  58 # STDERR will be closed, no T::B diag in blocks
  59 my $exception = try {
  60   close(STDERR);
  61   $schema->resultset('CD')->search({})->count;
  62 } catch {
  63   $_
  64 } finally {
  65   # restore STDERR
  66   open(STDERR, '>&STDERRCOPY');
  67 };
  68
  69 like $exception, qr/\QDuplication of STDERR for debug output failed (perhaps your STDERR is closed?)/;
  70
  71
  72
  73 # test debugcb and debugobj protocol
  74 {
  75   my $rs = $schema->resultset('CD')->search( {
  76     artist => 1,
  77     cdid => { -between => [ 1, 3 ] },
  78     title => { '!=' => \[ '?', undef ] }
  79   });
  80
  81   my $sql_trace = 'SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track FROM cd me WHERE ( ( artist = ? AND ( cdid BETWEEN ? AND ? ) AND title != ? ) )';
  82   my @bind_trace = qw( '1' '1' '3' NULL );  # quotes are in fact part of the trace </facepalm>
  83
  84
  85   my @args;
  86   $schema->storage->debugcb(sub { push @args, @_ } );
  87
  88   $rs->all;
  89
  90   is_deeply( \@args, [
  91     "SELECT",
  92     sprintf( "%s: %s\n", $sql_trace, join ', ', @bind_trace ),
  93   ]);
  94
  95   {
  96     package DBICTest::DebugObj;
  97     our @ISA = 'DBIx::Class::Storage::Statistics';
  98
  99     sub query_start {
 100       my $self = shift;
 101       ( $self->{_traced_sql}, @{$self->{_traced_bind}} ) = @_;
 102     }
 103   }
 104
 105   my $do = $schema->storage->debugobj(DBICTest::DebugObj->new);
 106
 107   $rs->all;
 108
 109   is( $do->{_traced_sql}, $sql_trace );
 110
 111   is_deeply ( $do->{_traced_bind}, \@bind_trace );
 112 }
 113
 114 done_testing;