summaryrefslogtreecommitdiffstats
path: root/test/vpp_igmp.py
blob: d1a308878c561a6070e439eccedcf3c04dc8cbff (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
from vpp_object import VppObject


class IgmpSG():
    def __init__(self, saddr, gaddr):
        self.saddr = saddr
        self.gaddr = gaddr


class VppIgmpConfig(VppObject):
    def __init__(self, test, sw_if_index, sg=None):
        self._test = test
        self.sw_if_index = sw_if_index
        if isinstance(sg, list):
            self.sg_list = sg
        else:
            self.sg_list = []
            self.sg_list.append(sg)

    def add_sg(self, sg):
        self.sg.append(sg)

    def add_vpp_config(self):
        for e in self.sg_list:
            self._test.vapi.igmp_listen(
                1, self.sw_if_index, e.saddr, e.gaddr)

    def remove_vpp_config(self):
        self._test.vapi.igmp_clear_interface(self.sw_if_index)

    def __str__(self):
        return self.object_id()

    def object_id(self):
        return "%s:%d" % (self.sg_list, self.sw_if_index)

    def query_vpp_config(self):
        return self._test.vapi.igmp_dump()