YARA-Java
diff --git a/‎yara-java/src/test/java/com/virustotal/yara/YaraJavaTests.java
Lines changed: 47 additions & 5 deletions b/‎yara-java/src/test/java/com/virustotal/yara/YaraJavaTests.java
Lines changed: 47 additions & 5 deletions
diff --git a/‎yara-java/src/test/resources/yara/doc/rules/AlternativesExample1.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/AlternativesExample1.yara b/‎yara-java/src/test/resources/yara/doc/rules/AlternativesExample1.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/AlternativesExample1.yara
diff --git a/‎yara-java/src/test/resources/yara/doc/rules/AlternativesExample2.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/AlternativesExample2.yara b/‎yara-java/src/test/resources/yara/doc/rules/AlternativesExample2.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/AlternativesExample2.yara
diff --git a/‎yara-java/src/test/resources/yara/doc/rules/CaseInsensitiveTextExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/CaseInsensitiveTextExample.yara b/‎yara-java/src/test/resources/yara/doc/rules/CaseInsensitiveTextExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/CaseInsensitiveTextExample.yara
diff --git a/‎yara-java/src/test/resources/yara/doc/rules/CommentExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/CommentExample.yara b/‎yara-java/src/test/resources/yara/doc/rules/CommentExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/CommentExample.yara
diff --git a/‎yara-java/src/test/resources/yara/doc/rules/DummyExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/DummyExample.yara b/‎yara-java/src/test/resources/yara/doc/rules/DummyExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/DummyExample.yara
diff --git a/‎yara-java/src/test/resources/yara/doc/rules/ExampleRule.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/ExampleRule.yara b/‎yara-java/src/test/resources/yara/doc/rules/ExampleRule.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/ExampleRule.yara
diff --git a/‎yara-java/src/test/resources/yara/doc/rules/JumpExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/JumpExample.yara b/‎yara-java/src/test/resources/yara/doc/rules/JumpExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/JumpExample.yara
diff --git a/‎yara-java/src/test/resources/yara/doc/rules/TextExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/TextExample.yara b/‎yara-java/src/test/resources/yara/doc/rules/TextExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/TextExample.yara
diff --git a/‎yara-java/src/test/resources/yara/doc/rules/WildcardExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/WildcardExample.yara b/‎yara-java/src/test/resources/yara/doc/rules/WildcardExample.yara renamed to ‎yara-java/src/test/resources/yara/rules/doc/WildcardExample.yara
@@ -5,10 +5,14 @@
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 
+import java.io.File;
+import java.io.IOException;
 import java.lang.foreign.Arena;
 import java.lang.foreign.MemorySegment;
+import java.net.URL;
 import java.nio.ByteBuffer;
 import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
 
 import static com.virustotal.yara.yara_h.C_POINTER;
 import static com.virustotal.yara.yara_h.ERROR_SUCCESS;
@@ -17,14 +21,16 @@
 
 public class YaraJavaTests {
     public static final int NULL_CHAR = 1;
+    public static final int ZERO_ERRORS = 0;
 
     @BeforeAll
-    public static void init(){
-        yr_initialize();
+    public static void init() {
+        int initialized = yr_initialize();
+        Assertions.assertEquals(ERROR_SUCCESS(), initialized);
     }
 
     @Test
-    public void testYaraVersion(){
+    public void testYaraVersion() {
         ByteBuffer versionBuffer = YR_VERSION().asByteBuffer();
         byte[] versionBytes = new byte[versionBuffer.remaining() - NULL_CHAR];
         versionBuffer.get(versionBytes);
@@ -49,8 +55,44 @@ public void testYaraCompilerCreateAndDestroy() {
         }
     }
 
+    @Test
+    public void testYaraCompilerAddString() {
+        try (Arena arena = Arena.openConfined()) {
+            MemorySegment compilerAddress = arena.allocate(C_POINTER); // YR_COMPILER**
+            int created = yr_compiler_create(compilerAddress);
+            Assertions.assertEquals(ERROR_SUCCESS(), created);
+            MemorySegment compiler = compilerAddress.get(C_POINTER, 0); // YR_COMPILER*
+
+            URL path = this.getClass().getResource("/yara/rules/doc");
+            Assertions.assertNotNull(path);
+
+            File yaraDocRulesDir = new File(path.getPath());
+            File[] rules = yaraDocRulesDir.listFiles((dir, name) -> name.endsWith(".yara"));
+
+            Assertions.assertNotNull(rules);
+            Assertions.assertTrue(rules.length > 0);
+
+            MemorySegment namespaceSegment = arena.allocateUtf8String("doc");  // char*
+
+            for (File rule : rules) {
+                try {
+                    String ruleString = Files.readString(rule.toPath());
+                    MemorySegment ruleStringSegment = arena.allocateUtf8String(ruleString); // char*
+
+                    int compilationErrors = yr_compiler_add_string(compiler, ruleStringSegment, namespaceSegment);
+                    Assertions.assertEquals(ZERO_ERRORS, compilationErrors);
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }
+
+            yr_compiler_destroy(compiler);
+        }
+    }
+
     @AfterAll
-    public static void finish(){
-        yr_finalize();
+    public static void finish() {
+        int finalized = yr_finalize();
+        Assertions.assertEquals(ERROR_SUCCESS(), finalized);
     }
 }