Adding protect behavior to LLM tests

src/test/java/com/owlmaddie/tests/BehaviorTests.java

@@ -49,6 +49,10 @@ public class BehaviorTests {
             "<attacked you directly with Stone Axe>",
             "<attacked you indirectly with Arrow>",
             "DIEEE!");
+    List<String> protectMessages = Arrays.asList(
+            "Please protect me",
+            "Please keep me safe friend",
+            "Don't let them hurt me please");
     List<String> friendshipUpMessages = Arrays.asList(
             "Hi friend! I am so happy to see you again!",
             "Looking forward to hanging out with you.",
@@ -109,6 +113,20 @@ public class BehaviorTests {
     }
 
     @Test
+    public void protectBrave() {
+        for (String message : protectMessages) {
+            testPromptForBehavior(bravePath, List.of(message), "PROTECT");
+        }
+    }
+
+    @Test
+    public void protectNervous() {
+        for (String message : protectMessages) {
+            testPromptForBehavior(nervousPath, List.of(message), "PROTECT");
+        }
+    }
+
+    @Test
     public void attackBrave() {
         for (String message : attackMessages) {
             testPromptForBehavior(bravePath, List.of(message), "ATTACK");