Pipes
[vpp.git] / test / test_tcp.py
1 #!/usr/bin/env python
2
3 import unittest
4
5 from framework import VppTestCase, VppTestRunner
6 from vpp_ip_route import VppIpTable, VppIpRoute, VppRoutePath
7
8
9 class TestTCP(VppTestCase):
10     """ TCP Test Case """
11
12     @classmethod
13     def setUpClass(cls):
14         super(TestTCP, cls).setUpClass()
15
16     def setUp(self):
17         super(TestTCP, self).setUp()
18         self.vapi.session_enable_disable(is_enabled=1)
19         self.create_loopback_interfaces(2)
20
21         table_id = 0
22
23         for i in self.lo_interfaces:
24             i.admin_up()
25
26             if table_id != 0:
27                 tbl = VppIpTable(self, table_id)
28                 tbl.add_vpp_config()
29
30             i.set_table_ip4(table_id)
31             i.config_ip4()
32             table_id += 1
33
34         # Configure namespaces
35         self.vapi.app_namespace_add(namespace_id="0",
36                                     sw_if_index=self.loop0.sw_if_index)
37         self.vapi.app_namespace_add(namespace_id="1",
38                                     sw_if_index=self.loop1.sw_if_index)
39
40     def tearDown(self):
41         for i in self.lo_interfaces:
42             i.unconfig_ip4()
43             i.set_table_ip4(0)
44             i.admin_down()
45         self.vapi.session_enable_disable(is_enabled=0)
46         super(TestTCP, self).tearDown()
47
48     def test_tcp_unittest(self):
49         """ TCP Unit Tests """
50         error = self.vapi.cli("test tcp all")
51
52         if error:
53             self.logger.critical(error)
54         self.assertEqual(error.find("failed"), -1)
55
56     def test_tcp_transfer(self):
57         """ TCP echo client/server transfer """
58
59         # Add inter-table routes
60         ip_t01 = VppIpRoute(self, self.loop1.local_ip4, 32,
61                             [VppRoutePath("0.0.0.0",
62                                           0xffffffff,
63                                           nh_table_id=1)])
64         ip_t10 = VppIpRoute(self, self.loop0.local_ip4, 32,
65                             [VppRoutePath("0.0.0.0",
66                                           0xffffffff,
67                                           nh_table_id=0)], table_id=1)
68         ip_t01.add_vpp_config()
69         ip_t10.add_vpp_config()
70
71         # Start builtin server and client
72         uri = "tcp://" + self.loop0.local_ip4 + "/1234"
73         error = self.vapi.cli("test echo server appns 0 fifo-size 4 uri " +
74                               uri)
75         if error:
76             self.logger.critical(error)
77             self.assertEqual(error.find("failed"), -1)
78
79         error = self.vapi.cli("test echo client mbytes 10 appns 1 " +
80                               "fifo-size 4 no-output test-bytes " +
81                               "syn-timeout 2 uri " + uri)
82         if error:
83             self.logger.critical(error)
84             self.assertEqual(error.find("failed"), -1)
85
86         # Delete inter-table routes
87         ip_t01.remove_vpp_config()
88         ip_t10.remove_vpp_config()
89
90 if __name__ == '__main__':
91     unittest.main(testRunner=VppTestRunner)