--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/test/api-poll.tcl Thu Apr 21 14:57:45 2011 +0100
@@ -0,0 +1,118 @@
+#
+# Copyright 2007 Intel Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+test::name api-poll
+net::num_nodes 1
+
+manifest::file apps/dtntest/dtntest dtntest
+
+dtn::config
+dtn::config_topology_common false
+
+test::script {
+ testlog "Running dtnd and dtntest"
+ dtn::run_dtnd 0
+ dtn::run_dtntest 0
+
+ testlog "Waiting for dtnd and dtntest to start up"
+ dtn::wait_for_dtnd 0
+ dtn::wait_for_dtntest 0
+
+ testlog "Creating two handles"
+ set h1 [dtn::tell_dtntest 0 dtn_open]
+ set h2 [dtn::tell_dtntest 0 dtn_open]
+
+ testlog "Creating registrations for source and dest"
+ set regid0 [dtn::tell_dtntest 0 dtn_register $h1 endpoint=dtn://host-0/src expiration=30]
+ set regid1 [dtn::tell_dtntest 0 dtn_register $h2 endpoint=dtn://host-0/dst expiration=30]
+
+ testlog "Setting up a callback handler"
+ set chan [dtn::tell_dtntest 0 dtn_poll_channel $h2]
+ dtn::tell_dtntest 0 { proc arrived {h} { global bundles; \
+ lappend bundles [dtn_recv $h payload_mem=true timeout=0]; } }
+
+ proc send_recv {dest {do_poll 1}} {
+ global chan h1 h2
+
+ if {$do_poll} {
+ dtn::tell_dtntest 0 dtn_begin_poll $h2 10000
+ dtn::tell_dtntest 0 fileevent $chan readable [list arrived $h2]
+ }
+
+ dtn::tell_dtntest 0 "dtn_send $h1 source=dtn://host-0/src $dest\
+ expiration=10 payload_data=test_payload"
+
+ # when polling, we first wait for delivery and then check the
+ # bundles var, otherwise we need to call recv to get the
+ # contents, then it will be delivered
+ if {$do_poll} {
+ dtn::wait_for_bundle_stat 0 1 delivered
+ puts [dtn::tell_dtntest 0 set bundles]
+ dtn::tell_dtntest 0 set bundles {}
+ dtn::tell_dtntest 0 fileevent $chan readable ""
+ } else {
+ puts [dtn::tell_dtntest 0 dtn_recv $h2 payload_mem=true timeout=10000]
+ dtn::wait_for_bundle_stat 0 1 delivered
+ }
+
+ dtn::wait_for_bundle_stat 0 0 pending
+ dtn::tell_dtnd 0 bundle reset_stats
+ }
+
+ testlog "Sending and polling for a couple bundles"
+ for {set i 0} {$i < 5} {incr i} {
+ send_recv dest=dtn://host-0/dst
+ after 1000
+ }
+
+ testlog "Testing poll timeout"
+ dtn::tell_dtntest 0 "dtn_begin_poll $h2 1000"
+ after 2000
+ catch {dtn::tell_dtntest 0 dtn_recv $h2 payload_mem=true timeout=0} err
+ if {$err != "error: error in dtn_recv: operation timed out"} {
+ error "unexpected error from dtn_recv: $err"
+ }
+
+ testlog "Testing a normal delivery"
+ send_recv dest=dtn://host-0/dst 0
+
+ testlog "Testing a poll delivery"
+ send_recv dest=dtn://host-0/dst
+
+ testlog "Testing poll cancel"
+ dtn::tell_dtntest 0 "dtn_begin_poll $h2 10000"
+ dtn::tell_dtntest 0 fileevent $chan readable \
+ "error: unexpected notification on idle channel"
+ after 2000
+ dtn::tell_dtntest 0 fileevent $chan readable ""
+ dtn::tell_dtntest 0 "dtn_cancel_poll $h2"
+
+ testlog "Testing a normal delivery"
+ send_recv dest=dtn://host-0/dst 0
+
+ testlog "Testing a poll delivery"
+ send_recv dest=dtn://host-0/dst
+
+ testlog "Test success!"
+}
+
+test::exit_script {
+ testlog "Stopping dtnd and dtntest"
+ dtn::tell_dtntest 0 dtn_close $h1
+ dtn::tell_dtntest 0 dtn_close $h2
+ dtn::stop_dtntest 0
+ dtn::stop_dtnd 0
+}