1 files changed, 85 insertions, 0 deletions
diff --git a/src/test/java/com/code_intelligence/jazzer/autofuzz/TestHelpers.java b/src/test/java/com/code_intelligence/jazzer/autofuzz/TestHelpers.java
new file mode 100644
index 00000000..89f9c968
--- /dev/null
+++ b/src/test/java/com/code_intelligence/jazzer/autofuzz/TestHelpers.java
@@ -0,0 +1,85 @@
+// Copyright 2021 Code Intelligence GmbH
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.code_intelligence.jazzer.autofuzz;
+
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertEquals;
+
+import com.code_intelligence.jazzer.api.CannedFuzzedDataProvider;
+import com.code_intelligence.jazzer.api.FuzzedDataProvider;
+import java.io.ByteArrayInputStream;
+import java.lang.reflect.Constructor;
+import java.lang.reflect.Executable;
+import java.lang.reflect.Method;
+import java.lang.reflect.Type;
+import java.util.List;
+
+class TestHelpers {
+  static void assertGeneralEquals(Object expected, Object actual) {
+    Class<?> type = expected != null ? expected.getClass() : Object.class;
+    if (type.isArray()) {
+      if (type.getComponentType() == boolean.class) {
+        assertArrayEquals((boolean[]) expected, (boolean[]) actual);
+      } else if (type.getComponentType() == char.class) {
+        assertArrayEquals((char[]) expected, (char[]) actual);
+      } else if (type.getComponentType() == short.class) {
+        assertArrayEquals((short[]) expected, (short[]) actual);
+      } else if (type.getComponentType() == long.class) {
+        assertArrayEquals((long[]) expected, (long[]) actual);
+      } else {
+        assertArrayEquals((Object[]) expected, (Object[]) actual);
+      }
+    } else if (type == ByteArrayInputStream.class) {
+      ByteArrayInputStream expectedStream = (ByteArrayInputStream) expected;
+      ByteArrayInputStream actualStream = (ByteArrayInputStream) actual;
+      assertArrayEquals(readAllBytes(expectedStream), readAllBytes(actualStream));
+    } else {
+      assertEquals(expected, actual);
+    }
+  }
+
+  static void consumeTestCase(
+      Object expectedResult, String expectedResultString, List<Object> cannedData) {
+    Class<?> type = expectedResult != null ? expectedResult.getClass() : Object.class;
+    consumeTestCase(type, expectedResult, expectedResultString, cannedData);
+  }
+
+  static void consumeTestCase(
+      Type type, Object expectedResult, String expectedResultString, List<Object> cannedData) {
+    AutofuzzCodegenVisitor visitor = new AutofuzzCodegenVisitor();
+    FuzzedDataProvider data = CannedFuzzedDataProvider.create(cannedData);
+    assertGeneralEquals(expectedResult, new Meta(null).consume(data, type, visitor));
+    assertEquals(expectedResultString, visitor.generate());
+  }
+
+  static void autofuzzTestCase(Object expectedResult, String expectedResultString, Executable func,
+      List<Object> cannedData) {
+    AutofuzzCodegenVisitor visitor = new AutofuzzCodegenVisitor();
+    FuzzedDataProvider data = CannedFuzzedDataProvider.create(cannedData);
+    if (func instanceof Method) {
+      assertGeneralEquals(expectedResult, new Meta(null).autofuzz(data, (Method) func, visitor));
+    } else {
+      assertGeneralEquals(
+          expectedResult, new Meta(null).autofuzz(data, (Constructor<?>) func, visitor));
+    }
+    assertEquals(expectedResultString, visitor.generate());
+  }
+
+  private static byte[] readAllBytes(ByteArrayInputStream in) {
+    byte[] result = new byte[in.available()];
+    in.read(result, 0, in.available());
+    return result;
+  }
+}