aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_vhost.py
blob: 4e5a62d45351b16f1ada072ea5599c6083c1e7a7 (plain) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

#!/usr/bin/env python

import unittest

from framework import VppTestCase, VppTestRunner

from vpp_vhost_interface import VppVhostInterface


class TesVhostInterface(VppTestCase):
    """Vhost User Test Case

    """
    @classmethod
    def setUpClass(cls):
        super(TesVhostInterface, cls).setUpClass()

    @classmethod
    def tearDownClass(cls):
        super(TesVhostInterface, cls).tearDownClass()

    def tearDown(self):
        super(TesVhostInterface, self).tearDown()
        if not self.vpp_dead:
            if_dump = self.vapi.sw_interface_vhost_user_dump()
            for ifc in if_dump:
                self.vapi.delete_vhost_user_if(ifc.sw_if_index)

    def test_vhost(self):
        """ Vhost User add/delete interface test """
        self.logger.info("Vhost User add interfaces")

        # create interface 1 (VirtualEthernet0/0/0)
        vhost_if1 = VppVhostInterface(self, sock_filename=b'/tmp/sock1')
        vhost_if1.add_vpp_config()
        vhost_if1.admin_up()

        # create interface 2 (VirtualEthernet0/0/1)
        vhost_if2 = VppVhostInterface(self, sock_filename=b'/tmp/sock2')
        vhost_if2.add_vpp_config()
        vhost_if2.admin_up()

        # verify both interfaces in the show
        ifs = self.vapi.cli("show interface")
        self.assertIn('VirtualEthernet0/0/0', ifs)
        self.assertIn('VirtualEthernet0/0/1', ifs)

        # verify they are in the dump also
        if_dump = self.vapi.sw_interface_vhost_user_dump()
        self.assertTrue(vhost_if1.is_interface_config_in_dump(if_dump))
        self.assertTrue(vhost_if2.is_interface_config_in_dump(if_dump))

        # delete VirtualEthernet0/0/1
        self.logger.info("Deleting VirtualEthernet0/0/1")
        vhost_if2.remove_vpp_config()

        self.logger.info("Verifying VirtualEthernet0/0/1 is deleted")

        ifs = self.vapi.cli("show interface")
        # verify VirtualEthernet0/0/0 still in the show
        self.assertIn('VirtualEthernet0/0/0', ifs)

        # verify VirtualEthernet0/0/1 not in the show
        self.assertNotIn('VirtualEthernet0/0/1', ifs)

        # verify VirtualEthernet0/0/1 is not in the dump
        if_dump = self.vapi.sw_interface_vhost_user_dump()
        self.assertFalse(vhost_if2.is_interface_config_in_dump(if_dump))

        # verify VirtualEthernet0/0/0 is still in the dump
        self.assertTrue(vhost_if1.is_interface_config_in_dump(if_dump))

        # delete VirtualEthernet0/0/0
        self.logger.info("Deleting VirtualEthernet0/0/0")
        vhost_if1.remove_vpp_config()

        self.logger.info("Verifying VirtualEthernet0/0/0 is deleted")

        # verify VirtualEthernet0/0/0 not in the show
        ifs = self.vapi.cli("show interface")
        self.assertNotIn('VirtualEthernet0/0/0', ifs)

        # verify VirtualEthernet0/0/0 is not in the dump
        if_dump = self.vapi.sw_interface_vhost_user_dump()
        self.assertFalse(vhost_if1.is_interface_config_in_dump(if_dump))

    def test_vhost_interface_state(self):
        """ Vhost User interface states and events test """

        self.vapi.want_interface_events()

        # clear outstanding events
        # (like delete interface events from other tests)
        self.vapi.collect_events()

        vhost_if = VppVhostInterface(self, sock_filename=b'/tmp/sock1')

        # create vhost interface
        vhost_if.add_vpp_config()
        self.sleep(0.1)
        events = self.vapi.collect_events()
        # creating interface does now create events
        self.assert_equal(len(events), 1, "number of events")

        vhost_if.admin_up()
        vhost_if.assert_interface_state(1, 0, expect_event=True)

        vhost_if.admin_down()
        vhost_if.assert_interface_state(0, 0, expect_event=True)

        # delete vhost interface
        vhost_if.remove_vpp_config()
        event = self.vapi.wait_for_event(timeout=1)
        self.assert_equal(event.sw_if_index, vhost_if.sw_if_index,
                          "sw_if_index")
        self.assert_equal(event.deleted, 1, "deleted flag")

        # verify there are no more events
        events = self.vapi.collect_events()
        self.assert_equal(len(events), 0, "number of events")

if __name__ == '__main__':
    unittest.main(testRunner=VppTestRunner)
