cli/pkg/application/app_test.go

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

import (
	"os"
	"path/filepath"
	"testing"

	"github.com/luxfi/cli/v2/pkg/constants"
	"github.com/luxfi/cli/v2/pkg/models"
	"github.com/luxfi/ids"
	luxlog "github.com/luxfi/log"
	"github.com/stretchr/testify/require"
)

const (
	subnetName1 = "TEST_subnet"
	subnetName2 = "TEST_copied_subnet"
)

func TestUpdateSideCar(t *testing.T) {
	require := require.New(t)
	sc := &models.Sidecar{
		Name:      "TEST",
		VM:        models.EVM,
		TokenName: "TEST",
		ChainID:   "42",
	}

	ap := newTestApp(t)

	err := ap.CreateSidecar(sc)
	require.NoError(err)
	control, err := ap.LoadSidecar(sc.Name)
	require.NoError(err)
	require.Equal(*sc, control)
	sc.Networks = make(map[string]models.NetworkData)
	sc.Networks["local"] = models.NetworkData{
		BlockchainID: ids.GenerateTestID(),
		SubnetID:     ids.GenerateTestID(),
	}

	err = ap.UpdateSidecar(sc)
	require.NoError(err)
	control, err = ap.LoadSidecar(sc.Name)
	require.NoError(err)
	require.Equal(*sc, control)
}

func Test_writeGenesisFile_success(t *testing.T) {
	require := require.New(t)
	genesisBytes := []byte("genesis")
	genesisFile := constants.GenesisFileName

	ap := newTestApp(t)
	// Write genesis
	err := ap.WriteGenesisFile(subnetName1, genesisBytes)
	require.NoError(err)

	// Check file exists
	createdPath := filepath.Join(ap.GetSubnetDir(), subnetName1, genesisFile)
	_, err = os.Stat(createdPath)
	require.NoError(err)

	// Cleanup file
	err = os.Remove(createdPath)
	require.NoError(err)
}

func Test_copyGenesisFile_success(t *testing.T) {
	require := require.New(t)
	genesisBytes := []byte("genesis")

	ap := newTestApp(t)
	// Create original genesis
	err := ap.WriteGenesisFile(subnetName1, genesisBytes)
	require.NoError(err)

	// Copy genesis
	createdGenesis := ap.GetGenesisPath(subnetName1)
	err = ap.CopyGenesisFile(createdGenesis, subnetName2)
	require.NoError(err)

	// Check copied file exists
	copiedGenesis := ap.GetGenesisPath(subnetName2)
	_, err = os.Stat(copiedGenesis)
	require.NoError(err)

	// Cleanup files
	err = os.Remove(createdGenesis)
	require.NoError(err)
	err = os.Remove(copiedGenesis)
	require.NoError(err)
}

func Test_copyGenesisFile_failure(t *testing.T) {
	require := require.New(t)
	// copy genesis that doesn't exist

	ap := newTestApp(t)
	// Copy genesis
	createdGenesis := ap.GetGenesisPath(subnetName1)
	err := ap.CopyGenesisFile(createdGenesis, subnetName2)
	require.Error(err)

	// Check no copied file exists
	copiedGenesis := ap.GetGenesisPath(subnetName2)
	_, err = os.Stat(copiedGenesis)
	require.Error(err)
}

func Test_createSidecar_success(t *testing.T) {
	type test struct {
		name              string
		subnetName        string
		tokenName         string
		expectedTokenName string
		chainID           string
	}

	tests := []test{
		{
			name:              "Success",
			subnetName:        subnetName1,
			tokenName:         "TOKEN",
			expectedTokenName: "TOKEN",
			chainID:           "999",
		},
		{
			name:              "no token name",
			subnetName:        subnetName1,
			tokenName:         "",
			expectedTokenName: "TEST",
			chainID:           "888",
		},
	}

	ap := newTestApp(t)

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			require := require.New(t)
			const vm = models.EVM

			sc := &models.Sidecar{
				Name:      tt.subnetName,
				VM:        vm,
				TokenName: tt.tokenName,
				ChainID:   tt.chainID,
			}

			// Write sidecar
			err := ap.CreateSidecar(sc)
			require.NoError(err)

			// Check file exists
			createdPath := ap.GetSidecarPath(tt.subnetName)
			_, err = os.Stat(createdPath)
			require.NoError(err)

			control, err := ap.LoadSidecar(tt.subnetName)
			require.NoError(err)
			require.Equal(*sc, control)

			require.Equal(sc.TokenName, tt.expectedTokenName)

			// Cleanup file
			err = os.Remove(createdPath)
			require.NoError(err)
		})
	}
}

