chore(tests): fix python tests style

- fix test line spacing, import ordering, indentation, formatting
  function calls

Signed-off-by: aerosouund <aerosound161@gmail.com>

Author: aerosouund

tests/conftest.py

@@ -240,7 +240,11 @@ def bin_vsock_path(test_fc_session_root_path):
 def bin_vsock_seqpacket_listener_path(test_fc_session_root_path):
     """Build a simple vsock seqpacket server application."""
     vsock_seq_srv_bin_path = os.path.join(test_fc_session_root_path, "vsock_seq_server")
-    build_tools.gcc_compile("host_tools/vsock_seq_server.c", vsock_seq_srv_bin_path, extra_flags="-lpthread -O3")
+    build_tools.gcc_compile(
+        "host_tools/vsock_seq_server.c",
+        vsock_seq_srv_bin_path,
+        extra_flags="-lpthread -O3",
+    )
     yield vsock_seq_srv_bin_path
 
 

tests/framework/utils_vsock.py

@@ -111,6 +111,7 @@ def start_guest_echo_server(vm):
 
     return os.path.join(vm.jailer.chroot_path(), VSOCK_UDS_PATH)
 
+
 def start_seqpacket_echo_server(vm):
     """Start a vsock seqpacket echo server in the microVM.
 
@@ -147,16 +148,20 @@ def check_host_connections(uds_path, blob_path, blob_hash, vsock_type=SOCK_STREA
         assert wrk.hash == blob_hash
 
 
-def check_guest_connections_seqpacket(vm, server_port_path, server_bin_path ,blob_path, blob_hash):
+def check_guest_connections_seqpacket(
+    vm, server_port_path, server_bin_path, blob_path, blob_hash
+):
     """Test guest-initiated connections.
 
     This will start an echo server on the host (in its own thread), then
     start `TEST_CONNECTION_COUNT` workers inside the guest VM, all
     communicating with the echo server.
     """
-    port = server_port_path.split("_")[-1]  
+    port = server_port_path.split("_")[-1]
     if Path(server_port_path).exists():
-        Path(server_port_path).unlink()  # the vsock server program doesn't have reuseaddr
+        Path(
+            server_port_path
+        ).unlink()  # the vsock server program doesn't have reuseaddr
 
     echo_server = Popen([server_bin_path, "serve", port, "af_unix", server_port_path])
 
@@ -189,7 +194,9 @@ def check_guest_connections_seqpacket(vm, server_port_path, server_bin_path ,blo
         # comparison sets the exit status of the worker command.
         worker_cmd = "hash=$("
         worker_cmd += "cat {}".format(blob_path)
-        worker_cmd += " | /tmp/vsock_helper echo 2 {} seqpacket".format(ECHO_SERVER_PORT)
+        worker_cmd += " | /tmp/vsock_helper echo 2 {} seqpacket".format(
+            ECHO_SERVER_PORT
+        )
         worker_cmd += " | md5sum | cut -f1 -d\\ "
         worker_cmd += ")"
         worker_cmd += ' && [[ "$hash" = "{}" ]]'.format(blob_hash)
@@ -209,7 +216,7 @@ def check_guest_connections_seqpacket(vm, server_port_path, server_bin_path ,blo
     finally:
         echo_server.terminate()
         echo_server.wait()
-        
+
 
 def check_guest_connections(vm, server_port_path, blob_path, blob_hash):
     """Test guest-initiated connections.
@@ -295,7 +302,9 @@ def _vsock_connect_to_guest(uds_path, port, vsock_type):
     return sock
 
 
-def _copy_vsock_data_to_guest(ssh_connection, blob_path, vm_blob_path, vsock_helper=None, vsock_seq_server=None):
+def _copy_vsock_data_to_guest(
+    ssh_connection, blob_path, vm_blob_path, vsock_helper=None, vsock_seq_server=None
+):
     # Copy the data file and a vsock helper to the guest.
 
     cmd = "mkdir -p /tmp/vsock"

tests/integration_tests/functional/test_snapshot_basic.py

