cli/tests/e2e/testcases/node/devnet/suite.go

// Copyright (C) 2025, Lux Industries Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package root

import (
	"encoding/json"
	"fmt"
	"os"
	"os/user"
	"path/filepath"
	"regexp"
	"time"

	"github.com/luxfi/cli/v2/pkg/constants"
	"github.com/luxfi/cli/v2/pkg/models"
	"github.com/luxfi/cli/v2/tests/e2e/commands"
	ginkgo "github.com/onsi/ginkgo/v2"
	"github.com/onsi/gomega"
	"github.com/pborman/ansi"
)

var (
	hostName    string
	NodeID      string
	apiHostName string
	apiNodeID   string
)

const (
	NumNodes           = 1
	NumAPINodes        = 1
	luxdVersion = "v1.13.0"
)

var _ = ginkgo.Describe("[Node devnet]", func() {
	ginkgo.It("can create a node", func() {
		outputB, err := ansi.Strip([]byte(commands.NodeDevnet(luxdVersion, NumNodes, NumAPINodes)))
		gomega.Expect(err).Should(gomega.BeNil())
		output := string(outputB)
		fmt.Println(output)
		gomega.Expect(output).To(gomega.ContainSubstring("Luxd and Lux-CLI installed and node(s) are bootstrapping!"))
		// parse hostName
		// Parse validator node
		re := regexp.MustCompile(`Cloud Instance ID: (\S+) \| Public IP: (\S+) \| NodeID-(\S+)`)
		match := re.FindStringSubmatch(output)
		if len(match) >= 3 {
			hostName = match[1]
			NodeID = fmt.Sprintf("NodeID-%s", match[3])
			fmt.Println(hostName)
			fmt.Println(NodeID)
			// This is a validator node
		} else {
			ginkgo.Fail("failed to parse validator hostName and NodeID")
		}

		// Parse API node
		apiRe := regexp.MustCompile(`\[API\] Cloud Instance ID: (\S+) \| Public IP: (\S+) \| NodeID-(\S+)`)
		apiMatch := apiRe.FindStringSubmatch(output)
		if len(apiMatch) >= 3 {
			apiHostName = apiMatch[1]
			apiNodeID = fmt.Sprintf("NodeID-%s", apiMatch[3])
			fmt.Println(apiHostName)
			fmt.Println(apiNodeID)
			// This is an API node
		} else {
			ginkgo.Fail("[API] failed to parse hostName and NodeID")
		}
	})
	ginkgo.It("has correct cluster config record for API", func() {
		usr, err := user.Current()
		gomega.Expect(err).Should(gomega.BeNil())
		homeDir := usr.HomeDir
		relativePath := "nodes"
		content, err := os.ReadFile(filepath.Join(homeDir, constants.E2EBaseDirName, relativePath, constants.ClustersConfigFileName))
		gomega.Expect(err).Should(gomega.BeNil())
		clustersConfig := models.ClustersConfig{}
		err = json.Unmarshal(content, &clustersConfig)
		gomega.Expect(err).Should(gomega.BeNil())
		gomega.Expect(clustersConfig.Clusters[constants.E2EClusterName].APINodes).To(gomega.HaveLen(NumAPINodes))
	})
	ginkgo.It("installs and runs luxd", func() {
		luxdProcess := commands.NodeSSH(constants.E2EClusterName, "docker ps --no-trunc")
		gomega.Expect(luxdProcess).To(gomega.ContainSubstring("avaplatform/luxd:"))
	})
	ginkgo.It("configured luxd", func() {
		luxdConfig := commands.NodeSSH(constants.E2EClusterName, "cat /home/ubuntu/.luxd/configs/node.json")
		gomega.Expect(luxdConfig).To(gomega.ContainSubstring("\"genesis-file\": \"/.luxd/configs/genesis.json\""))
		gomega.Expect(luxdConfig).To(gomega.ContainSubstring("\"network-id\": \"network-1338\""))
		gomega.Expect(luxdConfig).To(gomega.ContainSubstring("\"public-ip\": \"" + constants.E2ENetworkPrefix))
		luxdConfigCChain := commands.NodeSSH(constants.E2EClusterName, "cat /home/ubuntu/.luxd/configs/chains/C/config.json")
		gomega.Expect(luxdConfigCChain).To(gomega.ContainSubstring("\"state-sync-enabled\": true"))
	})
	ginkgo.It("provides luxd with staking certs", func() {
		stakingFiles := commands.NodeSSH(constants.E2EClusterName, "ls /home/ubuntu/.luxd/staking/")
		gomega.Expect(stakingFiles).To(gomega.ContainSubstring("signer.key"))
		gomega.Expect(stakingFiles).To(gomega.ContainSubstring("staker.crt"))
		gomega.Expect(stakingFiles).To(gomega.ContainSubstring("staker.key"))
	})
	ginkgo.It("provides luxd with genesis", func() {
		genesisFile := commands.NodeSSH(constants.E2EClusterName, "cat /home/ubuntu/.luxd/configs/genesis.json")
		gomega.Expect(genesisFile).To(gomega.ContainSubstring("luxAddr"))
		gomega.Expect(genesisFile).To(gomega.ContainSubstring("initialStakers"))
		gomega.Expect(genesisFile).To(gomega.ContainSubstring("cChainGenesis"))
		gomega.Expect(genesisFile).To(gomega.ContainSubstring(NodeID))
		gomega.Expect(genesisFile).To(gomega.ContainSubstring("\"rewardAddress\": \"X-custom"))
		gomega.Expect(genesisFile).To(gomega.ContainSubstring("\"startTime\":"))
		gomega.Expect(genesisFile).To(gomega.ContainSubstring("\"networkID\": 1338,"))
		// make sure there is no API node in the genesis
		gomega.Expect(genesisFile).To(gomega.Not(gomega.ContainSubstring(apiNodeID)))
	})
	ginkgo.It("can wait for 20 seconds for luxd to startup", func() {
		time.Sleep(20 * time.Second)
	})
	/*ginkgo.It("can get cluster status", func() {
		output := commands.NodeStatus()
		fmt.Println(output)
		gomega.Expect(output).To(gomega.ContainSubstring("Checking if node(s) are bootstrapped to Primary Network"))
		gomega.Expect(output).To(gomega.ContainSubstring("Checking if node(s) are healthy"))
		gomega.Expect(output).To(gomega.ContainSubstring("Getting luxd version of node(s)"))
		gomega.Expect(output).To(gomega.ContainSubstring(constants.E2ENetworkPrefix))
		gomega.Expect(output).To(gomega.ContainSubstring(hostName))
		gomega.Expect(output).To(gomega.ContainSubstring(NodeID))
		gomega.Expect(output).To(gomega.ContainSubstring(apiHostName))
		gomega.Expect(output).To(gomega.ContainSubstring(apiNodeID))
		gomega.Expect(output).To(gomega.ContainSubstring("Devnet"))
	})*/
	ginkgo.It("can ssh to a created node", func() {
		output := commands.NodeSSH(constants.E2EClusterName, "echo hello")
		gomega.Expect(output).To(gomega.ContainSubstring("hello"))
	})
	ginkgo.It("can list created nodes", func() {
		output := commands.NodeList()
		fmt.Println(output)
		gomega.Expect(output).To(gomega.ContainSubstring("Devnet"))
		gomega.Expect(output).To(gomega.ContainSubstring("docker1"))
		gomega.Expect(output).To(gomega.ContainSubstring("NodeID"))
		gomega.Expect(output).To(gomega.ContainSubstring(constants.E2ENetworkPrefix))
	})
	ginkgo.It("can cleanup", func() {
		commands.DeleteE2EInventory()
		commands.DeleteE2ECluster()
		commands.DeleteNode(hostName)
	})
})