--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/utils/python-unit-tests.py Tue Jul 08 10:43:58 2008 -0700
@@ -0,0 +1,86 @@
+import unittest
+import ns3
+
+class TestSimulator(unittest.TestCase):
+
+ def testScheduleNow(self):
+ def callback(args):
+ self._args_received = args
+ self._cb_time = ns3.Simulator.Now()
+ ns3.Simulator.Destroy()
+ self._args_received = None
+ self._cb_time = None
+ ns3.Simulator.ScheduleNow(callback, "args")
+ ns3.Simulator.Run()
+ self.assertEqual(self._args_received, "args")
+ self.assertEqual(self._cb_time.GetSeconds(), 0.0)
+
+ def testSchedule(self):
+ def callback(args):
+ self._args_received = args
+ self._cb_time = ns3.Simulator.Now()
+ ns3.Simulator.Destroy()
+ self._args_received = None
+ self._cb_time = None
+ ns3.Simulator.Schedule(ns3.Seconds(123), callback, "args")
+ ns3.Simulator.Run()
+ self.assertEqual(self._args_received, "args")
+ self.assertEqual(self._cb_time.GetSeconds(), 123.0)
+
+ def testScheduleDestroy(self):
+ def callback(args):
+ self._args_received = args
+ self._cb_time = ns3.Simulator.Now()
+ ns3.Simulator.Destroy()
+ self._args_received = None
+ self._cb_time = None
+ def null(): pass
+ ns3.Simulator.Schedule(ns3.Seconds(123), null)
+ ns3.Simulator.ScheduleDestroy(callback, "args")
+ ns3.Simulator.Run()
+ ns3.Simulator.Destroy()
+ self.assertEqual(self._args_received, "args")
+ self.assertEqual(self._cb_time.GetSeconds(), 123.0)
+
+ if 0: # these tests are known to fail for now (pybindgen limitation)
+ def testTime_EQ(self):
+ self.assert_(ns3.Seconds(123) == ns3.Seconds(123))
+ def testTime_GE(self):
+ self.assert_(ns3.Seconds(123) >= ns3.Seconds(123))
+ def testTime_LE(self):
+ self.assert_(ns3.Seconds(123) <= ns3.Seconds(123))
+ def testTime_GT(self):
+ self.assert_(ns3.Seconds(124) > ns3.Seconds(123))
+ def testTime_LT(self):
+ self.assert_(ns3.Seconds(123) < ns3.Seconds(124))
+
+ def testConfig(self):
+ ns3.Config.Set("ns3::OnOffApplication::PacketSize", ns3.UintegerValue(123))
+ ns3.Config.SetDefault("ns3::OnOffApplication::PacketSize", ns3.UintegerValue(123))
+ # hm.. no Config.Get?
+
+ if 0:
+ # not yet: https://bugs.launchpad.net/pybindgen/+bug/246069
+ def testSocket(self):
+ node = ns3.Node()
+ ns3.AddInternetStack(node)
+ self._received_packet = None
+
+ def rx_callback(socket):
+ assert self._received_packet is None
+ self._received_packet = socket.Recv()
+
+ sink = ns3.Socket.CreateSocket(node, ns3.TypeId.LookupByName("ns3::UdpSocketFactory"))
+ sink.Bind(ns3.InetSocketAddress(ns3.Ipv4Address.GetAny(), 80))
+ sink.SetRecvCallback(rx_callback)
+
+ source = ns3.Socket.CreateSocket(node, ns3.TypeId.LookupByName("ns3::UdpSocketFactory"))
+ source.SendTo(ns3.InetSocketAddress(ns3.Ipv4Address.GetAny(), 80), ns3.Packet(19))
+
+ ns3.Simulator.Run()
+ self.assert_(self._received_packet is not None)
+ self.assertEqual(self._received_packet.GetSize(), 19)
+
+
+if __name__ == '__main__':
+ unittest.main()