1 files changed, 61 insertions, 0 deletions
diff --git a/dedupe_test.go b/dedupe_test.go
new file mode 100644
index 0000000..6388360
--- /dev/null
+++ b/dedupe_test.go
@@ -0,0 +1,61 @@
+package meshtastic_test
+
+import (
+	"crypto/md5"
+	"github.com/crypto-smoke/meshtastic-go"
+	"testing"
+	"time"
+)
+
+func TestPacketDeduplicatorSeen(t *testing.T) {
+	hasher := md5.New()
+	expiresAfter := 100 * time.Millisecond
+	dedup := meshtastic.NewDeduplicator(hasher, expiresAfter)
+
+	sender := uint32(1)
+	packetID := uint32(1)
+
+	// Test Seen with new packetID
+	if dedup.Seen(sender, packetID) {
+		t.Error("Expected the packet to not have been seen")
+	}
+
+	// Test Seen with same packetID again
+	if !dedup.Seen(sender, packetID) {
+		t.Error("Expected the packet to have been seen")
+	}
+
+	// Wait for expiration
+	time.Sleep(expiresAfter + 100*time.Millisecond)
+
+	// Test Seen with same packetID after expiration
+	if dedup.Seen(sender, packetID) {
+		t.Error("Expected the packet to not have been seen after expiration")
+	}
+}
+
+func TestPacketDeduplicatorSeenData(t *testing.T) {
+	hasher := md5.New()
+	expiresAfter := 100 * time.Millisecond
+	dedup := meshtastic.NewDeduplicator(hasher, expiresAfter)
+
+	data := []byte("test data")
+
+	// Test SeenData with new data
+	if dedup.SeenData(data) {
+		t.Error("Expected the data to not have been seen")
+	}
+
+	// Test SeenData with same data again
+	if !dedup.SeenData(data) {
+		t.Error("Expected the data to have been seen")
+	}
+
+	// Wait for expiration
+	time.Sleep(expiresAfter + 100*time.Millisecond)
+
+	// Test SeenData with same data after expiration
+	if dedup.SeenData(data) {
+		t.Error("Expected the data to not have been seen after expiration")
+	}
+}