setting aside in case its wanted
This commit is contained in:
ryanfkeepers
parent
5efbd966fd
commit
dab0808ca7
@ -1213,6 +1213,441 @@ func (suite *ExchangeRestoreIntgSuite) TestRestore_Run_exchangeWithAdvancedOptio
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
//
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
//
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
//
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
//
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
//
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
func (suite *ExchangeRestoreIntgSuite) TestRestore_Run_exchangePairwiseAdvancedOptions() {
|
||||||
|
var (
|
||||||
|
t = suite.T()
|
||||||
|
userID = suite.its.user.ID
|
||||||
|
initCfg = ctrlTD.DefaultRestoreConfig("exchange_adv_pairwise_init")
|
||||||
|
origItemIDs = map[path.CategoryType][]string{}
|
||||||
|
)
|
||||||
|
|
||||||
|
ctx, flush := tester.NewContext(t)
|
||||||
|
defer flush()
|
||||||
|
|
||||||
|
// a backup is required to run restores
|
||||||
|
|
||||||
|
// first: generate four items for each category to be restored
|
||||||
|
// that will allow us to restore a pairwise set of items each time:
|
||||||
|
// one item that was already restored to test skip/copy/replace,
|
||||||
|
// and one item that isn't yet restored.
|
||||||
|
|
||||||
|
acc := suite.its.ac.Contacts()
|
||||||
|
|
||||||
|
origItemIDs[path.ContactsCategory] = make([]string, 4)
|
||||||
|
|
||||||
|
gc, err := acc.CreateContainer(ctx, userID, "", initCfg.Location)
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
for i := 0; i < 4; i++ {
|
||||||
|
bb := exchMock.ContactBytes(fmt.Sprintf("pairwise_%d", i))
|
||||||
|
|
||||||
|
body, err := api.BytesToContactable(bb)
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
itm, err := acc.PostItem(ctx, userID, ptr.Val(gc.GetId()), body)
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
origItemIDs[path.ContactsCategory][i] = ptr.Val(itm.GetId())
|
||||||
|
}
|
||||||
|
|
||||||
|
// ace := suite.its.ac.Events()
|
||||||
|
|
||||||
|
// origItemIDs[path.EventsCategory] = make([]string, 4)
|
||||||
|
|
||||||
|
// gc, err = ace.CreateContainer(ctx, userID, "", initCfg.Location)
|
||||||
|
// require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
// for i := 0; i < 4; i++ {
|
||||||
|
// bb := exchMock.EventBytes(fmt.Sprintf("pairwise_%d", i))
|
||||||
|
|
||||||
|
// body, err := api.BytesToEventable(bb)
|
||||||
|
// require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
// itm, err := ace.PostItem(ctx, userID, ptr.Val(gc.GetId()), body)
|
||||||
|
// require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
// origItemIDs[path.EventsCategory][i] = ptr.Val(itm.GetId())
|
||||||
|
// }
|
||||||
|
|
||||||
|
acm := suite.its.ac.Mail()
|
||||||
|
|
||||||
|
origItemIDs[path.EmailCategory] = make([]string, 4)
|
||||||
|
|
||||||
|
gc, err = acm.CreateContainer(ctx, userID, api.MsgFolderRoot, initCfg.Location)
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
for i := 0; i < 4; i++ {
|
||||||
|
bb := exchMock.MessageBytes(fmt.Sprintf("pairwise_%d", i))
|
||||||
|
|
||||||
|
body, err := api.BytesToMessageable(bb)
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
itm, err := acm.PostItem(ctx, userID, ptr.Val(gc.GetId()), body)
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
origItemIDs[path.EmailCategory][i] = ptr.Val(itm.GetId())
|
||||||
|
}
|
||||||
|
|
||||||
|
// then backup the folders generated for this test
|
||||||
|
|
||||||
|
baseSel := selectors.NewExchangeBackup([]string{suite.its.user.ID})
|
||||||
|
baseSel.Include(
|
||||||
|
// events cannot be run, for the same reason as incremental backups: the user needs
|
||||||
|
// to have their account recycled.
|
||||||
|
// baseSel.EventCalendars([]string{initCfg.Location}, selectors.PrefixMatch()),
|
||||||
|
baseSel.ContactFolders([]string{initCfg.Location}, selectors.PrefixMatch()),
|
||||||
|
baseSel.MailFolders([]string{initCfg.Location}, selectors.PrefixMatch()))
|
||||||
|
|
||||||
|
baseSel.DiscreteOwner = suite.its.user.ID
|
||||||
|
|
||||||
|
var (
|
||||||
|
mb = evmock.NewBus()
|
||||||
|
opts = control.DefaultOptions()
|
||||||
|
)
|
||||||
|
|
||||||
|
bo, bod := prepNewTestBackupOp(t, ctx, mb, baseSel.Selector, opts, version.Backup)
|
||||||
|
defer bod.close(t, ctx)
|
||||||
|
|
||||||
|
runAndCheckBackup(t, ctx, &bo, mb, false)
|
||||||
|
|
||||||
|
rsel, err := baseSel.ToExchangeRestore()
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
var (
|
||||||
|
restoreCfg = ctrlTD.DefaultRestoreConfig("exchange_adv_pairwise")
|
||||||
|
sel = rsel.Selector
|
||||||
|
countItemsInRestore int
|
||||||
|
|
||||||
|
itemIDs = map[path.CategoryType]map[string]struct{}{}
|
||||||
|
collisionKeys = map[path.CategoryType]map[string]string{}
|
||||||
|
containerIDs = map[path.CategoryType]string{}
|
||||||
|
parentContainerIDs = map[path.CategoryType]string{
|
||||||
|
path.EmailCategory: api.MsgFolderRoot,
|
||||||
|
}
|
||||||
|
|
||||||
|
testCategories = map[path.CategoryType]clientItemPager{
|
||||||
|
path.ContactsCategory: suite.its.ac.Contacts(),
|
||||||
|
path.EmailCategory: suite.its.ac.Mail(),
|
||||||
|
// path.EventsCategory: suite.its.ac.Events(),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// initial restore: only restore the first item of each four.
|
||||||
|
|
||||||
|
suite.Run("baseline", func() {
|
||||||
|
t := suite.T()
|
||||||
|
|
||||||
|
ctx, flush := tester.NewContext(t)
|
||||||
|
defer flush()
|
||||||
|
|
||||||
|
rSel := selectors.NewExchangeBackup([]string{userID})
|
||||||
|
rsel.Include(
|
||||||
|
rSel.Events(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.EventsCategory][0]}),
|
||||||
|
rSel.Contacts(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.ContactsCategory][0]}),
|
||||||
|
rSel.Mails(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.EmailCategory][0]}))
|
||||||
|
|
||||||
|
mb := evmock.NewBus()
|
||||||
|
ctr1 := count.New()
|
||||||
|
|
||||||
|
restoreCfg.OnCollision = control.Copy
|
||||||
|
|
||||||
|
ro, _ := prepNewTestRestoreOp(
|
||||||
|
t,
|
||||||
|
ctx,
|
||||||
|
bod.st,
|
||||||
|
bo.Results.BackupID,
|
||||||
|
mb,
|
||||||
|
ctr1,
|
||||||
|
rSel.Selector,
|
||||||
|
opts,
|
||||||
|
restoreCfg)
|
||||||
|
|
||||||
|
runAndCheckRestore(t, ctx, &ro, mb, false)
|
||||||
|
|
||||||
|
// get all files in folder, use these as the base
|
||||||
|
// set of files to compare against.
|
||||||
|
|
||||||
|
for cat, ac := range testCategories {
|
||||||
|
suite.Run(cat.String(), func() {
|
||||||
|
t := suite.T()
|
||||||
|
|
||||||
|
ctx, flush := tester.NewContext(t)
|
||||||
|
defer flush()
|
||||||
|
|
||||||
|
itemIDs[cat], collisionKeys[cat], containerIDs[cat] = getCollKeysAndItemIDs(
|
||||||
|
t,
|
||||||
|
ctx,
|
||||||
|
ac,
|
||||||
|
userID,
|
||||||
|
parentContainerIDs[cat],
|
||||||
|
restoreCfg.Location)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
countItemsInRestore = len(collisionKeys[path.ContactsCategory]) +
|
||||||
|
len(collisionKeys[path.EmailCategory]) +
|
||||||
|
len(collisionKeys[path.EventsCategory])
|
||||||
|
checkRestoreCounts(t, ctr1, 0, 0, countItemsInRestore)
|
||||||
|
assert.Equal(t, len(collisionKeys), countItemsInRestore, "only one item per category should be restored")
|
||||||
|
})
|
||||||
|
|
||||||
|
// skip restore; skip item 0, restore item 1
|
||||||
|
|
||||||
|
suite.Run("skip collisions", func() {
|
||||||
|
t := suite.T()
|
||||||
|
|
||||||
|
ctx, flush := tester.NewContext(t)
|
||||||
|
defer flush()
|
||||||
|
|
||||||
|
rSel := selectors.NewExchangeBackup([]string{userID})
|
||||||
|
rsel.Include(
|
||||||
|
rSel.Events(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.EventsCategory][0], origItemIDs[path.EventsCategory][1]}),
|
||||||
|
rSel.Contacts(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.ContactsCategory][0], origItemIDs[path.ContactsCategory][1]}),
|
||||||
|
rSel.Mails(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.EmailCategory][0], origItemIDs[path.EmailCategory][1]}))
|
||||||
|
|
||||||
|
mb := evmock.NewBus()
|
||||||
|
ctr2 := count.New()
|
||||||
|
|
||||||
|
restoreCfg.OnCollision = control.Skip
|
||||||
|
|
||||||
|
ro, _ := prepNewTestRestoreOp(
|
||||||
|
t,
|
||||||
|
ctx,
|
||||||
|
bod.st,
|
||||||
|
bo.Results.BackupID,
|
||||||
|
mb,
|
||||||
|
ctr2,
|
||||||
|
rSel.Selector,
|
||||||
|
opts,
|
||||||
|
restoreCfg)
|
||||||
|
|
||||||
|
deets := runAndCheckRestore(t, ctx, &ro, mb, false)
|
||||||
|
|
||||||
|
assert.Equal(t, len(origItemIDs), len(deets.Entries), "one item per category should have been restored")
|
||||||
|
|
||||||
|
checkRestoreCounts(t, ctr2, len(origItemIDs), 0, len(origItemIDs))
|
||||||
|
|
||||||
|
result := map[string]string{}
|
||||||
|
|
||||||
|
for cat, ac := range testCategories {
|
||||||
|
suite.Run(cat.String(), func() {
|
||||||
|
t := suite.T()
|
||||||
|
|
||||||
|
ctx, flush := tester.NewContext(t)
|
||||||
|
defer flush()
|
||||||
|
|
||||||
|
m := filterCollisionKeyResults(
|
||||||
|
t,
|
||||||
|
ctx,
|
||||||
|
userID,
|
||||||
|
containerIDs[cat],
|
||||||
|
GetItemsInContainerByCollisionKeyer[string](ac),
|
||||||
|
collisionKeys[cat])
|
||||||
|
maps.Copy(result, m)
|
||||||
|
|
||||||
|
currentIDs, err := ac.GetItemIDsInContainer(ctx, userID, containerIDs[cat])
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
assert.Subset(t, maps.Keys(currentIDs), maps.Keys(itemIDs[cat]), "restore should contain all previous ids")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.Len(t, result, len(origItemIDs), "one item per category should get added")
|
||||||
|
})
|
||||||
|
|
||||||
|
// copy restore; copy item 0, restore item 2
|
||||||
|
|
||||||
|
suite.Run("copy collisions", func() {
|
||||||
|
t := suite.T()
|
||||||
|
|
||||||
|
ctx, flush := tester.NewContext(t)
|
||||||
|
defer flush()
|
||||||
|
|
||||||
|
rSel := selectors.NewExchangeBackup([]string{userID})
|
||||||
|
rsel.Include(
|
||||||
|
rSel.Events(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.EventsCategory][0], origItemIDs[path.EventsCategory][2]}),
|
||||||
|
rSel.Contacts(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.ContactsCategory][0], origItemIDs[path.ContactsCategory][2]}),
|
||||||
|
rSel.Mails(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.EmailCategory][0], origItemIDs[path.EmailCategory][2]}))
|
||||||
|
|
||||||
|
mb := evmock.NewBus()
|
||||||
|
ctr4 := count.New()
|
||||||
|
|
||||||
|
restoreCfg.OnCollision = control.Copy
|
||||||
|
|
||||||
|
ro, _ := prepNewTestRestoreOp(
|
||||||
|
t,
|
||||||
|
ctx,
|
||||||
|
bod.st,
|
||||||
|
bo.Results.BackupID,
|
||||||
|
mb,
|
||||||
|
ctr4,
|
||||||
|
sel,
|
||||||
|
opts,
|
||||||
|
restoreCfg)
|
||||||
|
|
||||||
|
deets := runAndCheckRestore(t, ctx, &ro, mb, false)
|
||||||
|
filtEnts := []details.Entry{}
|
||||||
|
|
||||||
|
for _, e := range deets.Entries {
|
||||||
|
if e.Folder == nil {
|
||||||
|
filtEnts = append(filtEnts, e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.Len(t, filtEnts, countItemsInRestore, "every item should have been copied")
|
||||||
|
|
||||||
|
checkRestoreCounts(t, ctr4, 0, 0, countItemsInRestore)
|
||||||
|
|
||||||
|
result := map[string]string{}
|
||||||
|
|
||||||
|
for cat, ac := range testCategories {
|
||||||
|
suite.Run(cat.String(), func() {
|
||||||
|
t := suite.T()
|
||||||
|
|
||||||
|
ctx, flush := tester.NewContext(t)
|
||||||
|
defer flush()
|
||||||
|
|
||||||
|
m := filterCollisionKeyResults(
|
||||||
|
t,
|
||||||
|
ctx,
|
||||||
|
userID,
|
||||||
|
containerIDs[cat],
|
||||||
|
GetItemsInContainerByCollisionKeyer[string](ac),
|
||||||
|
collisionKeys[cat])
|
||||||
|
maps.Copy(result, m)
|
||||||
|
|
||||||
|
currentIDs, err := ac.GetItemIDsInContainer(ctx, userID, containerIDs[cat])
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
assert.Equal(t, 2*len(itemIDs[cat]), len(currentIDs), "count of ids should be double from before")
|
||||||
|
assert.Subset(t, maps.Keys(currentIDs), maps.Keys(itemIDs[cat]), "original item should exist after copy")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: we have the option of modifying copy creations in exchange
|
||||||
|
// so that the results don't collide. But we haven't made that
|
||||||
|
// decision yet.
|
||||||
|
assert.Len(t, result, 0, "no items should have been added as copies")
|
||||||
|
})
|
||||||
|
|
||||||
|
// replace restore; replace item 0, restore item 3
|
||||||
|
|
||||||
|
suite.Run("replace collisions", func() {
|
||||||
|
t := suite.T()
|
||||||
|
|
||||||
|
ctx, flush := tester.NewContext(t)
|
||||||
|
defer flush()
|
||||||
|
|
||||||
|
rSel := selectors.NewExchangeBackup([]string{userID})
|
||||||
|
rsel.Include(
|
||||||
|
rSel.Events(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.EventsCategory][0], origItemIDs[path.EventsCategory][2]}),
|
||||||
|
rSel.Contacts(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.ContactsCategory][0], origItemIDs[path.ContactsCategory][2]}),
|
||||||
|
rSel.Mails(
|
||||||
|
[]string{initCfg.Location},
|
||||||
|
[]string{origItemIDs[path.EmailCategory][0], origItemIDs[path.EmailCategory][2]}))
|
||||||
|
|
||||||
|
mb := evmock.NewBus()
|
||||||
|
ctr3 := count.New()
|
||||||
|
|
||||||
|
restoreCfg.OnCollision = control.Replace
|
||||||
|
|
||||||
|
ro, _ := prepNewTestRestoreOp(
|
||||||
|
t,
|
||||||
|
ctx,
|
||||||
|
bod.st,
|
||||||
|
bo.Results.BackupID,
|
||||||
|
mb,
|
||||||
|
ctr3,
|
||||||
|
sel,
|
||||||
|
opts,
|
||||||
|
restoreCfg)
|
||||||
|
|
||||||
|
deets := runAndCheckRestore(t, ctx, &ro, mb, false)
|
||||||
|
filtEnts := []details.Entry{}
|
||||||
|
|
||||||
|
for _, e := range deets.Entries {
|
||||||
|
if e.Folder == nil {
|
||||||
|
filtEnts = append(filtEnts, e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.Len(t, filtEnts, len(origItemIDs), "one item for each category should have been replaced")
|
||||||
|
|
||||||
|
checkRestoreCounts(t, ctr3, 0, len(origItemIDs), len(origItemIDs))
|
||||||
|
|
||||||
|
result := map[string]string{}
|
||||||
|
|
||||||
|
for cat, ac := range testCategories {
|
||||||
|
suite.Run(cat.String(), func() {
|
||||||
|
t := suite.T()
|
||||||
|
|
||||||
|
ctx, flush := tester.NewContext(t)
|
||||||
|
defer flush()
|
||||||
|
|
||||||
|
m := filterCollisionKeyResults(
|
||||||
|
t,
|
||||||
|
ctx,
|
||||||
|
userID,
|
||||||
|
containerIDs[cat],
|
||||||
|
GetItemsInContainerByCollisionKeyer[string](ac),
|
||||||
|
collisionKeys[cat])
|
||||||
|
maps.Copy(result, m)
|
||||||
|
|
||||||
|
currentIDs, err := ac.GetItemIDsInContainer(ctx, userID, containerIDs[cat])
|
||||||
|
require.NoError(t, err, clues.ToCore(err))
|
||||||
|
|
||||||
|
assert.Equal(t, 3*len(itemIDs[cat]), len(currentIDs), "3x items have been restored so far")
|
||||||
|
for orig := range itemIDs[cat] {
|
||||||
|
assert.NotContains(t, currentIDs, orig, "original item should not exist after replacement")
|
||||||
|
}
|
||||||
|
|
||||||
|
// note: this changes the count of itemIDs from 1 to 3
|
||||||
|
itemIDs[cat] = currentIDs
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.Len(t, result, len(origItemIDs), "one item per category should have been replaced")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
func (suite *ExchangeRestoreIntgSuite) TestRestore_Run_exchangeAlternateProtectedResource() {
|
func (suite *ExchangeRestoreIntgSuite) TestRestore_Run_exchangeAlternateProtectedResource() {
|
||||||
t := suite.T()
|
t := suite.T()
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user