an class="n">_nodeconfig, value, path) def add_http_server_root(self, value=u"html"): """Add Http Server root configuration.""" path = [u"http", u"server", u"root"] self.add_config_item(self._nodeconfig, value, path) def add_http_server_index(self, value=u"index.html index.htm"): """Add Http Server index configuration.""" path = [u"http", u"server", u"index"] self.add_config_item(self._nodeconfig, value, path) def add_config_item(self, config, value, path): """Add NGINX configuration item. :param config: Startup configuration of node. :param value: Value to insert. :param path: Path where to insert item. :type config: dict :type value: str :type path: list """ if len(path) == 1: config[path[0]] = value return if path[0] not in config: config[path[0]] = dict() elif isinstance(config[path[0]], str): config[path[0]] = dict() if config[path[0]] == u"" \ else {config[path[0]]: u""} self.add_config_item(config[path[0]], value, path[1:]) def dump_config(self, obj, level=-1): """Dump the startup configuration in NGINX config format. :param obj: Python Object to print. :param level: Nested level for indentation. :type obj: Obj :type level: int :returns: nothing """ indent = u" " if level >= 0: self._nginx_config += f"{level * indent}{{\n" if isinstance(obj, dict): for key, val in obj.items(): if hasattr(val, u"__iter__") and not isinstance(val, str): self._nginx_config += f"{(level + 1) * indent}{key}\n" self.dump_config(val, level + 1) else: self._nginx_config += f"{(level + 1) * indent}" \ f"{key} {val};\n" else: for val in obj: self._nginx_config += f"{(level + 1) * indent}{val};\n" if level >= 0: self._nginx_config += f"{level * indent}}}\n" def write_config(self, filename=None): """Generate and write NGINX startup configuration to file. :param filename: NGINX configuration file name. :type filename: str """ if filename is None: filename = f"{self._nginx_path}/conf/nginx.conf" self.dump_config(self._nodeconfig) cmd = f"echo \"{self._nginx_config}\" | sudo tee {filename}" exec_cmd_no_error( self._node, cmd, message=u"Writing config file failed!" ) def add_http_server_location(self, size): """Add Http Server location configuration. :param size: File size. :type size: int """ if size == 0: files = u"return" elif size >= 1024: files = f"{int(size / 1024)}KB.json" else: files = f"{size}B.json" key = f"{files}" size_str = size * u"x" value = "200 '%s'" % size_str path = [u"http", u"server", f"location /{key}", u"return"] self.add_config_item(self._nodeconfig, value, path) def add_http_access_log(self, value=u"off"): """Add Http access_log configuration.""" path = [u"http", u"access_log"] self.add_config_item(self._nodeconfig, value, path) def add_http_include(self, value=u"mime.types"): """Add Http include configuration.""" path = [u"http", u"include"] self.add_config_item(self._nodeconfig, value, path) def add_http_default_type(self, value=u"application/octet-stream"): """Add Http default_type configuration.""" path = [u"http", u"default_type"] self.add_config_item(self._nodeconfig, value, path) def add_http_sendfile(self, value=u"on"): """Add Http sendfile configuration.""" path = [u"http", u"sendfile"] self.add_config_item(self._nodeconfig, value, path) def add_http_keepalive_timeout(self, value): """Add Http keepalive alive timeout configuration.""" path = [u"http", u"keepalive_timeout"] self.add_config_item(self._nodeconfig, value, path) def add_http_keepalive_requests(self, value): """Add Http keepalive alive requests configuration.""" path = [u"http", u"keepalive_requests"] self.add_config_item(self._nodeconfig, value, path) def add_events_use(self, value=u"epoll"): """Add Events use configuration.""" path = [u"events", u"use"] self.add_config_item(self._nodeconfig, value, path) def add_events_worker_connections(self, value=10240): """Add Events worker connections configuration.""" path = [u"events", u"worker_connections"] self.add_config_item(self._nodeconfig, value, path) def add_events_accept_mutex(self, value=u"off"): """Add Events accept mutex configuration.""" path = [u"events", u"accept_mutex"] self.add_config_item(self._nodeconfig, value, path) def add_events_multi_accept(self, value=u"off"): """Add Events multi accept configuration.""" path = [u"events", u"multi_accept"] self.add_config_item(self._nodeconfig, value, path) def add_worker_rlimit_nofile(self, value=10240): """Add Events worker rlimit nofile configuration.""" path = [u"worker_rlimit_nofile"] self.add_config_item(self._nodeconfig, value, path) def add_master_process(self, value=u"on"): """Add master process configuration.""" path = [u"master_process"] self.add_config_item(self._nodeconfig, value, path) def add_daemon(self, value=u"off"): """Add daemon configuration.""" path = [u"daemon"] self.add_config_item(self._nodeconfig, value, path) def add_worker_processes(self, value, smt_used): """Add worker processes configuration.""" # nginx workers : vpp used phy workers = 2:1 if smt_used: value = value * 4 else: value = value * 2 path = [u"worker_processes"] self.add_config_item(self._nodeconfig, value, path) def apply_config(self, filename=None, verify_nginx=True): """Generate and write NGINX configuration to file and verify configuration. :param filename: NGINX configuration file name. :param verify_nginx: Verify NGINX configuration. :type filename: str :type verify_nginx: bool """ self.write_config(filename=filename) app_path = f"{self._nginx_path}/sbin/nginx" if verify_nginx: NginxUtil.nginx_config_verify(self._node, app_path)