@@ -598,7 +598,9 @@ def test_snapshot_rename_vsock(
     """
 
     vm = uvm_nano
-    vm.api.vsock.put(vsock_id="vsock0", guest_cid=3, uds_path="/v.sock1", vsock_type="stream")
+    vm.api.vsock.put(
+        vsock_id="vsock0", guest_cid=3, uds_path="/v.sock1", vsock_type="stream"
+    )
     vm.add_net_iface()
     vm.start()
 

tests/integration_tests/functional/test_vsock.py

@@ -18,9 +18,9 @@
 import subprocess
 import time
 from pathlib import Path
+from socket import SOCK_SEQPACKET
 from socket import timeout as SocketTimeout
 from ssl import SOCK_STREAM
-from socket import SOCK_SEQPACKET
 from threading import Thread
 
 import pytest
@@ -32,13 +32,13 @@
     _copy_vsock_data_to_guest,
     _vsock_connect_to_guest,
     check_guest_connections,
-    check_host_connections,
     check_guest_connections_seqpacket,
+    check_host_connections,
     check_vsock_device,
     make_blob,
     make_host_port_path,
     start_guest_echo_server,
-    start_seqpacket_echo_server
+    start_seqpacket_echo_server,
 )
 from host_tools.fcmetrics import validate_fc_metrics
 
@@ -115,6 +115,7 @@ def negative_test_host_connections(vm, blob_path, blob_hash, vsock_type):
     metrics = vm.flush_metrics()
     validate_fc_metrics(metrics)
 
+
 def test_vsock_epipe(uvm_plain_any, bin_vsock_path, test_fc_session_root_path):
     """
     Vsock negative test to validate SIGPIPE/EPIPE handling.
@@ -313,7 +314,9 @@ def test_vsock_transport_reset_g2h(uvm_plain_any, microvm_factory):
         new_vm.kill()
 
 
-def test_vsock_seqpacket_h2g(uvm_plain_6_1, bin_vsock_seqpacket_listener_path, test_fc_session_root_path):
+def test_vsock_seqpacket_h2g(
+    uvm_plain_6_1, bin_vsock_seqpacket_listener_path, test_fc_session_root_path
+):
     vm = uvm_plain_6_1
     vm.spawn()
     vm.basic_config()
@@ -323,22 +326,32 @@ def test_vsock_seqpacket_h2g(uvm_plain_6_1, bin_vsock_seqpacket_listener_path, t
         guest_cid=3,
         uds_path=f"/{VSOCK_UDS_PATH}",
         vsock_type="seqpacket",
-        conn_buffer_size=16*1024,
+        conn_buffer_size=16 * 1024,
     )
     vm.start()
 
-    blob_path, blob_hash = make_blob(test_fc_session_root_path, 16*1024)
+    blob_path, blob_hash = make_blob(test_fc_session_root_path, 16 * 1024)
     vm_blob_path = "/tmp/vsock/test.blob"
 
-    _copy_vsock_data_to_guest(vm.ssh, blob_path, vm_blob_path, vsock_seq_server=bin_vsock_seqpacket_listener_path)
+    _copy_vsock_data_to_guest(
+        vm.ssh,
+        blob_path,
+        vm_blob_path,
+        vsock_seq_server=bin_vsock_seqpacket_listener_path,
+    )
     path = start_seqpacket_echo_server(vm)
 
     check_host_connections(path, blob_path, blob_hash, SOCK_SEQPACKET)
     metrics = vm.flush_metrics()
     validate_fc_metrics(metrics)
 
 
-def test_vsock_seqpacket_g2h(uvm_plain_6_1, bin_vsock_seqpacket_listener_path, bin_vsock_path, test_fc_session_root_path):
+def test_vsock_seqpacket_g2h(
+    uvm_plain_6_1,
+    bin_vsock_seqpacket_listener_path,
+    bin_vsock_path,
+    test_fc_session_root_path,
+):
     vm = uvm_plain_6_1
     vm.spawn()
     vm.basic_config()
@@ -348,23 +361,27 @@ def test_vsock_seqpacket_g2h(uvm_plain_6_1, bin_vsock_seqpacket_listener_path, b
         guest_cid=3,
         uds_path=f"/{VSOCK_UDS_PATH}",
         vsock_type="seqpacket",
-        conn_buffer_size=16*1024,
+        conn_buffer_size=16 * 1024,
     )
     vm.start()
 
-    blob_path, blob_hash = make_blob(test_fc_session_root_path, 16*1024)
+    blob_path, blob_hash = make_blob(test_fc_session_root_path, 16 * 1024)
     vm_blob_path = "/tmp/vsock/test.blob"
 
     _copy_vsock_data_to_guest(vm.ssh, blob_path, vm_blob_path, bin_vsock_path)
 
     path = os.path.join(vm.path, make_host_port_path(VSOCK_UDS_PATH, ECHO_SERVER_PORT))
-    check_guest_connections_seqpacket(vm, path, bin_vsock_seqpacket_listener_path, vm_blob_path, blob_hash)
+    check_guest_connections_seqpacket(
+        vm, path, bin_vsock_seqpacket_listener_path, vm_blob_path, blob_hash
+    )
 
     metrics = vm.flush_metrics()
     validate_fc_metrics(metrics)
 
 
-def test_vsock_seqpacket_h2g_overflow(uvm_plain_6_1, bin_vsock_seqpacket_listener_path, test_fc_session_root_path):
+def test_vsock_seqpacket_h2g_overflow(
+    uvm_plain_6_1, bin_vsock_seqpacket_listener_path, test_fc_session_root_path
+):
     """Test that sending a message larger than conn_buffer_size errors."""
     conn_buffer_size = 16 * 1024
 
@@ -383,7 +400,12 @@ def test_vsock_seqpacket_h2g_overflow(uvm_plain_6_1, bin_vsock_seqpacket_listene
 
     blob_path, _ = make_blob(test_fc_session_root_path, 1024)
     vm_blob_path = "/tmp/vsock/test.blob"
-    _copy_vsock_data_to_guest(vm.ssh, blob_path, vm_blob_path, vsock_seq_server=bin_vsock_seqpacket_listener_path)
+    _copy_vsock_data_to_guest(
+        vm.ssh,
+        blob_path,
+        vm_blob_path,
+        vsock_seq_server=bin_vsock_seqpacket_listener_path,
+    )
     path = start_seqpacket_echo_server(vm)
 
     worker_error = None
@@ -402,7 +424,9 @@ def worker():
     t.start()
     t.join()
 
-    assert worker_error is not None, "Expected an error when sending message larger than conn_buffer_size"
+    assert (
+        worker_error is not None
+    ), "Expected an error when sending message larger than conn_buffer_size"
 
 
 def test_vsock_after_override(