Rename test file for setup subcommand

Author: Chris Gilmer

cmd/main_test.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+<<<<<<< HEAD
 	"io/ioutil"
 	"log"
 	"os"
@@ -85,4 +86,87 @@ func TestGetPartition(t *testing.T) {
 
 	_, err = getPartition("aws-under-the-sea")
 	assert.Error(t, err)
+=======
+	"fmt"
+	"io/ioutil"
+	"log"
+	"os"
+	"strings"
+	"testing"
+
+	"github.com/spf13/pflag"
+	"github.com/spf13/viper"
+	"github.com/stretchr/testify/suite"
+)
+
+type commandTestSuite struct {
+	suite.Suite
+	viper  *viper.Viper
+	logger *log.Logger
+}
+
+type initFlags func(f *pflag.FlagSet)
+
+func (suite *commandTestSuite) Setup(fn initFlags, flagSet []string) {
+	// Disable any logging that isn't attached to the logger unless using the verbose flag
+	log.SetOutput(ioutil.Discard)
+	log.SetFlags(0)
+
+	// Setup logger
+	var logger = log.New(os.Stdout, "", log.LstdFlags)
+
+	// Remove the flags for the logger
+	logger.SetFlags(0)
+	suite.SetLogger(logger)
+
+	// Setup viper
+	suite.viper = nil
+
+	flag := pflag.NewFlagSet(os.Args[0], pflag.ExitOnError)
+	fn(flag)
+	errFlagParse := flag.Parse(flagSet)
+	if errFlagParse != nil {
+		suite.logger.Fatal(errFlagParse)
+	}
+
+	v := viper.New()
+	err := v.BindPFlags(flag)
+	if err != nil {
+		suite.logger.Fatal(fmt.Errorf("could not bind flags: %w", err))
+	}
+	v.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
+	v.AutomaticEnv()
+
+	suite.SetViper(v)
+}
+
+func (suite *commandTestSuite) SetViper(v *viper.Viper) {
+	suite.viper = v
+}
+
+func (suite *commandTestSuite) SetLogger(logger *log.Logger) {
+	suite.logger = logger
+}
+
+func TestCommandSuite(t *testing.T) {
+	suite.Run(t, &commandTestSuite{})
+}
+
+func (suite *commandTestSuite) TestAddProfileFlags() {
+	suite.Setup(AddProfileInitFlags, []string{
+		"--aws-profile-account", "test-new:012345678901",
+		"--iam-user", "me",
+		"--iam-role", "engineer",
+	})
+	suite.NoError(AddProfileCheckConfig(suite.viper))
+}
+
+func (suite *commandTestSuite) TestSetupFlags() {
+	suite.Setup(SetupUserInitFlags, []string{
+		"--aws-profile-account", "test-id:012345678901",
+		"--iam-user", "me",
+		"--iam-role", "engineer",
+	})
+	suite.NoError(SetupUserCheckConfig(suite.viper))
+>>>>>>> 887956c... Rename test file for setup subcommand
 }