Function TestMap_Set

immutable_test.go:958–1078

(t *testing.T)

}

func TestMap_Set(t *testing.T) {
	t.Run("Simple", func(t *testing.T) {
		m := NewMap[int, string](nil)
		itr := m.Iterator()
		if !itr.Done() {
			t.Fatal("MapIterator.Done()=true, expected false")
		} else if k, v, ok := itr.Next(); ok {
			t.Fatalf("MapIterator.Next()=<%v,%v>, expected nil", k, v)
		}
	})

	t.Run("Simple", func(t *testing.T) {
		m := NewMap[int, string](nil)
		m = m.Set(100, "foo")
		if v, ok := m.Get(100); !ok || v != "foo" {
			t.Fatalf("unexpected value: <%v,%v>", v, ok)
		}
	})

	t.Run("Multi", func(t *testing.T) {
		m := NewMapOf(nil, map[int]string{1: "foo"})
		itr := m.Iterator()
		if itr.Done() {
			t.Fatal("MapIterator.Done()=false, expected true")
		}
		if k, v, ok := itr.Next(); !ok {
			t.Fatalf("MapIterator.Next()!=ok, expected ok")
		} else if k != 1 || v != "foo" {
			t.Fatalf("MapIterator.Next()=<%v,%v>, expected <1, \"foo\">", k, v)
		}
		if k, v, ok := itr.Next(); ok {
			t.Fatalf("MapIterator.Next()=<%v,%v>, expected nil", k, v)
		}
	})

	t.Run("VerySmall", func(t *testing.T) {
		const n = 6
		m := NewMap[int, int](nil)
		for i := 0; i < n; i++ {
			m = m.Set(i, i+1)
		}
		for i := 0; i < n; i++ {
			if v, ok := m.Get(i); !ok || v != i+1 {
				t.Fatalf("unexpected value for key=%v: <%v,%v>", i, v, ok)
			}
		}

		// NOTE: Array nodes store entries in insertion order.
		itr := m.Iterator()
		for i := 0; i < n; i++ {
			if k, v, ok := itr.Next(); !ok || k != i || v != i+1 {
				t.Fatalf("MapIterator.Next()=<%v,%v>, exp <%v,%v>", k, v, i, i+1)
			}
		}
		if !itr.Done() {
			t.Fatal("expected iterator done")
		}
	})