namespace_test.go

package vector

import (
	"github.com/stretchr/testify/require"
	"testing"
	"time"
)

func TestNamespace(t *testing.T) {

	t.Run("list namespaces", func(t *testing.T) {
		client, err := newTestClient()
		require.NoError(t, err)

		for _, ns := range namespaces {
			createNamespace(t, client, ns)
		}

		ns, err := client.ListNamespaces()
		require.NoError(t, err)
		require.Exactly(t, []string{"", "ns"}, ns)
	})

	t.Run("delete namespaces", func(t *testing.T) {
		client, err := newTestClient()
		require.NoError(t, err)

		for _, ns := range namespaces {
			createNamespace(t, client, ns)
		}

		for _, ns := range namespaces {
			if ns == "" {
				continue
			}

			err := client.Namespace(ns).DeleteNamespace()
			require.NoError(t, err)
		}

		info, err := client.Info()
		require.NoError(t, err)
		require.Exactly(t, 1, len(info.Namespaces))
	})
}

func createNamespace(t *testing.T, client *Index, ns string) {
	if ns == "" {
		return
	}

	namespace := client.Namespace(ns)

	err := namespace.Upsert(Upsert{
		Vector: []float32{0.1, 0.1},
	})
	require.NoError(t, err)

	require.Eventually(t, func() bool {
		info, err := client.Info()
		require.NoError(t, err)
		return info.PendingVectorCount == 0
	}, 10*time.Second, 1*time.Second)

	err = namespace.Reset()
	require.NoError(t, err)
}