Refactor: apply pyformat for coding style consistency
add pyformat for presubmit hook Flag: EXEMPT mainline Bug: 419423721 Test: atest CtsConnectivityMultiDevicesTestCases Change-Id: I9abc5cee454e8c8e21ce7e3c134145857680251f
This commit is contained in:
Jimi Chen
parent
5e69785de3
commit
dcbe9b403d
1 changed files with 644 additions and 522 deletions
|
|
@ -25,22 +25,22 @@ from net_tests_utils.host.python.apf_utils import (
|
|||
get_apf_capabilities,
|
||||
get_apf_counter,
|
||||
get_apf_counters_from_cmd,
|
||||
get_ipv4_addresses,
|
||||
get_non_tentative_ipv6_addresses,
|
||||
get_exclude_all_host_ipv6_multicast_addresses,
|
||||
get_hardware_address,
|
||||
is_send_raw_packet_downstream_supported,
|
||||
is_packet_capture_supported,
|
||||
get_ipv4_addresses,
|
||||
get_matched_packet_counts,
|
||||
get_non_tentative_ipv6_addresses,
|
||||
is_apf_dump_counters_supported,
|
||||
is_packet_capture_supported,
|
||||
is_send_raw_packet_downstream_supported,
|
||||
send_raw_packet_downstream,
|
||||
start_capture_packets,
|
||||
stop_capture_packets,
|
||||
get_matched_packet_counts,
|
||||
send_raw_packet_downstream,
|
||||
)
|
||||
from net_tests_utils.host.python.assert_utils import UnexpectedBehaviorError
|
||||
|
||||
TEST_IFACE_NAME = "eth0"
|
||||
TEST_PACKET_IN_HEX = "AABBCCDDEEFF"
|
||||
TEST_IFACE_NAME = 'eth0'
|
||||
TEST_PACKET_IN_HEX = 'AABBCCDDEEFF'
|
||||
|
||||
|
||||
class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
||||
|
|
@ -51,27 +51,27 @@ class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
|||
def setup_test(self):
|
||||
self.mock_ad = MagicMock() # Mock Android device object
|
||||
|
||||
@patch("net_tests_utils.host.python.apf_utils.get_apf_counters_from_cmd")
|
||||
@patch('net_tests_utils.host.python.apf_utils.get_apf_counters_from_cmd')
|
||||
def test_get_apf_counter(self, mock_get_counters: MagicMock) -> None:
|
||||
iface = "wlan0"
|
||||
iface = 'wlan0'
|
||||
mock_get_counters.return_value = {
|
||||
"COUNTER_NAME1": "123",
|
||||
"COUNTER_NAME2": "456",
|
||||
"COUNTER_NAME3": "[ERROR: impossible]",
|
||||
'COUNTER_NAME1': '123',
|
||||
'COUNTER_NAME2': '456',
|
||||
'COUNTER_NAME3': '[ERROR: impossible]',
|
||||
}
|
||||
asserts.assert_equal(
|
||||
get_apf_counter(self.mock_ad, iface, "COUNTER_NAME1"), 123
|
||||
get_apf_counter(self.mock_ad, iface, 'COUNTER_NAME1'), 123
|
||||
)
|
||||
# Invalid counter value
|
||||
asserts.assert_equal(
|
||||
get_apf_counter(self.mock_ad, iface, "COUNTER_NAME3"), -1
|
||||
get_apf_counter(self.mock_ad, iface, 'COUNTER_NAME3'), -1
|
||||
)
|
||||
# Not found
|
||||
asserts.assert_equal(
|
||||
get_apf_counter(self.mock_ad, iface, "COUNTER_NAME4"), 0
|
||||
get_apf_counter(self.mock_ad, iface, 'COUNTER_NAME4'), 0
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_hardware_address_success(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
|
|
@ -83,21 +83,19 @@ class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
|||
47: wlan1: <BROADCAST,MULTICAST> mtu 1500 qdisc ...
|
||||
link/ether 6a:16:81:ff:82:9b brd ff:ff:ff:ff:ff:ff
|
||||
"""
|
||||
mac_address = get_hardware_address(self.mock_ad, "wlan0")
|
||||
asserts.assert_equal(mac_address, "72:05:77:82:21:E0")
|
||||
mac_address = get_hardware_address(self.mock_ad, 'wlan0')
|
||||
asserts.assert_equal(mac_address, '72:05:77:82:21:E0')
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_hardware_address_not_found(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = "Some output without MAC address"
|
||||
mock_adb_shell.return_value = 'Some output without MAC address'
|
||||
with asserts.assert_raises(PatternNotFoundException):
|
||||
get_hardware_address(self.mock_ad, "wlan0")
|
||||
get_hardware_address(self.mock_ad, 'wlan0')
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
def test_get_ipv4_addresses_success(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_ipv4_addresses_success(self, mock_adb_shell: MagicMock) -> None:
|
||||
mock_adb_shell.return_value = """
|
||||
54: wlan0: <BROADCAST,MULTICAST,UP,LOWER_UP> mtu 1500 state UP qlen 1000
|
||||
inet 192.168.195.162/24 brd 192.168.195.255 scope global wlan0
|
||||
|
|
@ -105,18 +103,18 @@ class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
|||
inet 192.168.200.1/24 brd 192.168.200.255 scope global wlan0
|
||||
valid_lft forever preferred_lft forever
|
||||
"""
|
||||
ip_addresses = get_ipv4_addresses(self.mock_ad, "wlan0")
|
||||
asserts.assert_equal(ip_addresses, ["192.168.195.162", "192.168.200.1"])
|
||||
ip_addresses = get_ipv4_addresses(self.mock_ad, 'wlan0')
|
||||
asserts.assert_equal(ip_addresses, ['192.168.195.162', '192.168.200.1'])
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_ipv4_addresses_not_found(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = ""
|
||||
ip_addresses = get_ipv4_addresses(self.mock_ad, "wlan0")
|
||||
mock_adb_shell.return_value = ''
|
||||
ip_addresses = get_ipv4_addresses(self.mock_ad, 'wlan0')
|
||||
asserts.assert_equal(ip_addresses, [])
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_non_tentative_ipv6_addresses_success(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
|
|
@ -129,22 +127,23 @@ class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
|||
inet6 fe80::3aff:2199:2d8e:20d1/64 scope link noprefixroute
|
||||
valid_lft forever preferred_lft forever
|
||||
"""
|
||||
ip_addresses = get_non_tentative_ipv6_addresses(self.mock_ad, "wlan0")
|
||||
asserts.assert_equal(ip_addresses,
|
||||
["fe80::10a3:5dff:fe52:de32",
|
||||
"2001:b400:e53f:164e:9c1e:780e:d1:4658",
|
||||
"fe80::3aff:2199:2d8e:20d1"])
|
||||
ip_addresses = get_non_tentative_ipv6_addresses(self.mock_ad, 'wlan0')
|
||||
asserts.assert_equal(
|
||||
ip_addresses,
|
||||
[
|
||||
'fe80::10a3:5dff:fe52:de32',
|
||||
'2001:b400:e53f:164e:9c1e:780e:d1:4658',
|
||||
'fe80::3aff:2199:2d8e:20d1',
|
||||
],
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
def test_get_ipv6_address_not_found(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = ""
|
||||
ip_addresses = get_non_tentative_ipv6_addresses(self.mock_ad, "wlan0")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_ipv6_address_not_found(self, mock_adb_shell: MagicMock) -> None:
|
||||
mock_adb_shell.return_value = ''
|
||||
ip_addresses = get_non_tentative_ipv6_addresses(self.mock_ad, 'wlan0')
|
||||
asserts.assert_equal(ip_addresses, [])
|
||||
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_exclude_all_host_ipv6_multicast_addresses_success(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
|
|
@ -155,39 +154,43 @@ class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
|||
inet6 ff02::1
|
||||
inet6 ff01::1
|
||||
"""
|
||||
ip_addresses = get_exclude_all_host_ipv6_multicast_addresses(self.mock_ad, "wlan0")
|
||||
asserts.assert_equal(ip_addresses,
|
||||
["ff02::1:ff99:37b0",
|
||||
"ff02::1:ffb7:cba2"])
|
||||
ip_addresses = get_exclude_all_host_ipv6_multicast_addresses(
|
||||
self.mock_ad, 'wlan0'
|
||||
)
|
||||
asserts.assert_equal(
|
||||
ip_addresses, ['ff02::1:ff99:37b0', 'ff02::1:ffb7:cba2']
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_exclude_all_host_ipv6_multicast_addresses_not_found(
|
||||
self, mock_adb_shell: MagicMock
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = ""
|
||||
ip_addresses = get_exclude_all_host_ipv6_multicast_addresses(self.mock_ad, "wlan0")
|
||||
mock_adb_shell.return_value = ''
|
||||
ip_addresses = get_exclude_all_host_ipv6_multicast_addresses(
|
||||
self.mock_ad, 'wlan0'
|
||||
)
|
||||
asserts.assert_equal(ip_addresses, [])
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_send_raw_packet_downstream_success(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = "" # Successful command output
|
||||
mock_adb_shell.return_value = '' # Successful command output
|
||||
send_raw_packet_downstream(
|
||||
self.mock_ad, TEST_IFACE_NAME, TEST_PACKET_IN_HEX
|
||||
)
|
||||
mock_adb_shell.assert_called_once_with(
|
||||
self.mock_ad,
|
||||
"cmd network_stack send-raw-packet-downstream"
|
||||
f" {TEST_IFACE_NAME} {TEST_PACKET_IN_HEX}",
|
||||
'cmd network_stack send-raw-packet-downstream'
|
||||
f' {TEST_IFACE_NAME} {TEST_PACKET_IN_HEX}',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_send_raw_packet_downstream_failure(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = ( # Unexpected command output
|
||||
"Any Unexpected Output"
|
||||
'Any Unexpected Output'
|
||||
)
|
||||
with asserts.assert_raises(UnexpectedBehaviorError):
|
||||
send_raw_packet_downstream(
|
||||
|
|
@ -195,15 +198,15 @@ class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
|||
)
|
||||
asserts.assert_true(
|
||||
is_send_raw_packet_downstream_supported(self.mock_ad),
|
||||
"Send raw packet should be supported.",
|
||||
'Send raw packet should be supported.',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_send_raw_packet_downstream_unsupported(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.side_effect = AdbError(
|
||||
cmd="", stdout="Unknown command", stderr="", ret_code=3
|
||||
cmd='', stdout='Unknown command', stderr='', ret_code=3
|
||||
)
|
||||
with asserts.assert_raises(UnsupportedOperationException):
|
||||
send_raw_packet_downstream(
|
||||
|
|
@ -211,152 +214,122 @@ class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
|||
)
|
||||
asserts.assert_false(
|
||||
is_send_raw_packet_downstream_supported(self.mock_ad),
|
||||
"Send raw packet should not be supported.",
|
||||
'Send raw packet should not be supported.',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
def test_start_capture_success(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = "success" # Successful command output
|
||||
start_capture_packets(
|
||||
self.mock_ad, TEST_IFACE_NAME
|
||||
)
|
||||
mock_adb_shell.assert_called_once_with(
|
||||
self.mock_ad,
|
||||
"cmd network_stack capture start"
|
||||
f" {TEST_IFACE_NAME}"
|
||||
)
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_start_capture_success(self, mock_adb_shell: MagicMock) -> None:
|
||||
mock_adb_shell.return_value = 'success' # Successful command output
|
||||
start_capture_packets(self.mock_ad, TEST_IFACE_NAME)
|
||||
mock_adb_shell.assert_called_once_with(
|
||||
self.mock_ad, f'cmd network_stack capture start {TEST_IFACE_NAME}'
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
def test_start_capture_failure(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = ( # Unexpected command output
|
||||
"Any Unexpected Output"
|
||||
)
|
||||
with asserts.assert_raises(UnexpectedBehaviorError):
|
||||
start_capture_packets(
|
||||
self.mock_ad, TEST_IFACE_NAME
|
||||
)
|
||||
asserts.assert_true(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
"Start capturing packets should be supported.",
|
||||
)
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_start_capture_failure(self, mock_adb_shell: MagicMock) -> None:
|
||||
mock_adb_shell.return_value = ( # Unexpected command output
|
||||
'Any Unexpected Output'
|
||||
)
|
||||
with asserts.assert_raises(UnexpectedBehaviorError):
|
||||
start_capture_packets(self.mock_ad, TEST_IFACE_NAME)
|
||||
asserts.assert_true(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
'Start capturing packets should be supported.',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
def test_start_capture_unsupported(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.side_effect = AdbError(
|
||||
cmd="", stdout="Unknown command", stderr="", ret_code=3
|
||||
)
|
||||
with asserts.assert_raises(UnsupportedOperationException):
|
||||
start_capture_packets(
|
||||
self.mock_ad, TEST_IFACE_NAME
|
||||
)
|
||||
asserts.assert_false(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
"Start capturing packets should not be supported.",
|
||||
)
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_start_capture_unsupported(self, mock_adb_shell: MagicMock) -> None:
|
||||
mock_adb_shell.side_effect = AdbError(
|
||||
cmd='', stdout='Unknown command', stderr='', ret_code=3
|
||||
)
|
||||
with asserts.assert_raises(UnsupportedOperationException):
|
||||
start_capture_packets(self.mock_ad, TEST_IFACE_NAME)
|
||||
asserts.assert_false(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
'Start capturing packets should not be supported.',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
def test_stop_capture_success(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = "success" # Successful command output
|
||||
stop_capture_packets(
|
||||
self.mock_ad, TEST_IFACE_NAME
|
||||
)
|
||||
mock_adb_shell.assert_called_once_with(
|
||||
self.mock_ad,
|
||||
"cmd network_stack capture stop"
|
||||
f" {TEST_IFACE_NAME}"
|
||||
)
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_stop_capture_success(self, mock_adb_shell: MagicMock) -> None:
|
||||
mock_adb_shell.return_value = 'success' # Successful command output
|
||||
stop_capture_packets(self.mock_ad, TEST_IFACE_NAME)
|
||||
mock_adb_shell.assert_called_once_with(
|
||||
self.mock_ad, f'cmd network_stack capture stop {TEST_IFACE_NAME}'
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
def test_stop_capture_failure(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = ( # Unexpected command output
|
||||
"Any Unexpected Output"
|
||||
)
|
||||
with asserts.assert_raises(UnexpectedBehaviorError):
|
||||
stop_capture_packets(
|
||||
self.mock_ad, TEST_IFACE_NAME
|
||||
)
|
||||
asserts.assert_true(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
"Stop capturing packets should be supported.",
|
||||
)
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_stop_capture_failure(self, mock_adb_shell: MagicMock) -> None:
|
||||
mock_adb_shell.return_value = ( # Unexpected command output
|
||||
'Any Unexpected Output'
|
||||
)
|
||||
with asserts.assert_raises(UnexpectedBehaviorError):
|
||||
stop_capture_packets(self.mock_ad, TEST_IFACE_NAME)
|
||||
asserts.assert_true(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
'Stop capturing packets should be supported.',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
def test_stop_capture_unsupported(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.side_effect = AdbError(
|
||||
cmd="", stdout="Unknown command", stderr="", ret_code=3
|
||||
)
|
||||
with asserts.assert_raises(UnsupportedOperationException):
|
||||
stop_capture_packets(
|
||||
self.mock_ad, TEST_IFACE_NAME
|
||||
)
|
||||
asserts.assert_false(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
"Stop capturing packets should not be supported.",
|
||||
)
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_stop_capture_unsupported(self, mock_adb_shell: MagicMock) -> None:
|
||||
mock_adb_shell.side_effect = AdbError(
|
||||
cmd='', stdout='Unknown command', stderr='', ret_code=3
|
||||
)
|
||||
with asserts.assert_raises(UnsupportedOperationException):
|
||||
stop_capture_packets(self.mock_ad, TEST_IFACE_NAME)
|
||||
asserts.assert_false(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
'Stop capturing packets should not be supported.',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_matched_packet_counts_success(
|
||||
self, mock_adb_shell: MagicMock
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = "10" # Successful command output
|
||||
mock_adb_shell.return_value = '10' # Successful command output
|
||||
get_matched_packet_counts(self.mock_ad, TEST_IFACE_NAME, TEST_PACKET_IN_HEX)
|
||||
mock_adb_shell.assert_called_once_with(
|
||||
self.mock_ad,
|
||||
'cmd network_stack capture matched-packet-counts'
|
||||
f' {TEST_IFACE_NAME} {TEST_PACKET_IN_HEX}',
|
||||
)
|
||||
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_matched_packet_counts_failure(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = ( # Unexpected command output
|
||||
'Any Unexpected Output'
|
||||
)
|
||||
with asserts.assert_raises(UnexpectedBehaviorError):
|
||||
get_matched_packet_counts(
|
||||
self.mock_ad, TEST_IFACE_NAME, TEST_PACKET_IN_HEX
|
||||
)
|
||||
mock_adb_shell.assert_called_once_with(
|
||||
self.mock_ad,
|
||||
"cmd network_stack capture matched-packet-counts"
|
||||
f" {TEST_IFACE_NAME} {TEST_PACKET_IN_HEX}"
|
||||
)
|
||||
asserts.assert_true(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
'Get matched packet counts should be supported.',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
def test_get_matched_packet_counts_failure(
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = ( # Unexpected command output
|
||||
"Any Unexpected Output"
|
||||
)
|
||||
with asserts.assert_raises(UnexpectedBehaviorError):
|
||||
get_matched_packet_counts(
|
||||
self.mock_ad, TEST_IFACE_NAME, TEST_PACKET_IN_HEX
|
||||
)
|
||||
asserts.assert_true(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
"Get matched packet counts should be supported.",
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_matched_packet_counts_unsupported(
|
||||
self, mock_adb_shell: MagicMock
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.side_effect = AdbError(
|
||||
cmd="", stdout="Unknown command", stderr="", ret_code=3
|
||||
)
|
||||
with asserts.assert_raises(UnsupportedOperationException):
|
||||
get_matched_packet_counts(
|
||||
self.mock_ad, TEST_IFACE_NAME, TEST_PACKET_IN_HEX
|
||||
)
|
||||
asserts.assert_false(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
"Get matched packet counts should not be supported.",
|
||||
mock_adb_shell.side_effect = AdbError(
|
||||
cmd='', stdout='Unknown command', stderr='', ret_code=3
|
||||
)
|
||||
with asserts.assert_raises(UnsupportedOperationException):
|
||||
get_matched_packet_counts(
|
||||
self.mock_ad, TEST_IFACE_NAME, TEST_PACKET_IN_HEX
|
||||
)
|
||||
asserts.assert_false(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
'Get matched packet counts should not be supported.',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_apf_counters_from_cmd_success(
|
||||
self, mock_adb_shell: MagicMock
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = """
|
||||
mock_adb_shell.return_value = """
|
||||
NDIANNESS: 305419896
|
||||
TOTAL_PACKETS: 157553
|
||||
FILTER_AGE_SECONDS: 356
|
||||
|
|
@ -380,53 +353,46 @@ class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
|||
DROPPED_ARP_OTHER_HOST: 89
|
||||
DROPPED_ARP_REQUEST_REPLIED: 13
|
||||
"""
|
||||
get_apf_counters_from_cmd(
|
||||
self.mock_ad, TEST_IFACE_NAME
|
||||
)
|
||||
mock_adb_shell.assert_called_once_with(
|
||||
self.mock_ad,
|
||||
f"cmd network_stack apf {TEST_IFACE_NAME} dump-counters"
|
||||
)
|
||||
get_apf_counters_from_cmd(self.mock_ad, TEST_IFACE_NAME)
|
||||
mock_adb_shell.assert_called_once_with(
|
||||
self.mock_ad, f'cmd network_stack apf {TEST_IFACE_NAME} dump-counters'
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_apf_counters_from_cmd_failure(
|
||||
self, mock_adb_shell: MagicMock
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.return_value = ( # Unexpected command output
|
||||
"Any Unexpected Output"
|
||||
)
|
||||
with asserts.assert_raises(UnexpectedBehaviorError):
|
||||
get_apf_counters_from_cmd(
|
||||
self.mock_ad, TEST_IFACE_NAME
|
||||
)
|
||||
asserts.assert_true(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
"Dump APF packet counters should be supported.",
|
||||
)
|
||||
mock_adb_shell.return_value = ( # Unexpected command output
|
||||
'Any Unexpected Output'
|
||||
)
|
||||
with asserts.assert_raises(UnexpectedBehaviorError):
|
||||
get_apf_counters_from_cmd(self.mock_ad, TEST_IFACE_NAME)
|
||||
asserts.assert_true(
|
||||
is_packet_capture_supported(self.mock_ad),
|
||||
'Dump APF packet counters should be supported.',
|
||||
)
|
||||
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_apf_counters_from_cmd_unsupported(
|
||||
self, mock_adb_shell: MagicMock
|
||||
self, mock_adb_shell: MagicMock
|
||||
) -> None:
|
||||
mock_adb_shell.side_effect = AdbError(
|
||||
cmd="", stdout="Unknown command", stderr="", ret_code=3
|
||||
)
|
||||
with asserts.assert_raises(UnsupportedOperationException):
|
||||
get_apf_counters_from_cmd(
|
||||
self.mock_ad, TEST_IFACE_NAME
|
||||
)
|
||||
asserts.assert_false(
|
||||
is_apf_dump_counters_supported(self.mock_ad),
|
||||
"Dump APF packet counters should not be supported.",
|
||||
)
|
||||
mock_adb_shell.side_effect = AdbError(
|
||||
cmd='', stdout='Unknown command', stderr='', ret_code=3
|
||||
)
|
||||
with asserts.assert_raises(UnsupportedOperationException):
|
||||
get_apf_counters_from_cmd(self.mock_ad, TEST_IFACE_NAME)
|
||||
asserts.assert_false(
|
||||
is_apf_dump_counters_supported(self.mock_ad),
|
||||
'Dump APF packet counters should not be supported.',
|
||||
)
|
||||
|
||||
@parameterized.parameters(
|
||||
("2,2048,1", ApfCapabilities(2, 2048, 1)), # Valid input
|
||||
("3,1024,0", ApfCapabilities(3, 1024, 0)), # Valid input
|
||||
("invalid,output", ApfCapabilities(0, 0, 0)), # Invalid input
|
||||
("", ApfCapabilities(0, 0, 0)), # Empty input
|
||||
('2,2048,1', ApfCapabilities(2, 2048, 1)), # Valid input
|
||||
('3,1024,0', ApfCapabilities(3, 1024, 0)), # Valid input
|
||||
('invalid,output', ApfCapabilities(0, 0, 0)), # Invalid input
|
||||
('', ApfCapabilities(0, 0, 0)), # Empty input
|
||||
)
|
||||
@patch("net_tests_utils.host.python.adb_utils.adb_shell")
|
||||
@patch('net_tests_utils.host.python.adb_utils.adb_shell')
|
||||
def test_get_apf_capabilities(
|
||||
self, mock_output, expected_result, mock_adb_shell
|
||||
):
|
||||
|
|
@ -435,7 +401,7 @@ class TestApfUtils(base_test.BaseTestClass, parameterized.TestCase):
|
|||
mock_adb_shell.return_value = mock_output
|
||||
|
||||
# Call the function under test
|
||||
result = get_apf_capabilities(self.mock_ad, "wlan0")
|
||||
result = get_apf_capabilities(self.mock_ad, 'wlan0')
|
||||
|
||||
# Assert that the result matches the expected result
|
||||
asserts.assert_equal(result, expected_result)
|
||||
|
|
|
|||
Loading…
Reference in a new issue