def test_revision(self, sock=None): myset = WiSet(name=self.name, attr_type="hash:net", sock=sock) myset.create() assert load_ipset(self.name, sock=sock).revision >= 6 myset.destroy()
def test_check_ipset_stats(self, sock=None): require_user('root') def test_stats(myset, res=None, counters=False): myset.counters = counters myset.create() myset.add("8.8.8.8", packets=res, bytes=res) myset.update_content() stats = myset.content myset.flush() assert stats["8.8.8.8"].packets == res assert stats["8.8.8.8"].bytes == res myset.add("8.8.8.8") myset.update_content() stats = myset.content if res is not None: res = 0 assert stats["8.8.8.8"].packets == res assert stats["8.8.8.8"].bytes == res myset.destroy() with WiSet(name=self.name, sock=sock) as myset: test_stats(myset, res=10, counters=True) test_stats(myset)
def test_force_attr_revision(self): sock = get_ipset_socket(attr_revision=2) myset = WiSet(name=self.name, attr_type="hash:net", sock=sock) myset.create() assert load_ipset(self.name, sock=sock).revision >= 2 myset.destroy() sock.close()
def test_ipset_with_comment(self, sock=None): comment = "test comment" with WiSet(name=self.name, sock=sock, comment=True) as myset: myset.create() myset.add("8.8.8.8", comment=comment) set_list = myset.content myset.destroy() assert set_list["8.8.8.8"].comment == comment
def test_hash_net_ipset(self, sock=None): to_add = ["192.168.1.0/24", "192.168.2.0/23", "10.0.0.0/8"] atype = "hash:net" with WiSet(name=self.name, attr_type=atype, sock=sock) as myset: myset.create() myset.insert_list(to_add) for value in to_add: assert value in myset.content myset.destroy()
def test_two_dimensions_ipset(self, sock=None): to_add = ["192.168.1.0/24,eth0", "192.168.2.0/23,eth1", "10.0.0.0/8,tun0"] atype = "hash:net,iface" with WiSet(name=self.name, attr_type=atype, sock=sock) as myset: myset.create() myset.insert_list(to_add) for value in to_add: assert value in myset.content myset.destroy()
def test_create_one_ipset(self, sock=None): with WiSet(name=self.name, sock=sock) as myset: myset.create() list_wiset = load_all_ipsets(sock=sock) assert test_ipset_exist(self.name, sock=sock) myset.destroy() assert not test_ipset_exist(self.name, sock=sock) assert self.name in list_wiset assert self.name not in load_all_ipsets(sock=sock)
def test_remove_entry(self, sock=None): ip = "1.1.1.1" with WiSet(name=self.name, sock=sock, counters=True) as myset: myset.create() myset.add(ip) assert ip in myset.content myset.delete(ip) myset.update_content() assert ip not in myset.content myset.destroy()
def test_physdev(self): myset = WiSet(name=self.name, attr_type="hash:net,iface") myset.create() myset.add("192.168.0.0/24,eth0", physdev=False) myset.add("192.168.1.0/24,eth0", physdev=True) content = myset.content myset.destroy() assert content["192.168.0.0/24,eth0"].physdev is False assert content["192.168.1.0/24,eth0"].physdev is True
def test_wildcard_entries(self): require_kernel(5, 5) myset = WiSet(name=self.name, attr_type="hash:net,iface") myset.create() myset.add("192.168.0.0/24,eth", wildcard=True) myset.add("192.168.1.0/24,wlan0", wildcard=False) content = myset.content myset.destroy() assert content["192.168.0.0/24,eth"].wildcard is True assert content["192.168.1.0/24,wlan0"].wildcard is False
def test_flush(self, sock=None): ip_list = ["1.2.3.4", "1.1.1.1", "7.7.7.7"] with WiSet(name=self.name, sock=sock) as myset: myset.create() for ip in ip_list: myset.add(ip) assert myset.test(ip) myset.flush() for ip in ip_list: assert not myset.test(ip) myset.destroy()
def test_add_ipstats(self, sock=None): data = IPStats(packets=10, bytes=1000, comment="hello world", skbmark="0x10/0x10", timeout=None) myset = WiSet(name=self.name, attr_type="hash:net", comment=True, skbinfo=True, counters=True, sock=sock) myset.create() myset.add("198.51.100.0/24", **data._asdict()) assert "198.51.100.0/24" in myset.content assert data == myset.content["198.51.100.0/24"] myset.destroy()
def test_replace(content_a, content_b): myset = WiSet(name=self.name, sock=sock) myset.create() myset.insert_list(content_a) myset.update_content() for value in content_a: assert value in myset.content myset.replace_entries(content_b) myset.update_content() for (old, new) in zip(content_a, content_b): assert old not in myset.content assert new in myset.content myset.destroy()
def test_basic_attribute_reads(self, sock=None): for value in [True, False]: myset = WiSet(name=self.name, sock=sock, counters=value, comment=value) if sock is None: myset.open_netlink() myset.create() from_netlink = load_ipset(self.name, sock=sock) assert value == from_netlink.comment assert value == from_netlink.counters myset.destroy() if sock is None: myset.close_netlink()
def test_create_ipset_twice(self, sock=None): with WiSet(name=self.name, sock=sock) as myset: myset.create() try: myset.create(exclusive=True) assert False except IPSetError: pass myset.create(exclusive=False) myset.destroy() assert self.name not in load_all_ipsets(sock=sock)
def test_ipset_with_skbinfo(self, sock=None): with WiSet(name=self.name, sock=sock, skbinfo=True) as myset: myset.create() myset.add("192.168.1.1", skbmark=(0xc8, 0xc8)) myset.add("192.168.1.2", skbmark=(0xc9, 0xffffffff)) myset.add("192.168.1.3", skbmark="0xca/0xca") myset.add("192.168.1.4", skbmark="0xCB") set_list = myset.content myset.destroy() assert set_list["192.168.1.1"].skbmark == "0xc8/0xc8" assert set_list["192.168.1.2"].skbmark == "0xc9" assert set_list["192.168.1.3"].skbmark == "0xca/0xca" assert set_list["192.168.1.4"].skbmark == "0xcb"
def test_list_in(self, sock=None): ip_list_good = ["1.2.3.4", "1.1.1.1", "7.7.7.7"] ip_list_bad = ["4.4.4.4", "5.5.5.5", "6.6.6.6"] with WiSet(name=self.name, sock=sock) as myset: myset.create() myset.replace_entries(ip_list_good) res_test = myset.test_list(ip_list_good + ip_list_bad) for ip in ip_list_good: assert ip in res_test for ip in ip_list_bad: assert ip not in res_test myset.destroy()
def test_timeout(self, sock=None): ip = "1.2.3.4" timeout = 2 with WiSet(name=self.name, sock=sock, timeout=timeout) as myset: myset.create() myset.add(ip) sleep(3) myset.update_content() assert ip not in myset.content assert timeout == load_ipset(self.name).timeout myset.add(ip, timeout=0) myset.update_content() assert 0 == myset.content[ip].timeout myset.destroy()
def test_list_on_large_set(self, sock=None): set_size = 30000 base_ip = "10.10.%d.%d" with WiSet(name=self.name, sock=sock) as myset: myset.create() for i in range(0, set_size): myset.add(base_ip % (i / 255, i % 255)) stats_len = len(myset.content) stats_len2 = len( load_ipset(self.name, content=True, sock=sock).content) myset.destroy() assert stats_len == set_size assert stats_len2 == set_size
def test_hashnet_with_comment(self, sock=None): comment = "abcdef" myset = WiSet(name=self.name, attr_type="hash:net", comment=True, sock=sock) myset.create() inherit_sock = sock is not None myset = load_ipset(self.name, sock=sock, inherit_sock=inherit_sock) assert myset.comment myset.add("192.168.1.1", comment=comment) myset.update_content() assert myset.content["192.168.1.1/32"].comment == comment myset.destroy()
def test_stats_consistency(self, sock=None): """ Test several way to fill the statistics of one IPSet """ entries = ["1.2.3.4", "1.2.3.5", "1.2.3.6"] myset = WiSet(name=self.name, sock=sock) myset.create() myset.insert_list(entries) myset_lists = load_all_ipsets(sock=sock, content=True)[self.name] for value in entries: assert value in myset_lists.content myset_list = load_ipset(self.name, sock=sock, content=True) for value in entries: assert value in myset_list.content myset.destroy()
def test_replace(content_a, content_b): myset = WiSet(name=self.name, sock=sock, comment=True) myset.create() myset.insert_list(content_a) myset.update_content() for value in content_a: assert value['entry'] in myset.content assert value['comment'] == ( myset.content[value['entry']].comment) myset.replace_entries(content_b) myset.update_content() for value in content_a: assert value['entry'] not in myset.content for value in content_b: assert value['entry'] in myset.content assert value['comment'] == ( myset.content[value['entry']].comment) myset.destroy()