From af12d067321a350f5526979123a2788cfa6730a9 Mon Sep 17 00:00:00 2001
From: Jason Ernst <ernstjason1@gmail.com>
Date: Mon, 9 Sep 2024 07:46:31 -0700
Subject: [PATCH] added an icmp test

---
 .../com/jasonernst/knet/icmp/ICMPTest.kt      | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 knet/src/test/kotlin/com/jasonernst/knet/icmp/ICMPTest.kt

diff --git a/knet/src/test/kotlin/com/jasonernst/knet/icmp/ICMPTest.kt b/knet/src/test/kotlin/com/jasonernst/knet/icmp/ICMPTest.kt
new file mode 100644
index 0000000..57cb900
--- /dev/null
+++ b/knet/src/test/kotlin/com/jasonernst/knet/icmp/ICMPTest.kt
@@ -0,0 +1,25 @@
+package com.jasonernst.knet.icmp
+
+import com.jasonernst.icmp_common.ICMPHeader
+import com.jasonernst.icmp_common.v4.ICMPv4EchoPacket
+import com.jasonernst.icmp_common.v6.ICMPv6EchoPacket
+import org.junit.jupiter.api.Assertions.assertEquals
+import org.junit.jupiter.api.Test
+import java.nio.ByteBuffer
+
+class ICMPTest {
+    @Test
+    fun icmpv4() {
+        val icmp = ICMPv4EchoPacket(id = 1u, sequence = 2u, data = "hello".toByteArray())
+        val stream = ByteBuffer.wrap(icmp.toByteArray())
+        val parsed = ICMPHeader.fromStream(stream)
+        assertEquals(icmp, parsed)
+    }
+
+    @Test fun icmpv6() {
+        val icmp = ICMPv6EchoPacket(id = 1u, sequence = 2u, data = "hello".toByteArray())
+        val stream = ByteBuffer.wrap(icmp.toByteArray())
+        val parsed = ICMPHeader.fromStream(stream, isIcmpV4 = false)
+        assertEquals(icmp, parsed)
+    }
+}