3a2d0876dd
introduces a new type wrapping a nested map so that aggregation of globally excluded items in driveish services don't need to manage the map updates themselves. --- #### Does this PR need a docs update or release note? - [x] ⛔ No #### Type of change - [x] 🧹 Tech Debt/Cleanup #### Issue(s) * #2340 #### Test Plan - [x] ⚡ Unit test - [x] 💚 E2E
45 lines
869 B
Go
45 lines
869 B
Go
package mock
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/alcionai/corso/src/internal/common/prefixmatcher"
|
|
)
|
|
|
|
var _ prefixmatcher.StringSetReader = &PrefixMap{}
|
|
|
|
type PrefixMap struct {
|
|
prefixmatcher.StringSetBuilder
|
|
}
|
|
|
|
func NewPrefixMap(m map[string]map[string]struct{}) *PrefixMap {
|
|
r := PrefixMap{StringSetBuilder: prefixmatcher.NewMatcher[map[string]struct{}]()}
|
|
|
|
for k, v := range m {
|
|
r.Add(k, v)
|
|
}
|
|
|
|
return &r
|
|
}
|
|
|
|
func (pm PrefixMap) AssertEqual(t *testing.T, r prefixmatcher.StringSetReader) {
|
|
if pm.Empty() {
|
|
require.True(t, r.Empty(), "both prefix maps are empty")
|
|
return
|
|
}
|
|
|
|
pks := pm.Keys()
|
|
rks := r.Keys()
|
|
|
|
assert.ElementsMatch(t, pks, rks, "prefix keys match")
|
|
|
|
for _, pk := range pks {
|
|
p, _ := pm.Get(pk)
|
|
r, _ := r.Get(pk)
|
|
assert.Equal(t, p, r, "values match")
|
|
}
|
|
}
|