//                           _       _
// __      _____  __ ___   ___  __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
//  \ V  V /  __/ (_| |\ V /| | (_| | ||  __/
//   \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
//  Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
//  CONTACT: hello@weaviate.io
//

package test

import (
	"fmt"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/weaviate/weaviate/client/nodes"
	"github.com/weaviate/weaviate/entities/models"
	"github.com/weaviate/weaviate/entities/verbosity"
	"github.com/weaviate/weaviate/test/helper"
	"github.com/weaviate/weaviate/test/helper/sample-schema/books"
	"github.com/weaviate/weaviate/test/helper/sample-schema/multishard"
)

func Test_WeaviateCluster_NodesAPI(t *testing.T) {
	helper.SetupClient(os.Getenv(weaviateNode1Endpoint))
	booksClass := books.ClassContextionaryVectorizer()
	multiShardClass := multishard.ClassContextionaryVectorizer()
	// we override multishard.ShardingConfig to make sure it matches nodes count
	// and exists in every node
	multiShardClass.ShardingConfig = map[string]interface{}{
		"actualCount":         float64(3),
		"actualVirtualCount":  float64(128),
		"desiredCount":        float64(3),
		"desiredVirtualCount": float64(128),
		"function":            "murmur3",
		"key":                 "_id",
		"strategy":            "hash",
		"virtualPerPhysical":  float64(128),
	}

	helper.CreateClass(t, booksClass)
	helper.CreateClass(t, multiShardClass)
	defer helper.DeleteClass(t, booksClass.Class)
	defer helper.DeleteClass(t, multiShardClass.Class)

	t.Run("import data", func(t *testing.T) {
		for _, book := range books.Objects() {
			helper.CreateObject(t, book)
			helper.AssertGetObjectEventually(t, book.Class, book.ID)
		}
		for _, multishard := range multishard.Objects() {
			helper.CreateObject(t, multishard)
			helper.AssertGetObjectEventually(t, multishard.Class, multishard.ID)
		}
	})

	t.Run("check nodes api", func(t *testing.T) {
		for _, endpoint := range []string{weaviateNode1Endpoint, weaviateNode2Endpoint, weaviateNode3Endpoint} {
			t.Run(endpoint, func(t *testing.T) {
				helper.SetupClient(os.Getenv(endpoint))
				verbose := verbosity.OutputVerbose
				params := nodes.NewNodesGetParams().WithOutput(&verbose)
				resp, err := helper.Client(t).Nodes.NodesGet(params, nil)
				require.Nil(t, err)

				nodeStatusResp := resp.GetPayload()
				require.NotNil(t, nodeStatusResp)

				nodes := nodeStatusResp.Nodes
				require.NotNil(t, nodes)
				require.Len(t, nodes, 3)

				assert.Equal(t, "node1", nodes[0].Name)
				assert.Equal(t, "node2", nodes[1].Name)
				assert.Equal(t, "node3", nodes[2].Name)

				for i, nodeStatus := range nodes {
					require.NotNil(t, nodeStatus)
					assert.Equal(t, models.NodeStatusStatusHEALTHY, *nodeStatus.Status)
					assert.Equal(t, fmt.Sprintf("node%d", i+1), nodeStatus.Name)
					assert.True(t, nodeStatus.GitHash != "" && nodeStatus.GitHash != "unknown")
					assert.Len(t, nodeStatus.Shards, 2)
					var objectCount int64
					var shardCount int64
					for _, shard := range nodeStatus.Shards {
						assert.True(t, len(shard.Name) > 0)
						assert.True(t, shard.Class == multiShardClass.Class || shard.Class == booksClass.Class)
						assert.GreaterOrEqual(t, shard.ObjectCount, int64(0))
						assert.GreaterOrEqual(t, shard.VectorQueueLength, int64(0))
						objectCount += shard.ObjectCount
						shardCount++
					}
					require.NotNil(t, nodeStatus.Stats)
					assert.Equal(t, objectCount, nodeStatus.Stats.ObjectCount)
					assert.Equal(t, shardCount, nodeStatus.Stats.ShardCount)
				}
			})
		}
	})
}
