clash/component/pool/pool_test.go

package pool

import (
	"context"
	"runtime"
	"testing"
	"time"

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

func lg() Factory {
	initial := -1
	return func(context.Context) (interface{}, error) {
		initial++
		return initial, nil
	}
}

func TestPool_Basic(t *testing.T) {
	g := lg()
	pool := New(g)

	elm, _ := pool.Get()
	assert.Equal(t, 0, elm.(int))
	pool.Put(elm)
	elm, _ = pool.Get()
	assert.Equal(t, 0, elm.(int))
	elm, _ = pool.Get()
	assert.Equal(t, 1, elm.(int))
}

func TestPool_MaxSize(t *testing.T) {
	g := lg()
	size := 5
	pool := New(g, WithSize(size))

	items := []interface{}{}

	for i := 0; i < size; i++ {
		item, _ := pool.Get()
		items = append(items, item)
	}

	extra, _ := pool.Get()
	assert.Equal(t, size, extra.(int))

	for _, item := range items {
		pool.Put(item)
	}

	pool.Put(extra)

	for _, item := range items {
		elm, _ := pool.Get()
		assert.Equal(t, item.(int), elm.(int))
	}
}

func TestPool_MaxAge(t *testing.T) {
	g := lg()
	pool := New(g, WithAge(20))

	elm, _ := pool.Get()
	pool.Put(elm)

	elm, _ = pool.Get()
	assert.Equal(t, 0, elm.(int))
	pool.Put(elm)

	time.Sleep(time.Millisecond * 22)
	elm, _ = pool.Get()
	assert.Equal(t, 1, elm.(int))
}

func TestPool_AutoGC(t *testing.T) {
	g := lg()

	sign := make(chan int)
	pool := New(g, WithEvict(func(item interface{}) {
		sign <- item.(int)
	}))

	elm, _ := pool.Get()
	assert.Equal(t, 0, elm.(int))
	pool.Put(2)
	pool = nil

	runtime.GC()

	select {
	case num := <-sign:
		assert.Equal(t, 2, num)
	case <-time.After(time.Second * 3):
		assert.Fail(t, "something wrong")
	}
}