nigiri/cli/cmd/start_stop_test.go

package cmd

import (
	"fmt"
	"strings"
	"testing"
	"time"

	"github.com/vulpemventures/nigiri/cli/constants"
	"github.com/vulpemventures/nigiri/cli/controller"
)

const (
	liquid  = true
	bitcoin = false
	delete  = true
)

var (
	stopCmd = []string{"stop"}
	// deleteCmd      = append(stopCmd, "--delete")
	startCmd = []string{"start"}
	// liquidStartCmd = append(startCmd, "--liquid")
)

func TestStartStopLiquid(t *testing.T) {
	// Start/Stop
	testStart(t, liquid)
	testStop(t)
	// Start/Delete
	testStart(t, liquid)
	testDelete(t)
}

func TestStartStopBitcoin(t *testing.T) {
	// Start/Stop
	testStart(t, bitcoin)
	testStop(t)
	// Start/Delete
	testStart(t, bitcoin)
	testDelete(t)
}

func TestStopBeforeStartShouldFail(t *testing.T) {
	expectedError := constants.ErrNigiriNotRunning.Error()

	err := testCommand("stop", "", !delete)
	if err == nil {
		t.Fatal("Should return error when trying to stop before starting")
	}
	if err.Error() != expectedError {
		t.Fatalf("Expected error: %s, got: %s", expectedError, err)
	}

	expectedError = constants.ErrNigiriNotExisting.Error()
	err = testCommand("stop", "", delete)
	if err == nil {
		t.Fatal("Should return error when trying to delete before starting")
	}
	if err.Error() != expectedError {
		t.Fatalf("Expected error: %s, got: %s", expectedError, err)
	}
}

func TestStartAfterStartShouldFail(t *testing.T) {
	expectedError := constants.ErrNigiriAlreadyRunning.Error()

	if err := testCommand("start", "", bitcoin); err != nil {
		t.Fatal(err)
	}

	err := testCommand("start", "", bitcoin)
	if err == nil {
		t.Fatal("Should return error when trying to start Nigiri if already started")
	}
	if err.Error() != expectedError {
		t.Fatalf("Expected error: %s, got: %s", expectedError, err)
	}

	err = testCommand("start", "", liquid)
	if err == nil {
		t.Fatal("Should return error when trying to start Nigiri if already started")
	}
	if err.Error() != expectedError {
		t.Fatalf("Expected error: %s, got: %s", expectedError, err)
	}

	if err := testCommand("stop", "", delete); err != nil {
		t.Fatal(err)
	}
}

func testStart(t *testing.T, flag bool) {
	ctl, err := controller.NewController()
	if err != nil {
		t.Fatal(err)
	}
	if err := testCommand("start", "", flag); err != nil {
		t.Fatal(err)
	}
	//Give some time to nigiri to be ready before calling
	time.Sleep(5 * time.Second)
	if isRunning, err := ctl.IsNigiriRunning(); err != nil {
		t.Fatal(err)
	} else if !isRunning {
		t.Fatal("Nigiri should have been started but services have not been found among running containers")
	}
}

func testStop(t *testing.T) {
	ctl, err := controller.NewController()
	if err != nil {
		t.Fatal(err)
	}
	if err := testCommand("stop", "", !delete); err != nil {
		t.Fatal(err)
	}
	//Give some time to nigiri to be ready before calling
	time.Sleep(5 * time.Second)
	if isStopped, err := ctl.IsNigiriStopped(); err != nil {
		t.Fatal(err)
	} else if !isStopped {
		t.Fatal("Nigiri should have been stopped but services have not been found among stopped containers")
	}
}

func testDelete(t *testing.T) {
	ctl, err := controller.NewController()
	if err != nil {
		t.Fatal(err)
	}

	if err := testCommand("stop", "", delete); err != nil {
		t.Fatal(err)
	}
	if isStopped, err := ctl.IsNigiriStopped(); err != nil {
		t.Fatal(err)
	} else if isStopped {
		t.Fatal("Nigiri should have been terminated at this point but services have been found among stopped containers")
	}
}

func testCommand(command, arg string, flag bool) error {
	cmd := RootCmd
	cmd.SetArgs(nil)

	if command == "start" {
		args := append(startCmd, fmt.Sprintf("--liquid=%t", flag))
		cmd.SetArgs(args)
	}
	if command == "stop" {
		args := append(stopCmd, fmt.Sprintf("--delete=%t", flag))
		cmd.SetArgs(args)
	}
	if command == "logs" {
		logsCmd := []string{command, arg, fmt.Sprintf("--liquid=%t", flag)}
		cmd.SetArgs(logsCmd)
	}
	if command == "faucet" {
		faucetCmd := []string{command, arg, fmt.Sprintf("--liquid=%t", flag)}
		cmd.SetArgs(faucetCmd)
	}
	if command == "rpc" {
		logsCmd := []string{command, fmt.Sprintf("--liquid=%t", flag)}
		args := strings.Fields(arg)
		logsCmd = append(logsCmd, args...)
		cmd.SetArgs(logsCmd)
	}
	if command == "mint" {
		logsCmd := []string{command}
		args := strings.Fields(arg)
		logsCmd = append(logsCmd, args...)
		cmd.SetArgs(logsCmd)
	}
	if command == "push" {
		pushCmd := []string{command, arg, fmt.Sprintf("--liquid=%t", flag)}
		cmd.SetArgs(pushCmd)
	}

	if err := cmd.Execute(); err != nil {
		return err
	}

	return nil
}