From 5ca341eb56cdf96cdf7b10a34f5c2a386fca3260 Mon Sep 17 00:00:00 2001 From: Danny Adams Date: Fri, 27 Jan 2023 15:33:13 -0500 Subject: [PATCH] Update to /connector/sharepoint/helper_test.go Creates a test update function for the package. --- src/internal/connector/sharepoint/helper_test.go | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/src/internal/connector/sharepoint/helper_test.go b/src/internal/connector/sharepoint/helper_test.go index 1094c36ab..b4148fd02 100644 --- a/src/internal/connector/sharepoint/helper_test.go +++ b/src/internal/connector/sharepoint/helper_test.go @@ -9,6 +9,7 @@ import ( "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/internal/connector/support" "github.com/alcionai/corso/src/pkg/account" ) @@ -17,6 +18,16 @@ import ( // --------------------------------------------------------------------------- type MockGraphService struct{} +type MockUpdater struct { + UpdateState func(*support.ConnectorOperationStatus) +} + +func (mu *MockUpdater) UpdateStatus(input *support.ConnectorOperationStatus) { + if mu.UpdateState != nil { + mu.UpdateState(input) + } +} + //------------------------------------------------------------ // Interface Functions: @See graph.Service //------------------------------------------------------------