ruff: reformat

Author: dfa1

bpf_to_kafka.py

@@ -27,7 +27,9 @@ def _shutdown(sig: int, _: object) -> None:
     signal.signal(signal.SIGINT, _shutdown)
     signal.signal(signal.SIGTERM, _shutdown)
 
-    print(f"Tracing TCP connects to port {args.port} -> {args.topic}@{args.bootstrap_servers}")
+    print(
+        f"Tracing TCP connects to port {args.port} -> {args.topic}@{args.bootstrap_servers}"
+    )
     for event in source.events():
         sink.write(event)
 

tests/unit/test_bpf_source.py

@@ -7,9 +7,7 @@
 from sources.bpf import TASK_COMM_LEN, _PAYLOAD_LEN, BpfEventSource, _BpfEvent
 
 
-def _make_raw(
-    ts: int, pid: int, comm: bytes, payload: bytes
-) -> _BpfEvent:
+def _make_raw(ts: int, pid: int, comm: bytes, payload: bytes) -> _BpfEvent:
     e = _BpfEvent()
     e.ts = ts
     e.pid = pid
@@ -43,7 +41,10 @@ def test_yields_event_from_perf_buffer() -> None:
     source = _source_with_events([raw])
     (event,) = islice(source.events(), 1)
     assert event == Event(
-        timestamp=1_000_000_000, pid=1234, process="bash", payload="tcp_connect port=443"
+        timestamp=1_000_000_000,
+        pid=1234,
+        process="bash",
+        payload="tcp_connect port=443",
     )
 
 

tests/unit/test_debug_bpf_source.py

@@ -20,12 +20,12 @@ def _source_with_rows(
 
 
 def test_yields_event_from_trace_fields() -> None:
-    source = _source_with_rows(
-        [_make_row(b"bash", 1234, 1.0, b"execve /bin/ls")]
-    )
+    source = _source_with_rows([_make_row(b"bash", 1234, 1.0, b"execve /bin/ls")])
     events = list(source.events())
     assert events == [
-        Event(timestamp=1_000_000_000, pid=1234, process="bash", payload="execve /bin/ls")
+        Event(
+            timestamp=1_000_000_000, pid=1234, process="bash", payload="execve /bin/ls"
+        )
     ]
 
 
@@ -63,7 +63,9 @@ def test_skips_row_with_non_decodable_task() -> None:
 
 def test_skips_row_with_invalid_pid() -> None:
     mock_bpf = MagicMock()
-    mock_bpf.trace_fields.return_value = iter([(b"proc", "not-an-int", 0, 0, 1.0, b"msg")])
+    mock_bpf.trace_fields.return_value = iter(
+        [(b"proc", "not-an-int", 0, 0, 1.0, b"msg")]
+    )
     source = DebugBpfEventSource(mock_bpf)
     assert list(source.events()) == []