iterator tests

Author: semihbkgr

e2e_test.go

@@ -208,7 +208,7 @@ func TestGetOrSet(t *testing.T) {
 	}
 }
 
-func TestIterator(t *testing.T) {
+func TestForEach(t *testing.T) {
 	m := New[int, *Animal]()
 
 	m.ForEach(func(i int, a *Animal) bool {

iterator_test.go

@@ -0,0 +1,53 @@
+//go:build go1.23
+// +build go1.23
+
+package haxmap
+
+import (
+	"testing"
+)
+
+func TestIterators(t *testing.T) {
+	type Value = struct {
+		key int
+	}
+
+	m := New[int, *Value]()
+
+	itemCount := 16
+	for i := itemCount; i > 0; i-- {
+		m.Set(i, &Value{i})
+	}
+
+	t.Run("iterator", func(t *testing.T) {
+		counter := 0
+		for k, v := range m.Iterator() {
+			if v == nil {
+				t.Error("Expecting an object.")
+			} else if k != v.key {
+				t.Error("Incorrect key/value pairs")
+			}
+
+			counter++
+		}
+
+		if counter != itemCount {
+			t.Error("Iterated item count did not match.")
+		}
+	})
+
+	t.Run("keys", func(t *testing.T) {
+		counter := 0
+		for k := range m.Keys() {
+			_, ok := m.Get(k)
+			if !ok {
+				t.Error("The key is not is the map")
+			}
+			counter++
+		}
+
+		if counter != itemCount {
+			t.Error("Iterated item count did not match.")
+		}
+	})
+}