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 //------------------------------------------------------------