689c5cc1e9
* separate tenantID from m365 creds Now that account.Account is in place, tenant id needs to get removed from the credential set (it isn't actually a secret) and placed in the account configuration instead.
123 lines
2.6 KiB
Go
123 lines
2.6 KiB
Go
package account_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/alcionai/corso/pkg/account"
|
|
"github.com/alcionai/corso/pkg/credentials"
|
|
)
|
|
|
|
type M365CfgSuite struct {
|
|
suite.Suite
|
|
}
|
|
|
|
func TestM365CfgSuite(t *testing.T) {
|
|
suite.Run(t, new(M365CfgSuite))
|
|
}
|
|
|
|
var goodM365Config = account.M365Config{
|
|
M365: credentials.M365{
|
|
ClientID: "cid",
|
|
ClientSecret: "cs",
|
|
},
|
|
TenantID: "tid",
|
|
}
|
|
|
|
func (suite *M365CfgSuite) TestM365Config_Config() {
|
|
m365 := goodM365Config
|
|
c, err := m365.StringConfig()
|
|
require.NoError(suite.T(), err)
|
|
|
|
table := []struct {
|
|
key string
|
|
expect string
|
|
}{
|
|
{"m365_clientID", m365.ClientID},
|
|
{"m365_clientSecret", m365.ClientSecret},
|
|
{"m365_tenantID", m365.TenantID},
|
|
}
|
|
for _, test := range table {
|
|
assert.Equal(suite.T(), test.expect, c[test.key])
|
|
}
|
|
}
|
|
|
|
func (suite *M365CfgSuite) TestAccount_M365Config() {
|
|
t := suite.T()
|
|
|
|
in := goodM365Config
|
|
a, err := account.NewAccount(account.ProviderM365, in)
|
|
require.NoError(t, err)
|
|
out, err := a.M365Config()
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, in.ClientID, out.ClientID)
|
|
assert.Equal(t, in.ClientSecret, out.ClientSecret)
|
|
assert.Equal(t, in.TenantID, out.TenantID)
|
|
}
|
|
|
|
func makeTestM365Cfg(cid, cs, tid string) account.M365Config {
|
|
return account.M365Config{
|
|
M365: credentials.M365{
|
|
ClientID: cid,
|
|
ClientSecret: cs,
|
|
},
|
|
TenantID: tid,
|
|
}
|
|
}
|
|
|
|
func (suite *M365CfgSuite) TestAccount_M365Config_InvalidCases() {
|
|
// missing required properties
|
|
table := []struct {
|
|
name string
|
|
cfg account.M365Config
|
|
}{
|
|
{"missing client ID", makeTestM365Cfg("", "cs", "tid")},
|
|
{"missing client secret", makeTestM365Cfg("cid", "", "tid")},
|
|
{"missing tenant ID", makeTestM365Cfg("cid", "cs", "")},
|
|
}
|
|
for _, test := range table {
|
|
suite.T().Run(test.name, func(t *testing.T) {
|
|
_, err := account.NewAccount(account.ProviderUnknown, test.cfg)
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
// required property not populated in account
|
|
table2 := []struct {
|
|
name string
|
|
amend func(account.Account)
|
|
}{
|
|
{
|
|
"missing clientID",
|
|
func(a account.Account) {
|
|
a.Config["m365_clientID"] = ""
|
|
},
|
|
},
|
|
{
|
|
"missing client secret",
|
|
func(a account.Account) {
|
|
a.Config["m365_clientSecret"] = ""
|
|
},
|
|
},
|
|
{
|
|
"missing tenant id",
|
|
func(a account.Account) {
|
|
a.Config["m365_tenantID"] = ""
|
|
},
|
|
},
|
|
}
|
|
for _, test := range table2 {
|
|
suite.T().Run(test.name, func(t *testing.T) {
|
|
st, err := account.NewAccount(account.ProviderUnknown, goodM365Config)
|
|
assert.NoError(t, err)
|
|
test.amend(st)
|
|
_, err = st.M365Config()
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
}
|