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()