Add fuzzing test (#13)

Author: Songki Choi

tests/fuzz/__init__.py

@@ -0,0 +1,36 @@
+# Copyright (C) 2024 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+import sys
+
+import atheris
+
+with atheris.instrument_imports():
+    from openvino_xai import Explainer, insert_xai
+
+
+def TestOneInput(input_bytes):
+    data_provider = atheris.FuzzedDataProvider(input_bytes)
+
+    # Test Explainer statefull class API
+    try:
+        explainer = Explainer(
+            model=data_provider.ConsumeBytes(8),
+            task=data_provider.ConsumeUInt(8),
+        )
+        explanation = explainer(data=data_provider.ConsumeBytes(8))
+    except ValueError as e:
+        pass
+
+    # Test insert_xai() stateless API
+    try:
+        insert_xai(
+            model=data_provider.ConsumeBytes(8),
+            task=data_provider.ConsumeUInt(8),
+        )
+    except ValueError as e:
+        pass
+
+
+atheris.Setup(sys.argv, TestOneInput)
+atheris.Fuzz()

tests/fuzz/test_fuzz.py

@@ -25,3 +25,10 @@ commands =
 extras = dev,val
 commands =
     pytest -ra --showlocals {posargs:tests/}
+
+[testenv:fuzz-{py310, py311}]
+deps =
+    atheris
+extras = dev
+commands =
+    python tests/fuzz/test_fuzz.py {posargs:-artifact_prefix={toxworkdir}/ -print_final_stats=1 -runs=10000000}