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