mirror of
https://github.com/smittix/intercept.git
synced 2026-07-07 09:08:12 -07:00
style: apply ruff-format to entire codebase
First-time run of ruff-format via pre-commit hook normalises quote style, trailing commas, and whitespace across 188 Python files. No logic changes. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -13,87 +13,106 @@ class TestFlightCorrelator:
|
||||
self.correlator = FlightCorrelator(max_messages=100)
|
||||
|
||||
def test_add_acars_message(self):
|
||||
self.correlator.add_acars_message({
|
||||
'flight': 'BAW123', 'tail': 'G-ABCD', 'text': 'Hello',
|
||||
})
|
||||
self.correlator.add_acars_message(
|
||||
{
|
||||
"flight": "BAW123",
|
||||
"tail": "G-ABCD",
|
||||
"text": "Hello",
|
||||
}
|
||||
)
|
||||
assert self.correlator.acars_count == 1
|
||||
|
||||
def test_add_vdl2_message(self):
|
||||
self.correlator.add_vdl2_message({
|
||||
'flight': 'DLH456', 'text': 'World',
|
||||
})
|
||||
self.correlator.add_vdl2_message(
|
||||
{
|
||||
"flight": "DLH456",
|
||||
"text": "World",
|
||||
}
|
||||
)
|
||||
assert self.correlator.vdl2_count == 1
|
||||
|
||||
def test_match_by_callsign(self):
|
||||
self.correlator.add_acars_message({
|
||||
'flight': 'BAW123', 'text': 'msg1',
|
||||
})
|
||||
self.correlator.add_acars_message({
|
||||
'flight': 'DLH456', 'text': 'msg2',
|
||||
})
|
||||
self.correlator.add_acars_message(
|
||||
{
|
||||
"flight": "BAW123",
|
||||
"text": "msg1",
|
||||
}
|
||||
)
|
||||
self.correlator.add_acars_message(
|
||||
{
|
||||
"flight": "DLH456",
|
||||
"text": "msg2",
|
||||
}
|
||||
)
|
||||
|
||||
result = self.correlator.get_messages_for_aircraft(callsign='BAW123')
|
||||
assert len(result['acars']) == 1
|
||||
assert result['acars'][0]['text'] == 'msg1'
|
||||
result = self.correlator.get_messages_for_aircraft(callsign="BAW123")
|
||||
assert len(result["acars"]) == 1
|
||||
assert result["acars"][0]["text"] == "msg1"
|
||||
|
||||
def test_match_by_icao(self):
|
||||
self.correlator.add_vdl2_message({
|
||||
'icao': 'ABC123', 'text': 'vdl2 msg',
|
||||
})
|
||||
self.correlator.add_vdl2_message(
|
||||
{
|
||||
"icao": "ABC123",
|
||||
"text": "vdl2 msg",
|
||||
}
|
||||
)
|
||||
|
||||
result = self.correlator.get_messages_for_aircraft(icao='ABC123')
|
||||
assert len(result['vdl2']) == 1
|
||||
assert result['vdl2'][0]['text'] == 'vdl2 msg'
|
||||
result = self.correlator.get_messages_for_aircraft(icao="ABC123")
|
||||
assert len(result["vdl2"]) == 1
|
||||
assert result["vdl2"][0]["text"] == "vdl2 msg"
|
||||
|
||||
def test_no_match_returns_empty(self):
|
||||
self.correlator.add_acars_message({'flight': 'BAW123', 'text': 'msg'})
|
||||
self.correlator.add_acars_message({"flight": "BAW123", "text": "msg"})
|
||||
|
||||
result = self.correlator.get_messages_for_aircraft(callsign='NOMATCH')
|
||||
assert result['acars'] == []
|
||||
assert result['vdl2'] == []
|
||||
result = self.correlator.get_messages_for_aircraft(callsign="NOMATCH")
|
||||
assert result["acars"] == []
|
||||
assert result["vdl2"] == []
|
||||
|
||||
def test_empty_search_returns_empty(self):
|
||||
result = self.correlator.get_messages_for_aircraft()
|
||||
assert result == {'acars': [], 'vdl2': []}
|
||||
assert result == {"acars": [], "vdl2": []}
|
||||
|
||||
def test_ring_buffer_limit(self):
|
||||
correlator = FlightCorrelator(max_messages=5)
|
||||
for i in range(10):
|
||||
correlator.add_acars_message({'flight': f'FL{i}', 'text': f'msg{i}'})
|
||||
correlator.add_acars_message({"flight": f"FL{i}", "text": f"msg{i}"})
|
||||
|
||||
assert correlator.acars_count == 5
|
||||
# First 5 messages should have been evicted
|
||||
result = correlator.get_messages_for_aircraft(callsign='FL0')
|
||||
assert len(result['acars']) == 0
|
||||
result = correlator.get_messages_for_aircraft(callsign="FL0")
|
||||
assert len(result["acars"]) == 0
|
||||
# Last message should still be there
|
||||
result = correlator.get_messages_for_aircraft(callsign='FL9')
|
||||
assert len(result['acars']) == 1
|
||||
result = correlator.get_messages_for_aircraft(callsign="FL9")
|
||||
assert len(result["acars"]) == 1
|
||||
|
||||
def test_case_insensitive_matching(self):
|
||||
self.correlator.add_acars_message({'flight': 'baw123', 'text': 'lowercase'})
|
||||
self.correlator.add_acars_message({"flight": "baw123", "text": "lowercase"})
|
||||
|
||||
result = self.correlator.get_messages_for_aircraft(callsign='BAW123')
|
||||
assert len(result['acars']) == 1
|
||||
result = self.correlator.get_messages_for_aircraft(callsign="BAW123")
|
||||
assert len(result["acars"]) == 1
|
||||
|
||||
def test_match_by_tail_field(self):
|
||||
self.correlator.add_acars_message({
|
||||
'tail': 'G-ABCD', 'text': 'tail match',
|
||||
})
|
||||
self.correlator.add_acars_message(
|
||||
{
|
||||
"tail": "G-ABCD",
|
||||
"text": "tail match",
|
||||
}
|
||||
)
|
||||
|
||||
result = self.correlator.get_messages_for_aircraft(callsign='G-ABCD')
|
||||
assert len(result['acars']) == 1
|
||||
result = self.correlator.get_messages_for_aircraft(callsign="G-ABCD")
|
||||
assert len(result["acars"]) == 1
|
||||
|
||||
def test_internal_fields_not_returned(self):
|
||||
self.correlator.add_acars_message({'flight': 'TEST', 'text': 'msg'})
|
||||
self.correlator.add_acars_message({"flight": "TEST", "text": "msg"})
|
||||
|
||||
result = self.correlator.get_messages_for_aircraft(callsign='TEST')
|
||||
msg = result['acars'][0]
|
||||
assert '_corr_time' not in msg
|
||||
result = self.correlator.get_messages_for_aircraft(callsign="TEST")
|
||||
msg = result["acars"][0]
|
||||
assert "_corr_time" not in msg
|
||||
|
||||
def test_both_acars_and_vdl2_returned(self):
|
||||
self.correlator.add_acars_message({'flight': 'UAL789', 'text': 'acars'})
|
||||
self.correlator.add_vdl2_message({'flight': 'UAL789', 'text': 'vdl2'})
|
||||
self.correlator.add_acars_message({"flight": "UAL789", "text": "acars"})
|
||||
self.correlator.add_vdl2_message({"flight": "UAL789", "text": "vdl2"})
|
||||
|
||||
result = self.correlator.get_messages_for_aircraft(callsign='UAL789')
|
||||
assert len(result['acars']) == 1
|
||||
assert len(result['vdl2']) == 1
|
||||
result = self.correlator.get_messages_for_aircraft(callsign="UAL789")
|
||||
assert len(result["acars"]) == 1
|
||||
assert len(result["vdl2"]) == 1
|
||||
|
||||
Reference in New Issue
Block a user