openstack
/
neutron-fwaas
Code Issues Proposed changes

Merge "python3: fix netlink_lib delete_entries"

This commit is contained in:
Zuul 2018-07-19 22:48:39 +00:00 committed by Gerrit Code Review
parent c601f34ab0 a3be36cd6e
commit 28d30b2bb0
2 changed files with 32 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
neutron_fwaas/privileged/netlink_lib.py
View File

@ -110,8 +110,9 @@ class ConntrackManager(object):
6: nfct.nfct_set_attr_u16},
'dport': {4: nfct.nfct_set_attr_u16,
6: nfct.nfct_set_attr_u16}, }
self.converters = {'src': str,
'dst': str,
self.converters = {'src': bytes,
'dst': bytes,
'ipversion': nl_constants.IPVERSION_SOCKET.get,
'protocol': constants.IP_PROTOCOL_MAP.get,
'code': int,
@ -169,7 +170,7 @@ class ConntrackManager(object):
dest = ctypes.create_string_buffer(
nl_constants.IPVERSION_BUFFER[addr_family])
libc.inet_pton(nl_constants.IPVERSION_SOCKET[addr_family],
source, dest)
source.encode('utf-8'), dest)
return dest.raw
def _set_attributes(self, conntrack, entry):

56
neutron_fwaas/tests/unit/privileged/test_netlink_lib.py
View File

@ -110,12 +110,12 @@ class NetlinkLibTestCase(base.BaseTestCase):
calls = [
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_SRC,
str(conntrack._convert_text_to_binary(
FAKE_ENTRY['src'], 4))),
conntrack._convert_text_to_binary(
FAKE_ENTRY['src'], 4)),
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_DST,
str(conntrack._convert_text_to_binary(
FAKE_ENTRY['dst'], 4))),
conntrack._convert_text_to_binary(
FAKE_ENTRY['dst'], 4)),
]
nl_lib.nfct.nfct_set_attr.assert_has_calls(calls, any_order=True)
nl_lib.nfct.nfct_destroy.assert_called_once()
@ -152,12 +152,12 @@ class NetlinkLibTestCase(base.BaseTestCase):
calls = [
mock.call(conntrack_filter,
nl_constants.ATTR_IPV6_SRC,
str(conntrack._convert_text_to_binary(
FAKE_ENTRY['src'], 6))),
conntrack._convert_text_to_binary(
FAKE_ENTRY['src'], 6)),
mock.call(conntrack_filter,
nl_constants.ATTR_IPV6_DST,
str(conntrack._convert_text_to_binary(
FAKE_ENTRY['dst'], 6))),
conntrack._convert_text_to_binary(
FAKE_ENTRY['dst'], 6)),
]
nl_lib.nfct.nfct_set_attr.assert_has_calls(calls, any_order=True)
nl_lib.nfct.nfct_destroy.assert_called_once()
@ -192,12 +192,12 @@ class NetlinkLibTestCase(base.BaseTestCase):
calls = [
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_SRC,
str(conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['src'], 4))),
conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['src'], 4)),
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_DST,
str(conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['dst'], 4))),
conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['dst'], 4)),
]
nl_lib.nfct.nfct_set_attr.assert_has_calls(calls, any_order=True)
nl_lib.nfct.nfct_destroy.assert_called_once()
@ -232,12 +232,12 @@ class NetlinkLibTestCase(base.BaseTestCase):
calls = [
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_SRC,
str(conntrack._convert_text_to_binary(
FAKE_TCP_ENTRY['src'], 4))),
conntrack._convert_text_to_binary(
FAKE_TCP_ENTRY['src'], 4)),
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_DST,
str(conntrack._convert_text_to_binary(
FAKE_TCP_ENTRY['dst'], 4))),
conntrack._convert_text_to_binary(
FAKE_TCP_ENTRY['dst'], 4)),
]
nl_lib.nfct.nfct_set_attr.assert_has_calls(calls, any_order=True)
nl_lib.nfct.nfct_destroy.assert_called_once()
@ -301,28 +301,28 @@ class NetlinkLibTestCase(base.BaseTestCase):
calls = [
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_SRC,
str(conntrack._convert_text_to_binary(
FAKE_TCP_ENTRY['src'], 4))),
conntrack._convert_text_to_binary(
FAKE_TCP_ENTRY['src'], 4)),
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_DST,
str(conntrack._convert_text_to_binary(
FAKE_TCP_ENTRY['dst'], 4))),
conntrack._convert_text_to_binary(
FAKE_TCP_ENTRY['dst'], 4)),
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_SRC,
str(conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['src'], 4))),
conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['src'], 4)),
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_DST,
str(conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['dst'], 4))),
conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['dst'], 4)),
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_SRC,
str(conntrack._convert_text_to_binary(
FAKE_ENTRY['src'], 4))),
conntrack._convert_text_to_binary(
FAKE_ENTRY['src'], 4)),
mock.call(conntrack_filter,
nl_constants.ATTR_IPV4_DST,
str(conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['dst'], 4))),
conntrack._convert_text_to_binary(
FAKE_UDP_ENTRY['dst'], 4)),
]
nl_lib.nfct.nfct_set_attr.assert_has_calls(calls, any_order=True)
nl_lib.nfct.nfct_destroy.assert_called_once()