3 # NEPI, a framework to manage network experiments
4 # Copyright (C) 2013 INRIA
6 # This program is free software: you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation, either version 3 of the License, or
9 # (at your option) any later version.
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 # GNU General Public License for more details.
16 # You should have received a copy of the GNU General Public License
17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
19 # Author: Julien Tribino <julien.tribino@inria.fr>
22 from nepi.execution.resource import ResourceFactory, ResourceManager, ResourceAction, ResourceState
23 from nepi.execution.ec import ExperimentController
25 from nepi.resources.omf.node import OMFNode
26 from nepi.resources.omf.application import OMFApplication
27 from nepi.resources.omf.interface import OMFWifiInterface
28 from nepi.resources.omf.channel import OMFChannel
29 from nepi.resources.omf.omf_api import OMFAPIFactory
31 from nepi.util.timefuncs import *
36 class OMFResourceFactoryTestCase(unittest.TestCase):
37 def test_creation_phase(self):
39 self.assertEquals(OMFNode.get_rtype(), "OMFNode")
40 self.assertEquals(len(OMFNode._attributes), 6)
42 self.assertEquals(OMFWifiInterface.get_rtype(), "OMFWifiInterface")
43 self.assertEquals(len(OMFWifiInterface._attributes), 10)
45 self.assertEquals(OMFChannel.get_rtype(), "OMFChannel")
46 self.assertEquals(len(OMFChannel._attributes), 6)
48 self.assertEquals(OMFApplication.get_rtype(), "OMFApplication")
49 self.assertEquals(len(OMFApplication._attributes), 13)
51 class OMFEachTestCase(unittest.TestCase):
53 self.ec = ExperimentController(exp_id = "99999")
55 self.node1 = self.ec.register_resource("OMFNode")
56 self.ec.set(self.node1, 'hostname', 'omf.plexus.wlab17')
57 self.ec.set(self.node1, 'xmppSlice', "nepi")
58 self.ec.set(self.node1, 'xmppHost', "xmpp-plexus.onelab.eu")
59 self.ec.set(self.node1, 'xmppPort', "5222")
60 self.ec.set(self.node1, 'xmppPassword', "1234")
62 self.iface1 = self.ec.register_resource("OMFWifiInterface")
63 self.ec.set(self.iface1, 'alias', "w0")
64 self.ec.set(self.iface1, 'mode', "adhoc")
65 self.ec.set(self.iface1, 'type', "g")
66 self.ec.set(self.iface1, 'essid', "vlcexp")
67 self.ec.set(self.iface1, 'ip', "10.0.0.17")
69 self.channel = self.ec.register_resource("OMFChannel")
70 self.ec.set(self.channel, 'channel', "6")
71 self.ec.set(self.channel, 'xmppSlice', "nepi")
72 self.ec.set(self.channel, 'xmppHost', "xmpp-plexus.onelab.eu")
73 self.ec.set(self.channel, 'xmppPort', "5222")
74 self.ec.set(self.channel, 'xmppPassword', "1234")
76 self.app1 = self.ec.register_resource("OMFApplication")
77 self.ec.set(self.app1, 'appid', 'Vlc#1')
78 self.ec.set(self.app1, 'path', "/opt/vlc-1.1.13/cvlc")
79 self.ec.set(self.app1, 'args', "/opt/10-by-p0d.avi --sout '#rtp{dst=10.0.0.37,port=1234,mux=ts}'")
80 self.ec.set(self.app1, 'env', "DISPLAY=localhost:10.0 XAUTHORITY=/root/.Xauthority")
82 self.app2 = self.ec.register_resource("OMFApplication")
84 self.app3 = self.ec.register_resource("OMFApplication")
85 self.app4 = self.ec.register_resource("OMFApplication")
86 self.app5 = self.ec.register_resource("OMFApplication")
88 self.ec.register_connection(self.app1, self.node1)
89 self.ec.register_connection(self.app2, self.node1)
90 self.ec.register_connection(self.app3, self.node1)
91 self.ec.register_connection(self.app4, self.node1)
92 self.ec.register_connection(self.app5, self.node1)
93 self.ec.register_connection(self.node1, self.iface1)
94 self.ec.register_connection(self.iface1, self.channel)
96 self.ec.register_condition(self.app2, ResourceAction.START, self.app1, ResourceState.STARTED , "3s")
97 self.ec.register_condition(self.app3, ResourceAction.START, self.app2, ResourceState.STARTED , "2s")
98 self.ec.register_condition(self.app4, ResourceAction.START, self.app3, ResourceState.STARTED , "3s")
99 self.ec.register_condition(self.app5, ResourceAction.START, [self.app3, self.app2], ResourceState.STARTED , "2s")
100 self.ec.register_condition(self.app5, ResourceAction.START, self.app1, ResourceState.STARTED , "25s")
102 self.ec.register_condition([self.app1, self.app2, self.app3,self.app4, self.app5], ResourceAction.STOP, self.app5, ResourceState.STARTED , "1s")
107 def test_creation_and_configuration_node(self):
108 self.assertEquals(self.ec.get(self.node1, 'hostname'), 'omf.plexus.wlab17')
109 self.assertEquals(self.ec.get(self.node1, 'xmppSlice'), 'nepi')
110 self.assertEquals(self.ec.get(self.node1, 'xmppHost'), 'xmpp-plexus.onelab.eu')
111 self.assertEquals(self.ec.get(self.node1, 'xmppPort'), '5222')
112 self.assertEquals(self.ec.get(self.node1, 'xmppPassword'), '1234')
114 def test_creation_and_configuration_interface(self):
115 self.assertEquals(self.ec.get(self.iface1, 'alias'), 'w0')
116 self.assertEquals(self.ec.get(self.iface1, 'mode'), 'adhoc')
117 self.assertEquals(self.ec.get(self.iface1, 'type'), 'g')
118 self.assertEquals(self.ec.get(self.iface1, 'essid'), 'vlcexp')
119 self.assertEquals(self.ec.get(self.iface1, 'ip'), '10.0.0.17')
121 def test_creation_and_configuration_channel(self):
122 self.assertEquals(self.ec.get(self.channel, 'channel'), '6')
123 self.assertEquals(self.ec.get(self.channel, 'xmppSlice'), 'nepi')
124 self.assertEquals(self.ec.get(self.channel, 'xmppHost'), 'xmpp-plexus.onelab.eu')
125 self.assertEquals(self.ec.get(self.channel, 'xmppPort'), '5222')
126 self.assertEquals(self.ec.get(self.channel, 'xmppPassword'), '1234')
128 def test_creation_and_configuration_application(self):
129 self.assertEquals(self.ec.get(self.app1, 'appid'), 'Vlc#1')
130 self.assertEquals(self.ec.get(self.app1, 'path'), '/opt/vlc-1.1.13/cvlc')
131 self.assertEquals(self.ec.get(self.app1, 'args'), "/opt/10-by-p0d.avi --sout '#rtp{dst=10.0.0.37,port=1234,mux=ts}'")
132 self.assertEquals(self.ec.get(self.app1, 'env'), 'DISPLAY=localhost:10.0 XAUTHORITY=/root/.Xauthority')
134 def test_connection(self):
135 self.assertEquals(len(self.ec.get_resource(self.node1).connections), 6)
136 self.assertEquals(len(self.ec.get_resource(self.iface1).connections), 2)
137 self.assertEquals(len(self.ec.get_resource(self.channel).connections), 1)
138 self.assertEquals(len(self.ec.get_resource(self.app1).connections), 1)
139 self.assertEquals(len(self.ec.get_resource(self.app2).connections), 1)
141 def test_condition(self):
142 self.assertEquals(len(self.ec.get_resource(self.app1).conditions[ResourceAction.STOP]), 1)
143 self.assertEquals(len(self.ec.get_resource(self.app2).conditions[ResourceAction.START]), 1)
144 self.assertEquals(len(self.ec.get_resource(self.app3).conditions[ResourceAction.START]), 1)
145 self.assertEquals(len(self.ec.get_resource(self.app4).conditions[ResourceAction.STOP]), 1)
146 self.assertEquals(len(self.ec.get_resource(self.app5).conditions[ResourceAction.START]), 2)
148 class OMFVLCNormalCase(unittest.TestCase):
149 def test_deploy(self):
150 ec = ExperimentController(exp_id = "5421" )
152 self.node1 = ec.register_resource("OMFNode")
153 ec.set(self.node1, 'hostname', 'omf.plexus.wlab17')
154 ec.set(self.node1, 'xmppSlice', "nepi")
155 ec.set(self.node1, 'xmppHost', "xmpp-plexus.onelab.eu")
156 ec.set(self.node1, 'xmppPort', "5222")
157 ec.set(self.node1, 'xmppPassword', "1234")
159 self.iface1 = ec.register_resource("OMFWifiInterface")
160 ec.set(self.iface1, 'alias', "w0")
161 ec.set(self.iface1, 'mode', "adhoc")
162 ec.set(self.iface1, 'type', "g")
163 ec.set(self.iface1, 'essid', "vlcexp")
164 ec.set(self.iface1, 'ip', "10.0.0.17")
166 self.channel = ec.register_resource("OMFChannel")
167 ec.set(self.channel, 'channel', "6")
168 ec.set(self.channel, 'xmppSlice', "nepi")
169 ec.set(self.channel, 'xmppHost', "xmpp-plexus.onelab.eu")
170 ec.set(self.channel, 'xmppPort', "5222")
171 ec.set(self.channel, 'xmppPassword', "1234")
173 self.app1 = ec.register_resource("OMFApplication")
174 ec.set(self.app1, 'appid', 'Vlc#1')
175 ec.set(self.app1, 'path', "/opt/vlc-1.1.13/cvlc")
176 ec.set(self.app1, 'args', "/opt/10-by-p0d.avi --sout '#rtp{dst=10.0.0.37,port=1234,mux=ts}'")
177 ec.set(self.app1, 'env', "DISPLAY=localhost:10.0 XAUTHORITY=/root/.Xauthority")
179 self.app2 = ec.register_resource("OMFApplication")
180 ec.set(self.app2, 'appid', 'Test#1')
181 ec.set(self.app2, 'path', "/usr/bin/test")
182 ec.set(self.app2, 'args', "-1")
183 ec.set(self.app2, 'env', " ")
185 self.app3 = ec.register_resource("OMFApplication")
186 ec.set(self.app3, 'appid', 'Test#2')
187 ec.set(self.app3, 'path', "/usr/bin/test")
188 ec.set(self.app3, 'args', "-2")
189 ec.set(self.app3, 'env', " ")
191 self.app4 = ec.register_resource("OMFApplication")
192 ec.set(self.app4, 'appid', 'Test#3')
193 ec.set(self.app4, 'path', "/usr/bin/test")
194 ec.set(self.app4, 'args', "-3")
195 ec.set(self.app4, 'env', " ")
197 self.app5 = ec.register_resource("OMFApplication")
198 ec.set(self.app5, 'appid', 'Kill#2')
199 ec.set(self.app5, 'path', "/usr/bin/killall")
200 ec.set(self.app5, 'args', "vlc")
201 ec.set(self.app5, 'env', " ")
203 ec.register_connection(self.app1, self.node1)
204 ec.register_connection(self.app2, self.node1)
205 ec.register_connection(self.app3, self.node1)
206 ec.register_connection(self.app4, self.node1)
207 ec.register_connection(self.app5, self.node1)
208 ec.register_connection(self.node1, self.iface1)
209 ec.register_connection(self.iface1, self.channel)
211 ec.register_condition(self.app2, ResourceAction.START, self.app1, ResourceState.STARTED , "3s")
212 ec.register_condition(self.app3, ResourceAction.START, self.app2, ResourceState.STARTED , "2s")
213 ec.register_condition(self.app4, ResourceAction.START, self.app3, ResourceState.STARTED , "3s")
214 ec.register_condition(self.app5, ResourceAction.START, [self.app3, self.app2], ResourceState.STARTED , "2s")
215 ec.register_condition(self.app5, ResourceAction.START, self.app1, ResourceState.STARTED , "25s")
217 ec.register_condition([self.app1, self.app2, self.app3,self.app4, self.app5], ResourceAction.STOP, self.app5, ResourceState.STARTED , "1s")
221 ec.wait_finished([self.app1, self.app2, self.app3,self.app4, self.app5])
223 self.assertEquals(round(tdiffsec(ec.get_resource(self.app2).start_time, ec.get_resource(self.app1).start_time),0), 3.0)
224 self.assertEquals(round(tdiffsec(ec.get_resource(self.app3).start_time, ec.get_resource(self.app2).start_time),0), 2.0)
225 self.assertEquals(round(tdiffsec(ec.get_resource(self.app4).start_time, ec.get_resource(self.app3).start_time),0), 3.0)
226 self.assertEquals(round(tdiffsec(ec.get_resource(self.app5).start_time, ec.get_resource(self.app3).start_time),0), 20.0)
227 self.assertEquals(round(tdiffsec(ec.get_resource(self.app5).start_time, ec.get_resource(self.app1).start_time),0), 25.0)
229 self.assertEquals(ec.get_resource(self.node1).state, ResourceState.STARTED)
230 self.assertEquals(ec.get_resource(self.iface1).state, ResourceState.STARTED)
231 self.assertEquals(ec.get_resource(self.channel).state, ResourceState.STARTED)
232 self.assertEquals(ec.get_resource(self.app1).state, ResourceState.STOPPED)
233 self.assertEquals(ec.get_resource(self.app2).state, ResourceState.STOPPED)
234 self.assertEquals(ec.get_resource(self.app3).state, ResourceState.STOPPED)
235 self.assertEquals(ec.get_resource(self.app4).state, ResourceState.STOPPED)
236 self.assertEquals(ec.get_resource(self.app5).state, ResourceState.STOPPED)
240 self.assertEquals(ec.get_resource(self.node1).state, ResourceState.RELEASED)
241 self.assertEquals(ec.get_resource(self.iface1).state, ResourceState.RELEASED)
242 self.assertEquals(ec.get_resource(self.channel).state, ResourceState.RELEASED)
243 self.assertEquals(ec.get_resource(self.app1).state, ResourceState.RELEASED)
244 self.assertEquals(ec.get_resource(self.app2).state, ResourceState.RELEASED)
245 self.assertEquals(ec.get_resource(self.app3).state, ResourceState.RELEASED)
246 self.assertEquals(ec.get_resource(self.app4).state, ResourceState.RELEASED)
247 self.assertEquals(ec.get_resource(self.app5).state, ResourceState.RELEASED)
250 if __name__ == '__main__':