Fixing wrong license
[nepi.git] / test / resources / linux / ccn / ccnping.py
1 #!/usr/bin/env python
2 #
3 #    NEPI, a framework to manage network experiments
4 #    Copyright (C) 2013 INRIA
5 #
6 #    This program is free software: you can redistribute it and/or modify
7 #    it under the terms of the GNU General Public License version 2 as
8 #    published by the Free Software Foundation;
9 #
10 #    This program is distributed in the hope that it will be useful,
11 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
12 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 #    GNU General Public License for more details.
14 #
15 #    You should have received a copy of the GNU General Public License
16 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
17 #
18 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
19
20 from nepi.execution.ec import ExperimentController 
21 from test_utils import skipIfAnyNotAlive
22
23 import os
24 import time
25 import tempfile
26 import unittest
27
28 class LinuxCCNPingTestCase(unittest.TestCase):
29     def setUp(self):
30         self.fedora_host = "nepi2.pl.sophia.inria.fr"
31         self.fedora_user = "inria_nepi"
32
33         self.ubuntu_host = "roseval.pl.sophia.inria.fr"
34         self.ubuntu_user = "alina"
35         
36         self.target = "nepi5.pl.sophia.inria.fr"
37
38     @skipIfAnyNotAlive
39     def t_count(self, user1, host1, user2, host2):
40
41         ec = ExperimentController(exp_id = "test-ccn-ping-count")
42         
43         node1 = ec.register_resource("linux::Node")
44         ec.set(node1, "hostname", host1)
45         ec.set(node1, "username", user1)
46         ec.set(node1, "cleanExperiment", True)
47         ec.set(node1, "cleanProcesses", True)
48
49         ccnd1 = ec.register_resource("linux::CCND")
50         ec.register_connection(ccnd1, node1)
51
52         entry1 = ec.register_resource("linux::FIBEntry")
53         ec.set(entry1, "host", host2)
54         ec.register_connection(entry1, ccnd1)
55  
56         server = ec.register_resource("linux::CCNPingServer")
57         ec.set(server, "prefix", "ccnx:/test")
58         ec.register_connection(server, ccnd1)
59  
60         node2 = ec.register_resource("linux::Node")
61         ec.set(node2, "hostname", host2)
62         ec.set(node2, "username", user2)
63         ec.set(node2, "cleanExperiment", True)
64         ec.set(node2, "cleanProcesses", True)
65
66         ccnd2 = ec.register_resource("linux::CCND")
67         ec.register_connection(ccnd2, node2)
68
69         entry2 = ec.register_resource("linux::FIBEntry")
70         ec.set(entry2, "host", host1)
71         ec.register_connection(entry2, ccnd2)
72  
73         client = ec.register_resource("linux::CCNPing")
74         ec.set(client, "c", 15)
75         ec.set(client, "prefix", "ccnx:/test")
76         ec.register_connection(client, ccnd2)
77         ec.register_connection(client, server)
78
79         ec.deploy()
80
81         ec.wait_finished(client)
82
83         stdout = ec.trace(client, "stdout")
84         expected = "15 Interests transmitted"
85         self.assertTrue(stdout.find(expected) > -1)
86
87         ec.shutdown()
88
89     def test_count_fedora(self):
90         self.t_count(self.fedora_user, self.fedora_host, self.fedora_user, 
91                 self.target)
92
93     def test_count_ubuntu(self):
94         self.t_count(self.ubuntu_user, self.ubuntu_host, self.ubuntu_user,
95                 self.target)
96
97 if __name__ == '__main__':
98     unittest.main()
99