1 files changed, 123 insertions, 0 deletions
diff --git a/test/conscript.tcl b/test/conscript.tcl
new file mode 100644
index 0000000..1a254e7
--- /dev/null
+++ b/test/conscript.tcl
@@ -0,0 +1,123 @@
+# See the file LICENSE for redistribution information.
+#
+# Copyright (c) 1999-2009 Oracle.  All rights reserved.
+#
+# $Id$
+#
+# Script for DB_CONSUME test (test070.tcl).
+# Usage: conscript dir file runtype nitems outputfile tnum args
+# dir: DBHOME directory
+# file: db file on which to operate
+# runtype: PRODUCE, CONSUME or WAIT -- which am I?
+# nitems: number of items to put or get
+# outputfile: where to log consumer results
+# tnum: test number
+
+proc consumescript_produce { db_cmd nitems tnum args } {
+	source ./include.tcl
+	global mydata
+
+	set pid [pid]
+	puts "\tTest$tnum: Producer $pid starting, producing $nitems items."
+
+	set db [eval $db_cmd]
+	error_check_good db_open:$pid [is_valid_db $db] TRUE
+
+	set oret -1
+	set ret 0
+	for { set ndx 0 } { $ndx < $nitems } { incr ndx } {
+		set oret $ret
+		if { 0xffffffff > 0 && $oret > 0x7fffffff } {
+			incr oret [expr 0 - 0x100000000]
+		}
+		set ret [$db put -append [chop_data q $mydata]]
+		error_check_good db_put \
+		    [expr $oret > $ret ? \
+		    ($oret > 0x7fffffff && $ret < 0x7fffffff) : 1] 1
+
+	}
+
+	set ret [catch {$db close} res]
+	error_check_good db_close:$pid $ret 0
+	puts "\t\tTest$tnum: Producer $pid finished."
+}
+
+proc consumescript_consume { db_cmd nitems tnum outputfile mode args } {
+	source ./include.tcl
+	global mydata
+	set pid [pid]
+	puts "\tTest$tnum: Consumer $pid starting, seeking $nitems items."
+
+	set db [eval $db_cmd]
+	error_check_good db_open:$pid [is_valid_db $db] TRUE
+
+	set oid [open $outputfile a]
+
+	for { set ndx 0 } { $ndx < $nitems } { } {
+		set ret [$db get $mode]
+		if { [llength $ret] > 0 } {
+			error_check_good correct_data:$pid \
+				[lindex [lindex $ret 0] 1] [pad_data q $mydata]
+			set rno [lindex [lindex $ret 0] 0]
+			puts $oid $rno
+			incr ndx
+		} else {
+			# No data to consume;  wait.
+		}
+	}
+
+	error_check_good output_close:$pid [close $oid] ""
+
+	set ret [catch {$db close} res]
+	error_check_good db_close:$pid $ret 0
+	puts "\t\tTest$tnum: Consumer $pid finished."
+}
+
+source ./include.tcl
+source $test_path/test.tcl
+
+# Verify usage
+if { $argc < 6 } {
+	puts stderr "FAIL:[timestamp] Usage: $usage"
+	exit
+}
+
+set usage "conscript.tcl dir file runtype nitems outputfile tnum"
+
+# Initialize arguments
+set dir [lindex $argv 0]
+set file [lindex $argv 1]
+set runtype [lindex $argv 2]
+set nitems [lindex $argv 3]
+set outputfile [lindex $argv 4]
+set tnum [lindex $argv 5]
+# args is the string "{ -len 20 -pad 0}", so we need to extract the
+# " -len 20 -pad 0" part.
+set args [lindex [lrange $argv 6 end] 0]
+
+set mydata "consumer data"
+
+# Open env
+set dbenv [berkdb_env -home $dir ]
+error_check_good db_env_create [is_valid_env $dbenv] TRUE
+
+# Figure out db opening command.
+set db_cmd [concat {berkdb_open -create -mode 0644 -queue -env}\
+	$dbenv $args $file]
+
+# Invoke consumescript_produce or consumescript_consume based on $runtype
+if { $runtype == "PRODUCE" } {
+	# Producers have nothing to log;  make sure outputfile is null.
+	error_check_good no_producer_outputfile $outputfile ""
+	consumescript_produce $db_cmd $nitems $tnum $args
+} elseif { $runtype == "CONSUME" } {
+	consumescript_consume $db_cmd $nitems $tnum $outputfile -consume $args
+} elseif { $runtype == "WAIT" } {
+	consumescript_consume $db_cmd $nitems $tnum $outputfile -consume_wait \
+		$args
+} else {
+	error_check_good bad_args $runtype \
+	    "either PRODUCE, CONSUME, or WAIT"
+}
+error_check_good env_close [$dbenv close] 0
+exit