323 lines
11 KiB
Python
323 lines
11 KiB
Python
#! /usr/bin/env python3
|
|
|
|
# Copyright (C) 2008-2011 INESC Porto
|
|
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
# Author: Gustavo J. A. M. Carneiro <gjc@inescporto.pt>
|
|
|
|
import unittest
|
|
from ns import ns
|
|
import sys
|
|
|
|
UINT32_MAX = 0xFFFFFFFF
|
|
|
|
|
|
## TestSimulator class
|
|
class TestSimulator(unittest.TestCase):
|
|
## @var _received_packet
|
|
# received packet
|
|
## @var _args_received
|
|
# args
|
|
## @var _cb_time
|
|
# current time
|
|
## @var _context_received
|
|
# context
|
|
|
|
def testScheduleNow(self):
|
|
"""! Test schedule now
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
def callback(args: ns.cppyy.gbl.std.vector) -> None:
|
|
"""! Callback function
|
|
@param args arguments
|
|
return none
|
|
"""
|
|
import copy
|
|
self._args_received = list(map(lambda x: x.decode("utf-8"), args))
|
|
self._cb_time = ns.Simulator.Now()
|
|
ns.Simulator.Destroy()
|
|
self._args_received = None
|
|
self._cb_time = None
|
|
ns.cppyy.cppdef("""
|
|
EventImpl* pythonMakeEvent(void (*f)(std::vector<std::string>), std::vector<std::string> l)
|
|
{
|
|
return MakeEvent(f, l);
|
|
}
|
|
""")
|
|
event = ns.cppyy.gbl.pythonMakeEvent(callback, sys.argv)
|
|
ns.Simulator.ScheduleNow(event)
|
|
ns.Simulator.Run()
|
|
self.assertListEqual(self._args_received, sys.argv)
|
|
self.assertEqual(self._cb_time.GetSeconds(), 0.0)
|
|
|
|
def testSchedule(self):
|
|
"""! Test schedule
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
def callback(args: ns.cppyy.gbl.std.vector):
|
|
"""! Callback function
|
|
@param args arguments
|
|
@return none
|
|
"""
|
|
self._args_received = list(map(lambda x: x.decode("utf-8"), args))
|
|
self._cb_time = ns.Simulator.Now()
|
|
ns.Simulator.Destroy()
|
|
self._args_received = None
|
|
self._cb_time = None
|
|
ns.cppyy.cppdef("""
|
|
EventImpl* pythonMakeEvent2(void (*f)(std::vector<std::string>), std::vector<std::string> l)
|
|
{
|
|
return MakeEvent(f, l);
|
|
}
|
|
""")
|
|
event = ns.cppyy.gbl.pythonMakeEvent2(callback, sys.argv)
|
|
ns.Simulator.Schedule(ns.Seconds(123), event)
|
|
ns.Simulator.Run()
|
|
self.assertListEqual(self._args_received, sys.argv)
|
|
self.assertEqual(self._cb_time.GetSeconds(), 123.0)
|
|
|
|
def testScheduleDestroy(self):
|
|
"""! Test schedule destroy
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
def callback(args: ns.cppyy.gbl.std.vector):
|
|
"""! Callback function
|
|
@param args
|
|
@return none
|
|
"""
|
|
self._args_received = list(map(lambda x: x.decode("utf-8"), args))
|
|
self._cb_time = ns.Simulator.Now()
|
|
ns.Simulator.Destroy()
|
|
self._args_received = None
|
|
self._cb_time = None
|
|
ns.cppyy.cppdef("void null(){ return; }")
|
|
ns.Simulator.Schedule(ns.Seconds(123), ns.cppyy.gbl.null)
|
|
ns.cppyy.cppdef("""
|
|
EventImpl* pythonMakeEvent3(void (*f)(std::vector<std::string>), std::vector<std::string> l)
|
|
{
|
|
return MakeEvent(f, l);
|
|
}
|
|
""")
|
|
event = ns.cppyy.gbl.pythonMakeEvent3(callback, sys.argv)
|
|
ns.Simulator.ScheduleDestroy(event)
|
|
ns.Simulator.Run()
|
|
ns.Simulator.Destroy()
|
|
self.assertListEqual(self._args_received, sys.argv)
|
|
self.assertEqual(self._cb_time.GetSeconds(), 123.0)
|
|
|
|
def testScheduleWithContext(self):
|
|
"""! Test schedule with context
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
def callback(context, args: ns.cppyy.gbl.std.vector):
|
|
"""! Callback
|
|
@param context the cntet
|
|
@param args the arguments
|
|
@return none
|
|
"""
|
|
self._context_received = context
|
|
self._args_received = list(map(lambda x: x.decode("utf-8"), args))
|
|
self._cb_time = ns.Simulator.Now()
|
|
ns.Simulator.Destroy()
|
|
self._args_received = None
|
|
self._cb_time = None
|
|
self._context_received = None
|
|
ns.cppyy.cppdef("""
|
|
EventImpl* pythonMakeEvent4(void (*f)(uint32_t, std::vector<std::string>), uint32_t context, std::vector<std::string> l)
|
|
{
|
|
return MakeEvent(f, context, l);
|
|
}
|
|
""")
|
|
event = ns.cppyy.gbl.pythonMakeEvent4(callback, 54321, sys.argv)
|
|
ns.Simulator.ScheduleWithContext(54321, ns.Seconds(123), event)
|
|
ns.Simulator.Run()
|
|
self.assertEqual(self._context_received, 54321)
|
|
self.assertListEqual(self._args_received, sys.argv)
|
|
self.assertEqual(self._cb_time.GetSeconds(), 123.0)
|
|
|
|
def testTimeComparison(self):
|
|
"""! Test time comparison
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
self.assertTrue(ns.Seconds(123) == ns.Seconds(123))
|
|
self.assertTrue(ns.Seconds(123) >= ns.Seconds(123))
|
|
self.assertTrue(ns.Seconds(123) <= ns.Seconds(123))
|
|
self.assertTrue(ns.Seconds(124) > ns.Seconds(123))
|
|
self.assertTrue(ns.Seconds(123) < ns.Seconds(124))
|
|
|
|
def testTimeNumericOperations(self):
|
|
"""! Test numeric operations
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
self.assertEqual(ns.Seconds(10) + ns.Seconds(5), ns.Seconds(15))
|
|
self.assertEqual(ns.Seconds(10) - ns.Seconds(5), ns.Seconds(5))
|
|
|
|
v1 = ns.int64x64_t(5.0)*ns.int64x64_t(10)
|
|
self.assertEqual(v1, ns.int64x64_t(50))
|
|
|
|
def testConfig(self):
|
|
"""! Test configuration
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
ns.Config.SetDefault("ns3::OnOffApplication::PacketSize", ns.core.UintegerValue(123))
|
|
# hm.. no Config.Get?
|
|
|
|
def testSocket(self):
|
|
"""! Test socket
|
|
@param self
|
|
@return none
|
|
"""
|
|
nc = ns.NodeContainer(1)
|
|
node = nc.Get(0)
|
|
internet = ns.CreateObject("InternetStackHelper")
|
|
internet.Install(node)
|
|
self._received_packet = None
|
|
|
|
def python_rx_callback(socket) -> None:
|
|
self._received_packet = socket.Recv(maxSize=UINT32_MAX, flags=0)
|
|
|
|
ns.cppyy.cppdef("""
|
|
Callback<void,ns3::Ptr<ns3::Socket> > make_rx_callback(void(*func)(Ptr<Socket>))
|
|
{
|
|
return MakeCallback(func);
|
|
}
|
|
""")
|
|
|
|
sink = ns.network.Socket.CreateSocket(node, ns.core.TypeId.LookupByName("ns3::UdpSocketFactory"))
|
|
sink.Bind(ns.addressFromInetSocketAddress(ns.network.InetSocketAddress(ns.network.Ipv4Address.GetAny(), 80)))
|
|
sink.SetRecvCallback(ns.cppyy.gbl.make_rx_callback(python_rx_callback))
|
|
|
|
source = ns.network.Socket.CreateSocket(node, ns.core.TypeId.LookupByName("ns3::UdpSocketFactory"))
|
|
source.SendTo(ns.network.Packet(19), 0, ns.addressFromInetSocketAddress(ns.network.InetSocketAddress(ns.network.Ipv4Address("127.0.0.1"), 80)))
|
|
|
|
ns.Simulator.Run()
|
|
self.assertTrue(self._received_packet is not None)
|
|
self.assertEqual(self._received_packet.GetSize(), 19)
|
|
|
|
# Delete Ptr<>'s on the python side to let C++ clean them
|
|
del internet
|
|
|
|
def testAttributes(self):
|
|
"""! Test attributes function
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
# Templated class DropTailQueue<Packet> in C++
|
|
queue = ns.CreateObject("DropTailQueue<Packet>")
|
|
queueSizeValue = ns.network.QueueSizeValue (ns.network.QueueSize ("500p"))
|
|
queue.SetAttribute("MaxSize", queueSizeValue)
|
|
|
|
limit = ns.network.QueueSizeValue()
|
|
queue.GetAttribute("MaxSize", limit)
|
|
self.assertEqual(limit.Get(), ns.network.QueueSize ("500p"))
|
|
|
|
## -- object pointer values
|
|
mobility = ns.CreateObject("RandomWaypointMobilityModel")
|
|
ptr = ns.CreateObject("PointerValue")
|
|
mobility.GetAttribute("PositionAllocator", ptr)
|
|
self.assertEqual(ptr.GetObject(), ns.core.Ptr["Object"](ns.cppyy.nullptr))
|
|
|
|
pos = ns.mobility.ListPositionAllocator()
|
|
ptr.SetObject(pos)
|
|
mobility.SetAttribute("PositionAllocator", ptr)
|
|
|
|
ptr2 = ns.CreateObject("PointerValue")
|
|
mobility.GetAttribute("PositionAllocator", ptr2)
|
|
self.assertNotEqual(ptr.GetObject(), ns.core.Ptr["Object"](ns.cppyy.nullptr))
|
|
|
|
# Delete Ptr<>'s on the python side to let C++ clean them
|
|
del queue, mobility, ptr, ptr2
|
|
|
|
def testIdentity(self):
|
|
"""! Test identify
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
csma = ns.CreateObject("CsmaNetDevice")
|
|
channel = ns.CreateObject("CsmaChannel")
|
|
csma.Attach(channel)
|
|
|
|
c1 = csma.GetChannel()
|
|
c2 = csma.GetChannel()
|
|
|
|
self.assertEqual(c1, c2)
|
|
|
|
# Delete Ptr<>'s on the python side to let C++ clean them
|
|
del csma, channel
|
|
|
|
def testTypeId(self):
|
|
"""! Test type ID
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
ok, typeId1 = ns.LookupByNameFailSafe("ns3::UdpSocketFactory")
|
|
self.assertTrue(ok)
|
|
self.assertEqual(typeId1.GetName (), "ns3::UdpSocketFactory")
|
|
|
|
ok, typeId1 = ns.LookupByNameFailSafe("ns3::__InvalidTypeName__")
|
|
self.assertFalse(ok)
|
|
|
|
# def testCommandLine(self):
|
|
# """! Test command line
|
|
# @param self this object
|
|
# @return none
|
|
# """
|
|
# cmd = ns.core.CommandLine(__file__)
|
|
# cmd.AddValue("Test1", "this is a test option")
|
|
# cmd.AddValue("Test2", "this is a test option")
|
|
# cmd.AddValue("Test3", "this is a test option", variable="test_xxx")
|
|
# cmd.Test1 = None
|
|
# cmd.Test2 = None
|
|
# cmd.test_xxx = None
|
|
# class Foo:
|
|
# pass
|
|
# foo = Foo()
|
|
# foo.test_foo = None
|
|
# cmd.AddValue("Test4", "this is a test option", variable="test_foo", namespace=foo)
|
|
#
|
|
# cmd.Parse(["python", "--Test1=value1", "--Test2=value2", "--Test3=123", "--Test4=xpto"])
|
|
#
|
|
# self.assertEqual(cmd.Test1, "value1")
|
|
# self.assertEqual(cmd.Test2, "value2")
|
|
# self.assertEqual(cmd.test_xxx, "123")
|
|
# self.assertEqual(foo.test_foo, "xpto")
|
|
|
|
def testSubclass(self):
|
|
"""! Test subclass
|
|
@param self this object
|
|
@return none
|
|
"""
|
|
## MyNode class
|
|
class MyNode(ns.network.Node):
|
|
def GetLocalTime(self) -> ns.Time:
|
|
return ns.Seconds(10)
|
|
|
|
node = MyNode()
|
|
forced_local_time = node.GetLocalTime()
|
|
self.assertEqual(forced_local_time, ns.Seconds(10))
|
|
del node
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=1, failfast=True)
|