1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
use Test;
BEGIN { plan tests => 18 }
use DBI;
sub now {
return time();
}
sub add2 {
my ( $a, $b ) = @_;
return $a + $b;
}
sub my_sum {
my $sum = 0;
foreach my $x (@_) {
$sum += $x;
}
return $sum;
}
sub error {
die "function is dying: ", @_, "\n";
}
sub void_return {
}
sub return2 {
return ( 1, 2 );
}
sub return_null {
return undef;
}
sub my_defined {
return defined $_[0];
}
sub noop {
return $_[0];
}
my $dbh = DBI->connect("dbi:SQLite2:dbname=foo", "", "", { PrintError => 0 } );
ok($dbh);
$dbh->func( "now", 0, \&now, "create_function" );
my $result = $dbh->selectrow_arrayref( "SELECT now()" );
ok( $result->[0] );
$dbh->do( 'CREATE TEMP TABLE func_test ( a, b )' );
$dbh->do( 'INSERT INTO func_test VALUES ( 1, 3 )' );
$dbh->do( 'INSERT INTO func_test VALUES ( 0, 4 )' );
$dbh->func( "add2", 2, \&add2, "create_function" );
$result = $dbh->selectrow_arrayref( "SELECT add2(1,3)" );
ok( $result->[0] == 4 );
$result = $dbh->selectall_arrayref( "SELECT add2(a,b) FROM func_test" );
ok( $result->[0][0] = 4 && $result->[1][0] == 4 );
$dbh->func( "my_sum", -1, \&my_sum, "create_function" );
$result = $dbh->selectrow_arrayref( "SELECT my_sum( '2', 3, 4, '5')" );
ok( $result->[0] == 14 );
$dbh->func( "error", -1, \&error, "create_function" );
$result = $dbh->selectrow_arrayref( "SELECT error( 'I died' )" );
ok( !$result );
ok( $DBI::errstr =~ /function is dying: I died/ );
$dbh->func( "void_return", -1, \&void_return, "create_function" );
$result = $dbh->selectrow_arrayref( "SELECT void_return( 'I died' )" );
ok( $result && !defined $result->[0] );
$dbh->func( "return_null", -1, \&return_null, "create_function" );
$result = $dbh->selectrow_arrayref( "SELECT return_null()" );
ok( $result && !defined $result->[0] );
$dbh->func( "return2", -1, \&return2, "create_function" );
$result = $dbh->selectrow_arrayref( "SELECT return2()" );
ok( $result && $result->[0] == 2 );
$dbh->func( "my_defined", 1, \&my_defined, "create_function" );
$result = $dbh->selectrow_arrayref( "SELECT my_defined(1)" );
ok( $result && $result->[0] );
$result = $dbh->selectrow_arrayref( "SELECT my_defined('')" );
ok( $result && $result->[0] );
$result = $dbh->selectrow_arrayref( "SELECT my_defined('abc')" );
ok( $result && $result->[0] );
$result = $dbh->selectrow_arrayref( "SELECT my_defined(NULL)" );
ok( $result && !$result->[0] );
$dbh->func( "noop", 1, \&noop, "create_function" );
$result = $dbh->selectrow_arrayref( "SELECT noop(NULL)" );
ok( $result && !defined $result->[0] );
$result = $dbh->selectrow_arrayref( "SELECT noop(1)" );
ok( $result && $result->[0] == 1);
$result = $dbh->selectrow_arrayref( "SELECT noop('')" );
ok( $result && $result->[0] eq '' );
$result = $dbh->selectrow_arrayref( "SELECT noop(1.1)" );
ok( $result && $result->[0] == 1.1 );
$dbh->disconnect;
|
