Improve tests

3 files changed, 32 insertions, 0 deletions

diff --git a/tests/api_wait_end_3n/node.py b/tests/api_wait_end_3n/node.py
new file mode 100644
index 0000000..a8fbcd7
--- /dev/null
+++ b/tests/api_wait_end_3n/node.py
@@ -0,0 +1,10 @@
+#!/usr/bin/env python
+
+def execute(api):
+    wait=api.node_id
+    api.log("Before wait for "+str(wait)+"s")
+    api.wait(wait) # Since 3 nodes max(wait)==2
+
+    # Ensure that wait end return back when simulation ends
+    api.wait_end()
+    api.log("Terminated")
diff --git a/tests/api_wait_end_3n/out b/tests/api_wait_end_3n/out
new file mode 100644
index 0000000..321b0c2
--- /dev/null
+++ b/tests/api_wait_end_3n/out
@@ -0,0 +1,7 @@
+[t=0.000,src=n0] Before wait for 0s
+[t=0.000,src=n1] Before wait for 1s
+[t=0.000,src=n2] Before wait for 2s
+[t=2.000,src=n0] Terminated
+[t=2.000,src=n1] Terminated
+[t=2.000,src=n2] Terminated
+[t=2.000,src=esds] Simulation ends
diff --git a/tests/api_wait_end_3n/simulator.py b/tests/api_wait_end_3n/simulator.py
new file mode 100755
index 0000000..4d1763e
--- /dev/null
+++ b/tests/api_wait_end_3n/simulator.py
@@ -0,0 +1,15 @@
+#!/usr/bin/env python
+
+# Load ESDS
+import esds
+import numpy as np
+
+B=np.full((3,3),2)
+L=np.full((3,3),0)
+s=esds.Simulator({"wlan0":{"bandwidth":B, "latency":L, "is_wired":False}, "eth0":{"bandwidth":B, "latency":L, "is_wired":True}})
+
+s.create_node("node")
+s.create_node("node")
+s.create_node("node")
+
+s.run()