From a7ea30e1096ed329c2ebedb964f1853ccf2d4cc3 Mon Sep 17 00:00:00 2001 From: Danny Adams Date: Fri, 27 Jan 2023 13:12:18 -0500 Subject: [PATCH] Updates to /connector/sharepoint/helper_test.go Updates to createTestService method. Creates createTestBetaService(). --- .../connector/sharepoint/helper_test.go | 21 +++++++++++++------ 1 file changed, 15 insertions(+), 6 deletions(-) diff --git a/src/internal/connector/sharepoint/helper_test.go b/src/internal/connector/sharepoint/helper_test.go index e716a5bae..1094c36ab 100644 --- a/src/internal/connector/sharepoint/helper_test.go +++ b/src/internal/connector/sharepoint/helper_test.go @@ -4,10 +4,10 @@ import ( "testing" msgraphsdk "github.com/microsoftgraph/msgraph-sdk-go" - "github.com/pkg/errors" "github.com/stretchr/testify/require" "github.com/alcionai/corso/src/internal/connector/graph" + "github.com/alcionai/corso/src/internal/connector/graph/betasdk" "github.com/alcionai/corso/src/internal/connector/onedrive" "github.com/alcionai/corso/src/pkg/account" ) @@ -33,17 +33,26 @@ func (ms *MockGraphService) Adapter() *msgraphsdk.GraphRequestAdapter { // Helper Functions // --------------------------------------------------------------------------- -func createTestService(credentials account.M365Config) (*graph.Service, error) { +func createTestService(t *testing.T, credentials account.M365Config) *graph.Service { adapter, err := graph.CreateAdapter( credentials.AzureTenantID, credentials.AzureClientID, credentials.AzureClientSecret, ) - if err != nil { - return nil, errors.Wrap(err, "creating microsoft graph service for exchange") - } + require.NoError(t, err, "creating microsoft graph service for exchange") - return graph.NewService(adapter), nil + return graph.NewService(adapter) +} + +func createTestBetaService(t *testing.T, credentials account.M365Config) *betasdk.Service { + adapter, err := graph.CreateAdapter( + credentials.AzureTenantID, + credentials.AzureClientID, + credentials.AzureClientSecret, + ) + require.NoError(t, err) + + return betasdk.NewService(adapter) } func expectedPathAsSlice(t *testing.T, tenant, user string, rest ...string) []string {