5e0014307c
#### Does this PR need a docs update or release note? - [x] ⛔ No #### Type of change - [x] 🧹 Tech Debt/Cleanup #### Test Plan - [x] ⚡ Unit test - [x] 💚 E2E
68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package common
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/alcionai/corso/src/internal/tester/tconfig"
|
|
"github.com/alcionai/corso/src/pkg/account"
|
|
"github.com/alcionai/corso/src/pkg/control"
|
|
"github.com/alcionai/corso/src/pkg/count"
|
|
"github.com/alcionai/corso/src/pkg/credentials"
|
|
"github.com/alcionai/corso/src/pkg/services/m365/api"
|
|
)
|
|
|
|
type PermissionInfo struct {
|
|
EntityID string
|
|
Roles []string
|
|
}
|
|
|
|
const (
|
|
sanityBackupID = "SANITY_BACKUP_ID"
|
|
sanityTestSourceContainer = "SANITY_TEST_SOURCE_CONTAINER"
|
|
sanityTestRestoreContainer = "SANITY_TEST_RESTORE_CONTAINER"
|
|
sanityTestUser = "SANITY_TEST_USER"
|
|
)
|
|
|
|
type Envs struct {
|
|
BackupID string
|
|
SourceContainer string
|
|
RestoreContainer string
|
|
GroupID string
|
|
SiteID string
|
|
UserID string
|
|
TeamSiteID string
|
|
}
|
|
|
|
func EnvVars(ctx context.Context) Envs {
|
|
folder := strings.TrimSpace(os.Getenv(sanityTestRestoreContainer))
|
|
|
|
e := Envs{
|
|
BackupID: os.Getenv(sanityBackupID),
|
|
SourceContainer: os.Getenv(sanityTestSourceContainer),
|
|
RestoreContainer: folder,
|
|
GroupID: tconfig.GetM365TeamID(ctx),
|
|
SiteID: tconfig.GetM365SiteID(ctx),
|
|
UserID: tconfig.GetM365UserID(ctx),
|
|
TeamSiteID: tconfig.GetM365TeamSiteID(ctx),
|
|
}
|
|
|
|
if len(os.Getenv(sanityTestUser)) > 0 {
|
|
e.UserID = os.Getenv(sanityTestUser)
|
|
}
|
|
|
|
Infof(ctx, "test env vars: %+v", e)
|
|
|
|
return e
|
|
}
|
|
|
|
func GetAC() (api.Client, error) {
|
|
creds := account.M365Config{
|
|
M365: credentials.GetM365(),
|
|
AzureTenantID: os.Getenv(account.AzureTenantID),
|
|
}
|
|
|
|
return api.NewClient(creds, control.DefaultOptions(), count.New())
|
|
}
|