linkedhashmap_test.go

package set

import (
	"math/rand"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestGet(t *testing.T) {
	t.Run("When the key exists", func(t *testing.T) {
		set := newLinkedHashMap()
		value := rand.Int()
		set.Put("test", value)
		require.Equal(t, value, set.Get("test"))
	})

	t.Run("When the key not exists", func(t *testing.T) {
		set := newLinkedHashMap()
		result := set.Get("bla")
		require.Nil(t, result)
	})
}

func TestPut(t *testing.T) {
	t.Run("valid key", func(t *testing.T) {
		set := newLinkedHashMap()
		value := rand.Int()
		set.Put("test", value)
		require.Equal(t, value, set.Get("test"))
	})

	t.Run("invalid key", func(t *testing.T) {
		set := newLinkedHashMap()

		value := rand.Int()
		set.Put(nil, value)

		require.Equal(t, 0, set.Length())
	})
}

func TestRemove(t *testing.T) {
	testNumbers := []int{1, 2, 3}
	t.Run("first value", func(t *testing.T) {
		set := newLinkedHashMap()
		for _, number := range testNumbers {
			set.Put(number, number)
		}

		set.Remove(1)
		require.Equal(t, 2, set.Length())
		require.Nil(t, set.Get(1))
	})

	t.Run("last value", func(t *testing.T) {
		set := newLinkedHashMap()
		for _, number := range testNumbers {
			set.Put(number, number)
		}

		set.Remove(3)
		require.Equal(t, 2, set.Length())
		require.Nil(t, set.Get(3))
	})

	t.Run("middle value", func(t *testing.T) {
		set := newLinkedHashMap()
		for _, number := range testNumbers {
			set.Put(number, number)
		}

		set.Remove(2)
		require.Equal(t, 2, set.Length())
		require.Nil(t, set.Get(2))
	})

	t.Run("single value", func(t *testing.T) {
		set := newLinkedHashMap()
		set.Put(1, 1)
		set.Remove(1)
		require.Equal(t, 0, set.Length())
		require.Nil(t, set.Get(1))
	})
}