larryh/corso
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Test case for reference attachment

Browse Source
This commit is contained in:
Vaibhav Kamra 2022-10-19 11:20:59 -07:00
parent 5d2bc1eb4e
commit 8840116380
2 changed files with 79 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/internal/connector/exchange/attachment.go
View File

@ -55,7 +55,7 @@ func uploadAttachment(ctx context.Context, service graph.Service, userID, folder
Attachments().
Post(ctx, attachment, nil)
return err
return errors.Wrapf(err, "failed to post attachment: "+support.ConnectorStackErrorTrace(err))
}
return uploadLargeAttachment(ctx, service, userID, folderID, messageID, attachment)

143
src/internal/connector/exchange/exchange_service_test.go
View File

@ -397,85 +397,98 @@ func (suite *ExchangeServiceSuite) TestRestoreExchangeObject() {
cleanupFunc func(context.Context, graph.Service, string, string) error
destination func(context.Context) string
}{
// {
// name: "Test Mail",
// bytes: mockconnector.GetMockMessageBytes("Restore Exchange Object"),
// category: path.EmailCategory,
// cleanupFunc: DeleteMailFolder,
// destination: func(ctx context.Context) string {
// folderName := "TestRestoreMailObject: " + common.FormatSimpleDateTime(now)
// folder, err := CreateMailFolder(ctx, suite.es, userID, folderName)
// require.NoError(t, err)
// return *folder.GetId()
// },
// },
// {
// name: "Test Mail: One Direct Attachment",
// bytes: mockconnector.GetMockMessageWithDirectAttachment("Restore 1 Attachment"),
// category: path.EmailCategory,
// cleanupFunc: DeleteMailFolder,
// destination: func(ctx context.Context) string {
// folderName := "TestRestoreMailwithAttachment: " + common.FormatSimpleDateTime(now)
// folder, err := CreateMailFolder(ctx, suite.es, userID, folderName)
// require.NoError(t, err)
// return *folder.GetId()
// },
// },
{
name: "Test Mail",
bytes: mockconnector.GetMockMessageBytes("Restore Exchange Object"),
name: "Test Mail: One Reference Attachment",
bytes: mockconnector.GetMessageWithOneDriveAttachment("Restore 1 Attachment"),
category: path.EmailCategory,
cleanupFunc: DeleteMailFolder,
destination: func(ctx context.Context) string {
folderName := "TestRestoreMailObject: " + common.FormatSimpleDateTime(now)
folderName := "TestRestoreMailwithReferenceAttachment: " + common.FormatSimpleDateTime(now)
folder, err := CreateMailFolder(ctx, suite.es, userID, folderName)
require.NoError(t, err)
return *folder.GetId()
},
},
{
name: "Test Mail: One Direct Attachment",
bytes: mockconnector.GetMockMessageWithDirectAttachment("Restore 1 Attachment"),
category: path.EmailCategory,
cleanupFunc: DeleteMailFolder,
destination: func(ctx context.Context) string {
folderName := "TestRestoreMailwithAttachment: " + common.FormatSimpleDateTime(now)
folder, err := CreateMailFolder(ctx, suite.es, userID, folderName)
require.NoError(t, err)
// {
// name: "Test Mail: One Large Attachment",
// bytes: mockconnector.GetMockMessageWithLargeAttachment("Restore Large Attachment"),
// category: path.EmailCategory,
// cleanupFunc: DeleteMailFolder,
// destination: func(ctx context.Context) string {
// folderName := "TestRestoreMailwithLargeAttachment: " + common.FormatSimpleDateTime(now)
// folder, err := CreateMailFolder(ctx, suite.es, userID, folderName)
// require.NoError(t, err)
return *folder.GetId()
},
},
{
name: "Test Mail: One Large Attachment",
bytes: mockconnector.GetMockMessageWithLargeAttachment("Restore Large Attachment"),
category: path.EmailCategory,
cleanupFunc: DeleteMailFolder,
destination: func(ctx context.Context) string {
folderName := "TestRestoreMailwithLargeAttachment: " + common.FormatSimpleDateTime(now)
folder, err := CreateMailFolder(ctx, suite.es, userID, folderName)
require.NoError(t, err)
// return *folder.GetId()
// },
// },
// {
// name: "Test Mail: Two Attachments",
// bytes: mockconnector.GetMockMessageWithTwoAttachments("Restore 2 Attachments"),
// category: path.EmailCategory,
// cleanupFunc: DeleteMailFolder,
// destination: func(ctx context.Context) string {
// folderName := "TestRestoreMailwithAttachments: " + common.FormatSimpleDateTime(now)
// folder, err := CreateMailFolder(ctx, suite.es, userID, folderName)
// require.NoError(t, err)
return *folder.GetId()
},
},
{
name: "Test Mail: Two Attachments",
bytes: mockconnector.GetMockMessageWithTwoAttachments("Restore 2 Attachments"),
category: path.EmailCategory,
cleanupFunc: DeleteMailFolder,
destination: func(ctx context.Context) string {
folderName := "TestRestoreMailwithAttachments: " + common.FormatSimpleDateTime(now)
folder, err := CreateMailFolder(ctx, suite.es, userID, folderName)
require.NoError(t, err)
// return *folder.GetId()
// },
// },
// // TODO: #884 - reinstate when able to specify root folder by name
// {
// name: "Test Contact",
// bytes: mockconnector.GetMockContactBytes("Test_Omega"),
// category: path.ContactsCategory,
// cleanupFunc: DeleteContactFolder,
// destination: func(ctx context.Context) string {
// folderName := "TestRestoreContactObject: " + common.FormatSimpleDateTime(now)
// folder, err := CreateContactFolder(ctx, suite.es, userID, folderName)
// require.NoError(t, err)
return *folder.GetId()
},
},
// TODO: #884 - reinstate when able to specify root folder by name
{
name: "Test Contact",
bytes: mockconnector.GetMockContactBytes("Test_Omega"),
category: path.ContactsCategory,
cleanupFunc: DeleteContactFolder,
destination: func(ctx context.Context) string {
folderName := "TestRestoreContactObject: " + common.FormatSimpleDateTime(now)
folder, err := CreateContactFolder(ctx, suite.es, userID, folderName)
require.NoError(t, err)
// return *folder.GetId()
// },
// },
// {
// name: "Test Events",
// bytes: mockconnector.GetDefaultMockEventBytes("Restored Event Object"),
// category: path.EventsCategory,
// cleanupFunc: DeleteCalendar,
// destination: func(ctx context.Context) string {
// calendarName := "TestRestoreEventObject: " + common.FormatSimpleDateTime(now)
// calendar, err := CreateCalendar(ctx, suite.es, userID, calendarName)
// require.NoError(t, err)
return *folder.GetId()
},
},
{
name: "Test Events",
bytes: mockconnector.GetDefaultMockEventBytes("Restored Event Object"),
category: path.EventsCategory,
cleanupFunc: DeleteCalendar,
destination: func(ctx context.Context) string {
calendarName := "TestRestoreEventObject: " + common.FormatSimpleDateTime(now)
calendar, err := CreateCalendar(ctx, suite.es, userID, calendarName)
require.NoError(t, err)
return *calendar.GetId()
},
},
// return *calendar.GetId()
// },
// },
}
for _, test := range tests {