1 files changed, 41 insertions, 0 deletions
diff --git a/src/test/kotlin/moe/nea89/mail/util/ProtocolSpec.kt b/src/test/kotlin/moe/nea89/mail/util/ProtocolSpec.kt
new file mode 100644
index 0000000..9a4bcbf
--- /dev/null
+++ b/src/test/kotlin/moe/nea89/mail/util/ProtocolSpec.kt
@@ -0,0 +1,41 @@
+package moe.nea89.mail.util
+
+import Protocol
+import findSubarray
+import io.kotest.core.spec.style.FreeSpec
+import io.kotest.datatest.withData
+import readUntil
+import java.io.ByteArrayInputStream
+import java.io.ByteArrayOutputStream
+
+
+data class ByteArrayString(val hay: String, val needle: String, val position: Int?, val offset: Int = 0)
+class ProtocolSpec : FreeSpec({
+    "ByteArray.findSubarray" - {
+        "should work correctly" - {
+            withData(
+                listOf(
+                    ByteArrayString("abc", "a", 0),
+                    ByteArrayString("abca", "a", 0),
+                    ByteArrayString("abca", "a", 3, 2),
+                    ByteArrayString("abca", "a", 3, 3),
+                    ByteArrayString("bc", "a", null),
+                    ByteArrayString("acbcab", "ab", 4),
+                    ByteArrayString("abcbcab", "ab", 5, 1),
+                )
+            ) { (hay, needle, position, offset) ->
+                val hay = hay.encodeToByteArray()
+                val needle = needle.encodeToByteArray()
+                assert(hay.findSubarray(needle, offset) == position)
+            }
+        }
+    }
+    "Protocol.IO" - {
+        "readUntil" {
+            val data = ("a".repeat(9) + "01" + "b".repeat(10) + "02").encodeToByteArray()
+            val protIO = Protocol.IO.FromStreams(ByteArrayInputStream(data), ByteArrayOutputStream())
+            assert(protIO.readUntil("01".encodeToByteArray()).decodeToString() == "a".repeat(9))
+            assert(protIO.readUntil("02".encodeToByteArray()).decodeToString() == "b".repeat(10))
+        }
+    }
+})