feat(scan): add new prompt_injection test type (#159)

closes #158
NeuraLegion · Jul 28, 2023 · cdfd944 · cdfd944
1 parent 5b01036
commit cdfd944
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/SecTester.Scan/Models/TestType.cs b/src/SecTester.Scan/Models/TestType.cs
@@ -41,6 +41,7 @@ public enum TestType
   OpenBuckets,
   OpenDatabase,
   Osi,
+  PromptInjection,
   ProtoPollution,
   RetireJs,
   Rfi,

diff --git a/test/SecTester.Bus.Tests/Dispatchers/MessageSerializerTests.cs b/test/SecTester.Bus.Tests/Dispatchers/MessageSerializerTests.cs
@@ -97,6 +97,7 @@ public class MessageSerializerTests
     new object[] { TestType.OpenBuckets, @"""open_buckets""" },
     new object[] { TestType.OpenDatabase, @"""open_database""" },
     new object[] { TestType.Osi, @"""osi""" },
+    new object[] { TestType.PromptInjection, @"""prompt_injection""" },
     new object[] { TestType.ProtoPollution, @"""proto_pollution""" },
     new object[] { TestType.RetireJs, @"""retire_js""" },
     new object[] { TestType.Rfi, @"""rfi""" },