1 files changed, 56 insertions, 0 deletions
diff --git a/t/14_progress_handler.t b/t/14_progress_handler.t
new file mode 100644
index 0000000..21abf5a
--- /dev/null
+++ b/t/14_progress_handler.t
@@ -0,0 +1,56 @@
+#!/usr/bin/perl
+
+use strict;
+BEGIN {
+	$|  = 1;
+	$^W = 1;
+}
+
+use t::lib::Test qw/connect_ok @CALL_FUNCS/;
+use Test::More;
+use Test::NoWarnings;
+
+plan tests => 5 * @CALL_FUNCS + 1;
+
+my $N_OPCODES = 50; # how many opcodes before calling the progress handler
+
+# our progress_handler just remembers how many times it was called
+my $n_callback = 0;
+sub progress_handler {
+  $n_callback += 1;
+  return 0;
+}
+
+foreach my $call_func (@CALL_FUNCS) {
+	$n_callback = 0;  # reinitialize
+
+	# connect and register the progress handler
+	my $dbh = connect_ok( RaiseError => 1 );
+	ok($dbh->$call_func( $N_OPCODES, \&progress_handler, "progress_handler" ));
+
+	# populate a temporary table with random numbers
+	$dbh->do( 'CREATE TEMP TABLE progress_test ( foo )' );
+	$dbh->begin_work;
+	for my $count (1 .. 1000) {
+	  my $rand = rand;
+	  $dbh->do( "INSERT INTO progress_test(foo) VALUES ( $rand )" );
+	}
+	$dbh->commit;
+
+	# let the DB do some work (sorting the random numbers)
+	my $result = $dbh->do( "SELECT * from progress_test ORDER BY foo  " );
+
+	# now the progress handler should have been called a number of times
+	ok($n_callback);
+
+
+	# unregister the progress handler, set counter back to zero, do more work
+	ok($dbh->$call_func( $N_OPCODES, undef, "progress_handler" ));
+	$n_callback = 0;
+	$result = $dbh->do( "SELECT * from progress_test ORDER BY foo DESC " );
+
+	# now the progress handler should have been called zero times
+	ok(!$n_callback);
+
+	$dbh->disconnect;
+}