Function TestClusterStore

store/store_test.go:32–107

(t *testing.T)

)

func TestClusterStore(t *testing.T) {
	ctx := context.Background()
	store := NewClusterStore(engine.NewMock())

	t.Run("create/get/list/delete namespace", func(t *testing.T) {
		namespaces := []string{"ns0", "ns1", "ns2"}
		for _, ns := range namespaces {
			err := store.CreateNamespace(ctx, ns)
			require.NoError(t, err)
			require.ErrorIs(t, store.CreateNamespace(ctx, ns), consts.ErrAlreadyExists)
			exists, err := store.ExistsNamespace(ctx, ns)
			require.NoError(t, err)
			require.True(t, exists)
		}

		exists, err := store.ExistsNamespace(ctx, "not-exits-ns")
		require.NoError(t, err)
		require.False(t, exists)

		gotNamespaces, err := store.ListNamespace(ctx)
		require.NoError(t, err)
		require.ElementsMatch(t, namespaces, gotNamespaces)

		for _, ns := range namespaces {
			err := store.RemoveNamespace(ctx, ns)
			require.NoError(t, err)
			exists, err := store.ExistsNamespace(ctx, ns)
			require.NoError(t, err)
			require.False(t, exists)
		}
	})

	t.Run("create/get/list/delete cluster", func(t *testing.T) {
		ns := "ns0"
		cluster0 := &Cluster{Name: "cluster0", Shards: Shards{NewShard()}}
		cluster1 := &Cluster{Name: "cluster1", Shards: Shards{NewShard()}}
		cluster0.Version.Store(2)
		cluster1.Version.Store(3)

		require.NoError(t, store.CreateCluster(ctx, ns, cluster0))
		require.ErrorIs(t, store.CreateCluster(ctx, ns, cluster0), consts.ErrAlreadyExists)
		require.NoError(t, store.CreateCluster(ctx, ns, cluster1))

		gotCluster, err := store.GetCluster(ctx, ns, "cluster0")
		require.NoError(t, err)
		require.Equal(t, cluster0.Name, gotCluster.Name)
		require.Equal(t, cluster0.Version.Load(), gotCluster.Version.Load())

		gotClusters, err := store.ListCluster(ctx, ns)
		require.NoError(t, err)
		require.ElementsMatch(t, []string{"cluster0", "cluster1"}, gotClusters)

		require.NoError(t, store.UpdateCluster(ctx, ns, cluster0))
		gotCluster, err = store.GetCluster(ctx, ns, "cluster0")
		require.NoError(t, err)
		require.Equal(t, cluster0.Name, gotCluster.Name)
		require.EqualValues(t, 3, gotCluster.Version.Load())