Fixed Python flake8 offenses
diff --git a/lib/py/test/fuzz/fuzz_common.py b/lib/py/test/fuzz/fuzz_common.py
index 33c8138..3fcbca9 100644
--- a/lib/py/test/fuzz/fuzz_common.py
+++ b/lib/py/test/fuzz/fuzz_common.py
@@ -22,6 +22,7 @@
import os
import atheris
+
def setup_thrift_imports():
"""Set up the Python path to include Thrift libraries and generated code."""
@@ -48,19 +49,21 @@
sys.path.append(gen_path)
print(sys.path)
+
setup_thrift_imports()
from thrift.transport import TTransport
from thrift.TSerialization import serialize, deserialize
from fuzz.ttypes import FuzzTest
+
def create_parser_fuzzer(protocol_factory_class):
"""
Create a parser fuzzer function for a specific protocol.
-
+
Args:
protocol_factory_class: The Thrift protocol factory class to use
-
+
Returns:
A function that can be used with atheris.Setup()
"""
@@ -71,25 +74,26 @@
try:
# Create a memory buffer with the fuzzed data
buf = TTransport.TMemoryBuffer(data)
- transport = TTransport.TBufferedTransportFactory().getTransport(buf)
+ TTransport.TBufferedTransportFactory().getTransport(buf)
factory = protocol_factory_class(string_length_limit=1000, container_length_limit=1000)
# Try to deserialize the fuzzed data into the test class
- test_instance = deserialize(FuzzTest(), data, factory)
+ deserialize(FuzzTest(), data, factory)
- except Exception as e:
+ except Exception:
# We expect various exceptions during fuzzing
pass
return TestOneInput
+
def create_roundtrip_fuzzer(protocol_factory_class):
"""
Create a roundtrip fuzzer function for a specific protocol.
-
+
Args:
protocol_factory_class: The Thrift protocol factory class to use
-
+
Returns:
A function that can be used with atheris.Setup()
"""
@@ -100,7 +104,7 @@
try:
# Create a memory buffer with the fuzzed data
buf = TTransport.TMemoryBuffer(data)
- transport = TTransport.TBufferedTransportFactory().getTransport(buf)
+ TTransport.TBufferedTransportFactory().getTransport(buf)
factory = protocol_factory_class(string_length_limit=1000, container_length_limit=1000)
# Try to deserialize the fuzzed data into the test class
@@ -112,18 +116,19 @@
# Verify the objects are equal after a second deserialization
assert test_instance == deserialized
- except AssertionError as e:
- raise e
- except Exception as e:
+ except AssertionError:
+ raise
+ except Exception:
# We expect various exceptions during fuzzing
pass
return TestOneInput
+
def _run_fuzzer(fuzzer_function):
"""
Set up and run the fuzzer for a specific protocol.
-
+
Args:
fuzzer_function: The fuzzer function to use
"""
@@ -136,7 +141,7 @@
def run_roundtrip_fuzzer(protocol_factory_class):
"""
Set up and run the fuzzer for a specific protocol.
-
+
Args:
protocol_factory_class: The Thrift protocol factory class to use
"""
@@ -146,8 +151,8 @@
def run_parser_fuzzer(protocol_factory_class):
"""
Set up and run the fuzzer for a specific protocol.
-
+
Args:
protocol_factory_class: The Thrift protocol factory class to use
"""
- _run_fuzzer(create_parser_fuzzer(protocol_factory_class))
\ No newline at end of file
+ _run_fuzzer(create_parser_fuzzer(protocol_factory_class))
diff --git a/lib/py/test/fuzz/fuzz_parse_TBinaryProtocol.py b/lib/py/test/fuzz/fuzz_parse_TBinaryProtocol.py
index a07de28..ecdf056 100644
--- a/lib/py/test/fuzz/fuzz_parse_TBinaryProtocol.py
+++ b/lib/py/test/fuzz/fuzz_parse_TBinaryProtocol.py
@@ -20,8 +20,10 @@
from fuzz_common import run_parser_fuzzer
from thrift.protocol.TBinaryProtocol import TBinaryProtocolFactory
+
def main():
run_parser_fuzzer(TBinaryProtocolFactory)
+
if __name__ == "__main__":
main()
diff --git a/lib/py/test/fuzz/fuzz_parse_TBinaryProtocolAccelerated.py b/lib/py/test/fuzz/fuzz_parse_TBinaryProtocolAccelerated.py
index 7c4f046..61c5e92 100644
--- a/lib/py/test/fuzz/fuzz_parse_TBinaryProtocolAccelerated.py
+++ b/lib/py/test/fuzz/fuzz_parse_TBinaryProtocolAccelerated.py
@@ -20,8 +20,10 @@
from fuzz_common import run_parser_fuzzer
from thrift.protocol.TBinaryProtocol import TBinaryProtocolAcceleratedFactory
+
def main():
run_parser_fuzzer(TBinaryProtocolAcceleratedFactory)
+
if __name__ == "__main__":
- main()
\ No newline at end of file
+ main()
diff --git a/lib/py/test/fuzz/fuzz_parse_TCompactProtocol.py b/lib/py/test/fuzz/fuzz_parse_TCompactProtocol.py
index ff05ec1..a8ebc51 100644
--- a/lib/py/test/fuzz/fuzz_parse_TCompactProtocol.py
+++ b/lib/py/test/fuzz/fuzz_parse_TCompactProtocol.py
@@ -20,8 +20,10 @@
from fuzz_common import run_parser_fuzzer
from thrift.protocol.TCompactProtocol import TCompactProtocolFactory
+
def main():
run_parser_fuzzer(TCompactProtocolFactory)
+
if __name__ == "__main__":
- main()
\ No newline at end of file
+ main()
diff --git a/lib/py/test/fuzz/fuzz_parse_TCompactProtocolAccelerated.py b/lib/py/test/fuzz/fuzz_parse_TCompactProtocolAccelerated.py
index 04ab96f..a7b8635 100644
--- a/lib/py/test/fuzz/fuzz_parse_TCompactProtocolAccelerated.py
+++ b/lib/py/test/fuzz/fuzz_parse_TCompactProtocolAccelerated.py
@@ -20,8 +20,10 @@
from fuzz_common import run_parser_fuzzer
from thrift.protocol.TCompactProtocol import TCompactProtocolAcceleratedFactory
+
def main():
run_parser_fuzzer(TCompactProtocolAcceleratedFactory)
+
if __name__ == "__main__":
- main()
\ No newline at end of file
+ main()
diff --git a/lib/py/test/fuzz/fuzz_roundtrip_TBinaryProtocol.py b/lib/py/test/fuzz/fuzz_roundtrip_TBinaryProtocol.py
index 7413707..a2cdafd 100644
--- a/lib/py/test/fuzz/fuzz_roundtrip_TBinaryProtocol.py
+++ b/lib/py/test/fuzz/fuzz_roundtrip_TBinaryProtocol.py
@@ -20,8 +20,10 @@
from fuzz_common import run_roundtrip_fuzzer
from thrift.protocol.TBinaryProtocol import TBinaryProtocolFactory
+
def main():
run_roundtrip_fuzzer(TBinaryProtocolFactory)
+
if __name__ == "__main__":
main()
diff --git a/lib/py/test/fuzz/fuzz_roundtrip_TBinaryProtocolAccelerated.py b/lib/py/test/fuzz/fuzz_roundtrip_TBinaryProtocolAccelerated.py
index 76f3114..4cd0b74 100644
--- a/lib/py/test/fuzz/fuzz_roundtrip_TBinaryProtocolAccelerated.py
+++ b/lib/py/test/fuzz/fuzz_roundtrip_TBinaryProtocolAccelerated.py
@@ -20,8 +20,10 @@
from fuzz_common import run_roundtrip_fuzzer
from thrift.protocol.TBinaryProtocol import TBinaryProtocolAcceleratedFactory
+
def main():
run_roundtrip_fuzzer(TBinaryProtocolAcceleratedFactory)
+
if __name__ == "__main__":
- main()
\ No newline at end of file
+ main()
diff --git a/lib/py/test/fuzz/fuzz_roundtrip_TCompactProtocol.py b/lib/py/test/fuzz/fuzz_roundtrip_TCompactProtocol.py
index 0fb9cce..a70270e 100644
--- a/lib/py/test/fuzz/fuzz_roundtrip_TCompactProtocol.py
+++ b/lib/py/test/fuzz/fuzz_roundtrip_TCompactProtocol.py
@@ -20,8 +20,10 @@
from fuzz_common import run_roundtrip_fuzzer
from thrift.protocol.TCompactProtocol import TCompactProtocolFactory
+
def main():
run_roundtrip_fuzzer(TCompactProtocolFactory)
+
if __name__ == "__main__":
- main()
\ No newline at end of file
+ main()
diff --git a/lib/py/test/fuzz/fuzz_roundtrip_TCompactProtocolAccelerated.py b/lib/py/test/fuzz/fuzz_roundtrip_TCompactProtocolAccelerated.py
index 0465b2b..d0f4c57 100644
--- a/lib/py/test/fuzz/fuzz_roundtrip_TCompactProtocolAccelerated.py
+++ b/lib/py/test/fuzz/fuzz_roundtrip_TCompactProtocolAccelerated.py
@@ -20,8 +20,10 @@
from fuzz_common import run_roundtrip_fuzzer
from thrift.protocol.TCompactProtocol import TCompactProtocolAcceleratedFactory
+
def main():
run_roundtrip_fuzzer(TCompactProtocolAcceleratedFactory)
+
if __name__ == "__main__":
- main()
\ No newline at end of file
+ main()
diff --git a/lib/py/test/test_socket.py b/lib/py/test/test_socket.py
index 5e25f1a..397c9bd 100644
--- a/lib/py/test/test_socket.py
+++ b/lib/py/test/test_socket.py
@@ -32,7 +32,7 @@
class TSocketTest(unittest.TestCase):
def test_failed_connection_raises_exception(self):
- sock = TSocket(host="localhost", port=60606) # unused port
+ sock = TSocket(host="localhost", port=60606) # unused port
with self.assertRaises(TTransportException) as ctx:
sock.open()
exc = ctx.exception
@@ -55,7 +55,7 @@
exc = ctx.exception
self.assertEqual(exc.message, "read timeout")
- acc.client.close() # this also blocks until the other thread is done
+ acc.client.close() # this also blocks until the other thread is done
acc.close()
sock.close()
diff --git a/lib/py/test/thrift_TBinaryProtocol.py b/lib/py/test/thrift_TBinaryProtocol.py
index b257626..c4777ad 100644
--- a/lib/py/test/thrift_TBinaryProtocol.py
+++ b/lib/py/test/thrift_TBinaryProtocol.py
@@ -266,10 +266,10 @@
def test_TBinaryProtocol_no_strict_write_read(self):
TMessageType = {"T_CALL": 1, "T_REPLY": 2, "T_EXCEPTION": 3, "T_ONEWAY": 4}
test_data = [("short message name", TMessageType['T_CALL'], 0),
- ("1", TMessageType['T_REPLY'], 12345),
- ("loooooooooooooooooooooooooooooooooong", TMessageType['T_EXCEPTION'], 1 << 16),
- ("one way push", TMessageType['T_ONEWAY'], 12),
- ("Janky", TMessageType['T_CALL'], 0)]
+ ("1", TMessageType['T_REPLY'], 12345),
+ ("loooooooooooooooooooooooooooooooooong", TMessageType['T_EXCEPTION'], 1 << 16),
+ ("one way push", TMessageType['T_ONEWAY'], 12),
+ ("Janky", TMessageType['T_CALL'], 0)]
try:
for dt in test_data:
diff --git a/lib/py/test/thrift_TSerializer.py b/lib/py/test/thrift_TSerializer.py
index 600dee7..396e68d 100644
--- a/lib/py/test/thrift_TSerializer.py
+++ b/lib/py/test/thrift_TSerializer.py
@@ -31,7 +31,6 @@
from thrift.protocol.TBinaryProtocol import TBinaryProtocolAcceleratedFactory
from thrift.protocol.TCompactProtocol import TCompactProtocolFactory
from thrift.protocol.TCompactProtocol import TCompactProtocolAcceleratedFactory
-from thrift.transport import TTransport
from thrift.TSerialization import serialize, deserialize
from TestServer.ttypes import Message