func Test_loadSidecar_success(t *testing.T) {
	require := require.New(t)
	const vm = models.EVM

	ap := newTestApp(t)

	// Write sidecar
	sidecarBytes := []byte("{  \"Name\": \"TEST_subnet\",\n  \"VM\": \"Lux EVM\",\n  \"Subnet\": \"TEST_subnet\"\n  }")
	sidecarPath := ap.GetSidecarPath(subnetName1)
	err := os.MkdirAll(filepath.Dir(sidecarPath), constants.DefaultPerms755)
	require.NoError(err)

	err = os.WriteFile(sidecarPath, sidecarBytes, 0o600)
	require.NoError(err)

	// Check file exists
	_, err = os.Stat(sidecarPath)
	require.NoError(err)

	// Check contents
	expectedSc := models.Sidecar{
		Name:      subnetName1,
		VM:        vm,
		Subnet:    subnetName1,
		TokenName: constants.DefaultTokenName,
	}

	sc, err := ap.LoadSidecar(subnetName1)
	require.NoError(err)
	require.Equal(sc, expectedSc)

	// Cleanup file
	err = os.Remove(sidecarPath)
	require.NoError(err)
}

func Test_loadSidecar_failure_notFound(t *testing.T) {
	require := require.New(t)

	ap := newTestApp(t)

	// Assert file doesn't exist at start
	sidecarPath := ap.GetSidecarPath(subnetName1)
	_, err := os.Stat(sidecarPath)
	require.Error(err)

	_, err = ap.LoadSidecar(subnetName1)
	require.Error(err)
}

func Test_loadSidecar_failure_malformed(t *testing.T) {
	require := require.New(t)

	ap := newTestApp(t)

	// Write sidecar
	sidecarBytes := []byte("bad_sidecar")
	sidecarPath := ap.GetSidecarPath(subnetName1)
	err := os.MkdirAll(filepath.Dir(sidecarPath), constants.DefaultPerms755)
	require.NoError(err)

	err = os.WriteFile(sidecarPath, sidecarBytes, 0o600)
	require.NoError(err)

	// Check file exists
	_, err = os.Stat(sidecarPath)
	require.NoError(err)

	// Check contents
	_, err = ap.LoadSidecar(subnetName1)
	require.Error(err)

	// Cleanup file
	err = os.Remove(sidecarPath)
	require.NoError(err)
}

func Test_genesisExists(t *testing.T) {
	require := require.New(t)

	ap := newTestApp(t)

	// Assert file doesn't exist at start
	result := ap.GenesisExists(subnetName1)
	require.False(result)

	// Create genesis
	genesisPath := ap.GetGenesisPath(subnetName1)
	genesisBytes := []byte("genesis")
	err := os.MkdirAll(filepath.Dir(genesisPath), constants.DefaultPerms755)
	require.NoError(err)
	err = os.WriteFile(genesisPath, genesisBytes, 0o600)
	require.NoError(err)

	// Verify genesis exists
	result = ap.GenesisExists(subnetName1)
	require.True(result)

	// Clean up created genesis
	err = os.Remove(genesisPath)
	require.NoError(err)
}

func newTestApp(t *testing.T) *Lux {
	tempDir := t.TempDir()
	return &Lux{
		baseDir: tempDir,
		Log:     luxlog.NewNoOpLogger(),
	}
}