adapters/repos/db/vector/hnsw/visited/pool_test.go

//                           _       _
// __      _____  __ ___   ___  __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
//  \ V  V /  __/ (_| |\ V /| | (_| | ||  __/
//   \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
//  Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
//  CONTACT: [email protected]
//

package visited

import (
	"testing"

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

func TestPool(t *testing.T) {
	// pool with two liss
	pool := NewPool(2, 2)
	assert.Equal(t, 2, len(pool.listSets))

	// get first list
	l1 := pool.Borrow()
	assert.Equal(t, 2, l1.Len())
	assert.Equal(t, 1, len(pool.listSets))

	// get second list
	l2 := pool.Borrow()
	assert.Equal(t, 0, len(pool.listSets))

	// get third list from empty pool and return it back
	l3 := pool.Borrow()
	assert.Equal(t, 0, len(pool.listSets))
	pool.Return(l3)
	assert.Equal(t, 1, len(pool.listSets))

	// get same list again and modify its size
	// so that it is not accepted when returned to the pool
	l3 = pool.Borrow()
	l3.Visit(2)
	pool.Return(l3)
	assert.Equal(t, 0, len(pool.listSets))

	// add two list and destroy the pool
	pool.Return(l1)
	pool.Return(l2)
	assert.Equal(t, 2, len(pool.listSets))
	pool.Destroy()
	assert.Equal(t, 0, len(pool.listSets))
}