Merge pull request #505 from synfinatic/80-percent

Increase test coverage to 80%

Author: synfinatic

sso/cache_test.go

@@ -335,6 +335,13 @@ func (suite *CacheTestSuite) TestExpired() {
 	assert.Error(t, suite.cache.Expired(&s))
 	s.settings.CacheRefresh = 0
 	assert.NoError(t, suite.cache.Expired(&s))
+
+	// invalid version
+	c := &Cache{
+		Version: 1, // invalid
+	}
+	err := c.Expired(&s)
+	assert.Error(t, err)
 }
 
 func (suite *CacheTestSuite) TestGetRole() {
@@ -560,3 +567,11 @@ func (suite *CacheTestSuite) TestPruneSSO() {
 	assert.Contains(t, c.SSO, "Primary")
 	assert.NotContains(t, c.SSO, "Invalid")
 }
+
+func TestOpenCacheFailure(t *testing.T) {
+	s := &Settings{}
+	c, err := OpenCache("/dev/null", s)
+	assert.Error(t, err)
+	assert.Equal(t, int64(0), c.ConfigCreatedAt)
+	assert.Equal(t, int64(1), c.Version)
+}

sso/config_test.go

@@ -0,0 +1,82 @@
+package sso
+
+/*
+ * AWS SSO CLI
+ * Copyright (c) 2021-2022 Aaron Turner  <synfinatic at gmail dot com>
+ *
+ * This program is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or with the authors permission any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetRoleMatches(t *testing.T) {
+	accounts := map[string]*SSOAccount{
+		"123456789012": {
+			Tags: map[string]string{
+				"Foo": "foo",
+				"Bar": "bar",
+			},
+			Name: "MyAccount",
+		},
+	}
+	roles := map[string]*SSORole{
+		"FirstRole": {
+			account: accounts["123456789012"],
+			ARN:     "arn:aws:iam::123456789012:role/FirstRole",
+			Tags: map[string]string{
+				"Hello": "There",
+			},
+		},
+		"SecondRole": {
+			account: accounts["123456789012"],
+			ARN:     "arn:aws:iam::123456789012:role/SecondRole",
+			Tags: map[string]string{
+				"Yes": "Please",
+			},
+		},
+	}
+	accounts["123456789012"].Roles = roles
+	s := &SSOConfig{
+		Accounts: accounts,
+	}
+
+	none := map[string]string{
+		"No": "Hits",
+	}
+	empty := s.GetRoleMatches(none)
+	assert.Empty(t, empty)
+
+	twohits := map[string]string{
+		"Foo": "foo",
+		"Bar": "bar",
+	}
+	two := s.GetRoleMatches(twohits)
+	assert.Equal(t, 2, len(two))
+
+	onehit := map[string]string{
+		"Hello": "There",
+	}
+	one := s.GetRoleMatches(onehit)
+	assert.Equal(t, 1, len(one))
+
+	yes := accounts["123456789012"].HasRole("arn:aws:iam::123456789012:role/FirstRole")
+	assert.True(t, yes)
+
+	no := accounts["123456789012"].HasRole("arn:aws:iam::123456789012:role/MissingRole")
+	assert.False(t, no)
+}

sso/options_test.go

@@ -0,0 +1,62 @@
+package sso
+
+/*
+ * AWS SSO CLI
+ * Copyright (c) 2021-2022 Aaron Turner  <synfinatic at gmail dot com>
+ *
+ * This program is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or with the authors permission any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/suite"
+)
+
+func eChecker(in string, breakline bool) bool {
+	return true
+}
+
+type OptionsTestSuite struct {
+	suite.Suite
+	settings *Settings
+}
+
+func TestOptionsTestSuite(t *testing.T) {
+	over := OverrideSettings{}
+	defaults := map[string]interface{}{}
+	// defined in settings_test.go
+	settings, err := LoadSettings(TEST_SETTINGS_FILE, TEST_CACHE_FILE, defaults, over)
+	assert.Nil(t, err)
+
+	s := &OptionsTestSuite{
+		settings: settings,
+	}
+	suite.Run(t, s)
+}
+
+func (suite *OptionsTestSuite) TestDefaultOptions() {
+	t := suite.T()
+	s := suite.settings
+	opts := s.DefaultOptions(eChecker)
+	assert.Equal(t, 4, len(opts))
+}
+
+func (suite *OptionsTestSuite) TestGetColorOptions() {
+	t := suite.T()
+	s := suite.settings
+	opts := s.GetColorOptions()
+	assert.Equal(t, 16, len(opts))
+}

sso/settings_test.go

@@ -356,3 +356,10 @@ func TestApplyDeprecations(t *testing.T) {
 	// AccountIdStr .AccountId => .AccountIdPad
 	assert.Equal(t, "{{ .AccountIdPad }}:{{ .RoleName }}", s.ProfileFormat)
 }
+
+func TestGetExecutable(t *testing.T) {
+	path, err := getExecutable()
+	assert.NoError(t, err)
+	// can't test the NIX path really can we??
+	assert.Contains(t, path, "sso.test")
+}