refactor(tests): inline _check_tx_bandwidth

The function is too small, so inline it.

Signed-off-by: Egor Lazarchuk <yegorlz@amazon.co.uk>

Author: ShadowCurse

tests/integration_tests/performance/test_rate_limiter.py

@@ -161,7 +161,10 @@ def _check_tx_rate_limiting(test_microvm):
 
     # Second step: check bandwidth when rate limiting is on.
     print("Run guest TX iperf for rate limiting without burst")
-    _check_tx_bandwidth(test_microvm, eth1.host_ip, expected_kbps)
+    observed_kbps = _get_tx_bandwidth_with_duration(
+        test_microvm, eth1.host_ip, IPERF_TRANSMIT_TIME
+    )
+    assert _diff(observed_kbps, expected_kbps) < MAX_BYTES_DIFF_PERCENTAGE
 
     # Third step: get the number of bytes when rate limiting is on and there is
     # an initial burst size from where to consume.
@@ -179,7 +182,10 @@ def _check_tx_rate_limiting(test_microvm):
     assert _diff(burst_kbps, expected_kbps) > 100
 
     # Since the burst should be consumed, check rate limit is in place.
-    _check_tx_bandwidth(test_microvm, eth2.host_ip, expected_kbps)
+    observed_kbps = _get_tx_bandwidth_with_duration(
+        test_microvm, eth2.host_ip, IPERF_TRANSMIT_TIME
+    )
+    assert _diff(observed_kbps, expected_kbps) < MAX_BYTES_DIFF_PERCENTAGE
 
 
 def _check_rx_rate_limiting(test_microvm):
@@ -241,11 +247,17 @@ def _check_tx_rate_limit_patch(test_microvm):
     # Check that a TX rate limiter can be applied to a previously unlimited
     # interface.
     _patch_iface_bw(test_microvm, "eth0", "TX", bucket_size, REFILL_TIME_MS)
-    _check_tx_bandwidth(test_microvm, eth0.host_ip, expected_kbps)
+    observed_kbps = _get_tx_bandwidth_with_duration(
+        test_microvm, eth0.host_ip, IPERF_TRANSMIT_TIME
+    )
+    assert _diff(observed_kbps, expected_kbps) < MAX_BYTES_DIFF_PERCENTAGE
 
     # Check that a TX rate limiter can be updated.
     _patch_iface_bw(test_microvm, "eth1", "TX", bucket_size, REFILL_TIME_MS)
-    _check_tx_bandwidth(test_microvm, eth1.host_ip, expected_kbps)
+    observed_kbps = _get_tx_bandwidth_with_duration(
+        test_microvm, eth1.host_ip, IPERF_TRANSMIT_TIME
+    )
+    assert _diff(observed_kbps, expected_kbps) < MAX_BYTES_DIFF_PERCENTAGE
 
     # Check that a TX rate limiter can be removed.
     _patch_iface_bw(test_microvm, "eth0", "TX", 0, 0)
@@ -283,15 +295,6 @@ def _check_rx_rate_limit_patch(test_microvm):
     assert _diff(rate_no_limit_kbps, expected_kbps) > 50
 
 
-def _check_tx_bandwidth(test_microvm, ip, expected_kbps):
-    """Check that the rate-limited TX bandwidth is close to what we expect."""
-    observed_kbps = _get_tx_bandwidth_with_duration(
-        test_microvm, ip, IPERF_TRANSMIT_TIME
-    )
-    diff_pc = _diff(observed_kbps, expected_kbps)
-    assert diff_pc < MAX_BYTES_DIFF_PERCENTAGE
-
-
 def _get_tx_bandwidth_with_duration(test_microvm, host_ip, duration):
     """Check that the rate-limited TX bandwidth is close to what we expect."""