github.com/m3db/m3@v1.5.0/src/dbnode/storage/bootstrap/util.go (about)

     1  // Copyright (c) 2019 Uber Technologies, Inc.
     2  //
     3  // Permission is hereby granted, free of charge, to any person obtaining a copy
     4  // of this software and associated documentation files (the "Software"), to deal
     5  // in the Software without restriction, including without limitation the rights
     6  // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
     7  // copies of the Software, and to permit persons to whom the Software is
     8  // furnished to do so, subject to the following conditions:
     9  //
    10  // The above copyright notice and this permission notice shall be included in
    11  // all copies or substantial portions of the Software.
    12  //
    13  // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    14  // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    15  // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    16  // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    17  // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    18  // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
    19  // THE SOFTWARE.
    20  
    21  package bootstrap
    22  
    23  import (
    24  	"bytes"
    25  	"fmt"
    26  	"io"
    27  	"math"
    28  	"sort"
    29  	"sync"
    30  
    31  	"github.com/m3db/m3/src/dbnode/encoding"
    32  	"github.com/m3db/m3/src/dbnode/encoding/m3tsz"
    33  	"github.com/m3db/m3/src/dbnode/namespace"
    34  	"github.com/m3db/m3/src/dbnode/persist/fs"
    35  	"github.com/m3db/m3/src/dbnode/storage/block"
    36  	"github.com/m3db/m3/src/dbnode/storage/bootstrap/result"
    37  	"github.com/m3db/m3/src/dbnode/storage/series"
    38  	"github.com/m3db/m3/src/dbnode/x/xio"
    39  	"github.com/m3db/m3/src/x/context"
    40  	"github.com/m3db/m3/src/x/ident"
    41  	"github.com/m3db/m3/src/x/pool"
    42  	xtest "github.com/m3db/m3/src/x/test"
    43  	xtime "github.com/m3db/m3/src/x/time"
    44  
    45  	"github.com/golang/mock/gomock"
    46  	"github.com/stretchr/testify/assert"
    47  	"github.com/stretchr/testify/require"
    48  )
    49  
    50  // ReadersForID is a slice of readers that share a series ID.
    51  type ReadersForID []ReaderAtTime
    52  
    53  // ReaderMap is a map containing all gathered block segment readers.
    54  type ReaderMap map[string]ReadersForID
    55  
    56  // Must implement NamespaceDataAccumulator.
    57  var _ NamespaceDataAccumulator = (*TestDataAccumulator)(nil)
    58  
    59  // TestDataAccumulator is a NamespaceDataAccumulator that captures any
    60  // series inserts for examination.
    61  type TestDataAccumulator struct {
    62  	sync.Mutex
    63  
    64  	t              require.TestingT
    65  	ctrl           *gomock.Controller
    66  	ns             string
    67  	pool           encoding.MultiReaderIteratorPool
    68  	loadedBlockMap ReaderMap
    69  	schema         namespace.SchemaDescr
    70  	// writeMap is a map to which values are written directly.
    71  	writeMap DecodedBlockMap
    72  	results  map[string]CheckoutSeriesResult
    73  }
    74  
    75  // DecodedValues is a slice of series datapoints.
    76  type DecodedValues []series.DecodedTestValue
    77  
    78  // DecodedBlockMap is a map of decoded datapoints per series ID.
    79  type DecodedBlockMap map[string]DecodedValues
    80  
    81  func testValuesEqual(
    82  	a series.DecodedTestValue,
    83  	b series.DecodedTestValue,
    84  ) bool {
    85  	return a.Timestamp.Equal(b.Timestamp) &&
    86  		math.Abs(a.Value-b.Value) < 0.000001 &&
    87  		a.Unit == b.Unit &&
    88  		bytes.Equal(a.Annotation, b.Annotation)
    89  }
    90  
    91  // VerifyEquals verifies that two DecodedBlockMap are equal; errors otherwise.
    92  func (m DecodedBlockMap) VerifyEquals(other DecodedBlockMap) error {
    93  	if len(m) != len(other) {
    94  		return fmt.Errorf("block maps of length %d and %d do not match",
    95  			len(m), len(other))
    96  	}
    97  
    98  	seen := make(map[string]struct{})
    99  	for k, v := range m {
   100  		otherSeries, found := other[k]
   101  		if !found {
   102  			return fmt.Errorf("series %s: values not found", k)
   103  		}
   104  
   105  		if len(otherSeries) != len(v) {
   106  			return fmt.Errorf("series %s: length of series %d does not match other %d",
   107  				k, len(v), len(otherSeries))
   108  		}
   109  
   110  		// NB: make a clone here to avoid mutating base data
   111  		// just in case any tests care about order.
   112  		thisVal := append([]series.DecodedTestValue(nil), v...)
   113  		otherVal := append([]series.DecodedTestValue(nil), otherSeries...)
   114  
   115  		sort.Sort(series.ValuesByTime(thisVal))
   116  		sort.Sort(series.ValuesByTime(otherVal))
   117  		for i, t := range thisVal {
   118  			o := otherVal[i]
   119  			if !testValuesEqual(t, o) {
   120  				return fmt.Errorf("series %s: value %+v does not match other value %+v",
   121  					k, t, o)
   122  			}
   123  		}
   124  
   125  		seen[k] = struct{}{}
   126  	}
   127  
   128  	for k := range other {
   129  		if _, beenFound := seen[k]; !beenFound {
   130  			return fmt.Errorf("series %s not found in this map", k)
   131  		}
   132  	}
   133  
   134  	return nil
   135  }
   136  
   137  // ReaderAtTime captures incoming block loads, including
   138  // their start times and tags.
   139  type ReaderAtTime struct {
   140  	// Start is the block start time.
   141  	Start xtime.UnixNano
   142  	// Reader is the block segment reader.
   143  	Reader xio.SegmentReader
   144  	// Tags is the list of tags in a basic string map format.
   145  	Tags map[string]string
   146  }
   147  
   148  // dumpLoadedBlocks decodes any accumulated values gathered from calls to
   149  // series.LoadBlock() and returns them as raw values.
   150  func (a *TestDataAccumulator) dumpLoadedBlocks() DecodedBlockMap {
   151  	if len(a.loadedBlockMap) == 0 {
   152  		return nil
   153  	}
   154  
   155  	decodedMap := make(DecodedBlockMap, len(a.loadedBlockMap))
   156  	iter := a.pool.Get()
   157  	defer iter.Close()
   158  	for k, v := range a.loadedBlockMap {
   159  		readers := make([]xio.SegmentReader, 0, len(v))
   160  		for _, r := range v {
   161  			readers = append(readers, r.Reader)
   162  		}
   163  
   164  		value, err := series.DecodeSegmentValues(readers, iter, a.schema)
   165  		if err != nil {
   166  			if err != io.EOF {
   167  				require.NoError(a.t, err)
   168  			}
   169  
   170  			// NB: print out that we encountered EOF here to assist debugging tests,
   171  			// but this is not necessarily a failure.
   172  			fmt.Println("EOF: segment had no values.")
   173  		}
   174  
   175  		sort.Sort(series.ValuesByTime(value))
   176  		decodedMap[k] = value
   177  	}
   178  
   179  	return decodedMap
   180  }
   181  
   182  // CheckoutSeriesWithLock will retrieve a series for writing to,
   183  // and when the accumulator is closed, it will ensure that the
   184  // series is released (with lock).
   185  func (a *TestDataAccumulator) CheckoutSeriesWithLock(
   186  	shardID uint32,
   187  	id ident.ID,
   188  	tags ident.TagIterator,
   189  ) (CheckoutSeriesResult, bool, error) {
   190  	a.Lock()
   191  	defer a.Unlock()
   192  	return a.checkoutSeriesWithLock(shardID, id, tags)
   193  }
   194  
   195  // CheckoutSeriesWithoutLock will retrieve a series for writing to,
   196  // and when the accumulator is closed, it will ensure that the
   197  // series is released (without lock).
   198  func (a *TestDataAccumulator) CheckoutSeriesWithoutLock(
   199  	shardID uint32,
   200  	id ident.ID,
   201  	tags ident.TagIterator,
   202  ) (CheckoutSeriesResult, bool, error) {
   203  	return a.checkoutSeriesWithLock(shardID, id, tags)
   204  }
   205  
   206  func (a *TestDataAccumulator) checkoutSeriesWithLock(
   207  	shardID uint32,
   208  	id ident.ID,
   209  	tags ident.TagIterator,
   210  ) (CheckoutSeriesResult, bool, error) {
   211  	var decodedTags map[string]string
   212  	if tags != nil {
   213  		decodedTags = make(map[string]string, tags.Len())
   214  		for tags.Next() {
   215  			tag := tags.Current()
   216  			name := tag.Name.String()
   217  			value := tag.Value.String()
   218  			if len(name) > 0 && len(value) > 0 {
   219  				decodedTags[name] = value
   220  			}
   221  		}
   222  
   223  		if err := tags.Err(); err != nil {
   224  			return CheckoutSeriesResult{}, false, err
   225  		}
   226  	} else {
   227  		// Ensure the decoded tags aren't nil.
   228  		decodedTags = make(map[string]string)
   229  	}
   230  
   231  	stringID := id.String()
   232  	if result, found := a.results[stringID]; found {
   233  		return result, true, nil
   234  	}
   235  
   236  	var streamErr error
   237  	mockSeries := series.NewMockDatabaseSeries(a.ctrl)
   238  
   239  	mockSeries.EXPECT().
   240  		LoadBlock(gomock.Any(), gomock.Any()).
   241  		DoAndReturn(func(bl block.DatabaseBlock, _ series.WriteType) error {
   242  			a.Lock()
   243  			defer a.Unlock()
   244  
   245  			reader, err := bl.Stream(context.NewBackground())
   246  			if err != nil {
   247  				streamErr = err
   248  				return err
   249  			}
   250  
   251  			a.loadedBlockMap[stringID] = append(a.loadedBlockMap[stringID],
   252  				ReaderAtTime{
   253  					Start:  bl.StartTime(),
   254  					Reader: reader,
   255  					Tags:   decodedTags,
   256  				})
   257  
   258  			return nil
   259  		}).AnyTimes()
   260  
   261  	mockSeries.EXPECT().Write(
   262  		gomock.Any(), gomock.Any(), gomock.Any(),
   263  		gomock.Any(), gomock.Any(), gomock.Any()).
   264  		DoAndReturn(
   265  			func(
   266  				_ context.Context,
   267  				ts xtime.UnixNano,
   268  				val float64,
   269  				unit xtime.Unit,
   270  				annotation []byte,
   271  				_ series.WriteOptions,
   272  			) (bool, series.WriteType, error) {
   273  				a.Lock()
   274  				a.writeMap[stringID] = append(
   275  					a.writeMap[stringID], series.DecodedTestValue{
   276  						Timestamp:  ts,
   277  						Value:      val,
   278  						Unit:       unit,
   279  						Annotation: annotation,
   280  					})
   281  				a.Unlock()
   282  				return true, series.WarmWrite, nil
   283  			}).AnyTimes()
   284  
   285  	result := CheckoutSeriesResult{
   286  		Shard:    shardID,
   287  		Resolver: &seriesStaticResolver{series: mockSeries},
   288  	}
   289  
   290  	a.results[stringID] = result
   291  	return result, true, streamErr
   292  }
   293  
   294  var _ SeriesRefResolver = (*seriesStaticResolver)(nil)
   295  
   296  type seriesStaticResolver struct {
   297  	series SeriesRef
   298  }
   299  
   300  func (r *seriesStaticResolver) SeriesRef() (SeriesRef, error) {
   301  	return r.series, nil
   302  }
   303  
   304  func (r *seriesStaticResolver) ReleaseRef() {}
   305  
   306  // Release is a no-op on the test accumulator.
   307  func (a *TestDataAccumulator) Release() {}
   308  
   309  // Close is a no-op on the test accumulator.
   310  func (a *TestDataAccumulator) Close() error { return nil }
   311  
   312  // NamespacesTester is a utility to assist testing bootstrapping.
   313  type NamespacesTester struct {
   314  	t    require.TestingT
   315  	ctrl *gomock.Controller
   316  	pool encoding.MultiReaderIteratorPool
   317  
   318  	// Accumulators are the accumulators which incoming blocks get loaded into.
   319  	// One per namespace.
   320  	Accumulators []*TestDataAccumulator
   321  
   322  	// Namespaces are the namespaces for this tester.
   323  	Namespaces Namespaces
   324  	// Cache is a snapshot of data useful during bootstrapping.
   325  	Cache Cache
   326  	// Results are the namespace results after bootstrapping.
   327  	Results NamespaceResults
   328  }
   329  
   330  func buildDefaultIterPool() encoding.MultiReaderIteratorPool {
   331  	iterPool := encoding.NewMultiReaderIteratorPool(pool.NewObjectPoolOptions())
   332  	iterPool.Init(m3tsz.DefaultReaderIteratorAllocFn(encoding.NewOptions()))
   333  	return iterPool
   334  }
   335  
   336  // BuildNamespacesTester builds a NamespacesTester.
   337  func BuildNamespacesTester(
   338  	t require.TestingT,
   339  	runOpts RunOptions,
   340  	ranges result.ShardTimeRanges,
   341  	mds ...namespace.Metadata,
   342  ) NamespacesTester {
   343  	return BuildNamespacesTesterWithReaderIteratorPool(
   344  		t,
   345  		runOpts,
   346  		ranges,
   347  		nil,
   348  		fs.NewOptions(),
   349  		mds...,
   350  	)
   351  }
   352  
   353  // BuildNamespacesTesterWithFilesystemOptions builds a NamespacesTester with fs.Options
   354  func BuildNamespacesTesterWithFilesystemOptions(
   355  	t require.TestingT,
   356  	runOpts RunOptions,
   357  	ranges result.ShardTimeRanges,
   358  	fsOpts fs.Options,
   359  	mds ...namespace.Metadata,
   360  ) NamespacesTester {
   361  	return BuildNamespacesTesterWithReaderIteratorPool(
   362  		t,
   363  		runOpts,
   364  		ranges,
   365  		nil,
   366  		fsOpts,
   367  		mds...,
   368  	)
   369  }
   370  
   371  // BuildNamespacesTesterWithReaderIteratorPool builds a NamespacesTester with a
   372  // given MultiReaderIteratorPool.
   373  func BuildNamespacesTesterWithReaderIteratorPool(
   374  	t require.TestingT,
   375  	runOpts RunOptions,
   376  	ranges result.ShardTimeRanges,
   377  	iterPool encoding.MultiReaderIteratorPool,
   378  	fsOpts fs.Options,
   379  	mds ...namespace.Metadata,
   380  ) NamespacesTester {
   381  	shards := make([]uint32, 0, ranges.Len())
   382  	for shard := range ranges.Iter() {
   383  		shards = append(shards, shard)
   384  	}
   385  
   386  	if iterPool == nil {
   387  		iterPool = buildDefaultIterPool()
   388  	}
   389  
   390  	ctrl := xtest.NewController(t)
   391  	namespacesMap := NewNamespacesMap(NamespacesMapOptions{})
   392  	accumulators := make([]*TestDataAccumulator, 0, len(mds))
   393  	finders := make([]NamespaceDetails, 0, len(mds))
   394  	for _, md := range mds {
   395  		nsCtx := namespace.NewContextFrom(md)
   396  		acc := &TestDataAccumulator{
   397  			t:              t,
   398  			ctrl:           ctrl,
   399  			pool:           iterPool,
   400  			ns:             md.ID().String(),
   401  			results:        make(map[string]CheckoutSeriesResult),
   402  			loadedBlockMap: make(ReaderMap),
   403  			writeMap:       make(DecodedBlockMap),
   404  			schema:         nsCtx.Schema,
   405  		}
   406  
   407  		accumulators = append(accumulators, acc)
   408  		namespacesMap.Set(md.ID(), Namespace{
   409  			Metadata:        md,
   410  			Shards:          shards,
   411  			DataAccumulator: acc,
   412  			DataRunOptions: NamespaceRunOptions{
   413  				ShardTimeRanges:       ranges.Copy(),
   414  				TargetShardTimeRanges: ranges.Copy(),
   415  				RunOptions:            runOpts,
   416  			},
   417  			IndexRunOptions: NamespaceRunOptions{
   418  				ShardTimeRanges:       ranges.Copy(),
   419  				TargetShardTimeRanges: ranges.Copy(),
   420  				RunOptions:            runOpts,
   421  			},
   422  		})
   423  		finders = append(finders, NamespaceDetails{
   424  			Namespace: md,
   425  			Shards:    shards,
   426  		})
   427  	}
   428  	cache, err := NewCache(NewCacheOptions().
   429  		SetFilesystemOptions(fsOpts).
   430  		SetInstrumentOptions(fsOpts.InstrumentOptions()).
   431  		SetNamespaceDetails(finders))
   432  	require.NoError(t, err)
   433  
   434  	return NamespacesTester{
   435  		t:            t,
   436  		ctrl:         ctrl,
   437  		pool:         iterPool,
   438  		Accumulators: accumulators,
   439  		Cache:        cache,
   440  		Namespaces: Namespaces{
   441  			Namespaces: namespacesMap,
   442  		},
   443  	}
   444  }
   445  
   446  // DecodedNamespaceMap is a map of decoded blocks per namespace ID.
   447  type DecodedNamespaceMap map[string]DecodedBlockMap
   448  
   449  // DumpLoadedBlocks dumps any loaded blocks as decoded series per namespace.
   450  func (nt *NamespacesTester) DumpLoadedBlocks() DecodedNamespaceMap {
   451  	nsMap := make(DecodedNamespaceMap, len(nt.Accumulators))
   452  	for _, acc := range nt.Accumulators {
   453  		block := acc.dumpLoadedBlocks()
   454  
   455  		if block != nil {
   456  			nsMap[acc.ns] = block
   457  		}
   458  	}
   459  
   460  	return nsMap
   461  }
   462  
   463  // EnsureDumpLoadedBlocksForNamespace dumps all loaded blocks as decoded series,
   464  // and fails if the namespace is not found.
   465  func (nt *NamespacesTester) EnsureDumpLoadedBlocksForNamespace(
   466  	md namespace.Metadata,
   467  ) DecodedBlockMap {
   468  	id := md.ID().String()
   469  	for _, acc := range nt.Accumulators {
   470  		if acc.ns == id {
   471  			return acc.dumpLoadedBlocks()
   472  		}
   473  	}
   474  
   475  	assert.FailNow(nt.t, fmt.Sprintf("namespace with id %s not found "+
   476  		"valid namespaces are %v", id, nt.Namespaces))
   477  	return nil
   478  }
   479  
   480  // EnsureNoLoadedBlocks ensures that no blocks have been loaded into any of this
   481  // testers accumulators.
   482  func (nt *NamespacesTester) EnsureNoLoadedBlocks() {
   483  	require.Equal(nt.t, 0, len(nt.DumpLoadedBlocks()))
   484  }
   485  
   486  // DumpWrites dumps the writes encountered for all namespaces.
   487  func (nt *NamespacesTester) DumpWrites() DecodedNamespaceMap {
   488  	nsMap := make(DecodedNamespaceMap, len(nt.Accumulators))
   489  	for _, acc := range nt.Accumulators {
   490  		if len(acc.writeMap) > 0 {
   491  			nsMap[acc.ns] = acc.writeMap
   492  		}
   493  	}
   494  
   495  	return nsMap
   496  }
   497  
   498  // EnsureDumpWritesForNamespace dumps the writes encountered for the
   499  // given namespace, and fails if the namespace is not found.
   500  func (nt *NamespacesTester) EnsureDumpWritesForNamespace(
   501  	md namespace.Metadata,
   502  ) DecodedBlockMap {
   503  	id := md.ID().String()
   504  	for _, acc := range nt.Accumulators {
   505  		if acc.ns == id {
   506  			return acc.writeMap
   507  		}
   508  	}
   509  
   510  	assert.FailNow(nt.t, fmt.Sprintf("namespace with id %s not found "+
   511  		"valid namespaces are %v", id, nt.Namespaces))
   512  	return nil
   513  }
   514  
   515  // EnsureNoWrites ensures that no writes have been written into any of this
   516  // testers accumulators.
   517  func (nt *NamespacesTester) EnsureNoWrites() {
   518  	require.Equal(nt.t, 0, len(nt.DumpWrites()))
   519  }
   520  
   521  // EnsureDumpAllForNamespace dumps all results for a single namespace, and
   522  // fails if the namespace is not found. The results are unsorted; if sorted
   523  // order is important for verification, they should be sorted afterwards.
   524  func (nt *NamespacesTester) EnsureDumpAllForNamespace(
   525  	md namespace.Metadata,
   526  ) (DecodedBlockMap, error) {
   527  	id := md.ID().String()
   528  	for _, acc := range nt.Accumulators {
   529  		if acc.ns != id {
   530  			continue
   531  		}
   532  
   533  		writeMap := acc.writeMap
   534  		loadedBlockMap := acc.dumpLoadedBlocks()
   535  		merged := make(DecodedBlockMap, len(writeMap)+len(loadedBlockMap))
   536  		for k, v := range writeMap {
   537  			merged[k] = v
   538  		}
   539  
   540  		for k, v := range loadedBlockMap {
   541  			if vals, found := merged[k]; found {
   542  				merged[k] = append(vals, v...)
   543  			} else {
   544  				merged[k] = v
   545  			}
   546  		}
   547  
   548  		return merged, nil
   549  	}
   550  
   551  	return nil, fmt.Errorf("namespace with id %s not found "+
   552  		"valid namespaces are %v", id, nt.Namespaces)
   553  }
   554  
   555  // EnsureDumpReadersForNamespace dumps the readers and their start times for a
   556  // given namespace, and fails if the namespace is not found.
   557  func (nt *NamespacesTester) EnsureDumpReadersForNamespace(
   558  	md namespace.Metadata,
   559  ) ReaderMap {
   560  	id := md.ID().String()
   561  	for _, acc := range nt.Accumulators {
   562  		if acc.ns == id {
   563  			return acc.loadedBlockMap
   564  		}
   565  	}
   566  
   567  	assert.FailNow(nt.t, fmt.Sprintf("namespace with id %s not found "+
   568  		"valid namespaces are %v", id, nt.Namespaces))
   569  	return nil
   570  }
   571  
   572  // ResultForNamespace gives the result for the given namespace, and fails if
   573  // the namespace is not found.
   574  func (nt *NamespacesTester) ResultForNamespace(id ident.ID) NamespaceResult {
   575  	result, found := nt.Results.Results.Get(id)
   576  	require.True(nt.t, found)
   577  	return result
   578  }
   579  
   580  // TestBootstrapWith bootstraps the current Namespaces with the
   581  // provided bootstrapper.
   582  func (nt *NamespacesTester) TestBootstrapWith(b Bootstrapper) {
   583  	ctx := context.NewBackground()
   584  	defer ctx.Close()
   585  	res, err := b.Bootstrap(ctx, nt.Namespaces, nt.Cache)
   586  	assert.NoError(nt.t, err)
   587  	nt.Results = res
   588  }
   589  
   590  // TestReadWith reads the current Namespaces with the
   591  // provided bootstrap source.
   592  func (nt *NamespacesTester) TestReadWith(s Source) {
   593  	ctx := context.NewBackground()
   594  	defer ctx.Close()
   595  	res, err := s.Read(ctx, nt.Namespaces, nt.Cache)
   596  	require.NoError(nt.t, err)
   597  	nt.Results = res
   598  }
   599  
   600  func validateRanges(ac xtime.Ranges, ex xtime.Ranges) error {
   601  	// Make range eclipses expected.
   602  	removedRange := ex.Clone()
   603  	removedRange.RemoveRanges(ac)
   604  	if !removedRange.IsEmpty() {
   605  		return fmt.Errorf("actual range %v does not match expected range %v "+
   606  			"diff: %v", ac, ex, removedRange)
   607  	}
   608  
   609  	// Now make sure no ranges outside of expected.
   610  	expectedWithAddedRanges := ex.Clone()
   611  	expectedWithAddedRanges.AddRanges(ac)
   612  	if ex.Len() != expectedWithAddedRanges.Len() {
   613  		return fmt.Errorf("expected with re-added ranges not equal")
   614  	}
   615  
   616  	iter := ex.Iter()
   617  	withAddedRangesIter := expectedWithAddedRanges.Iter()
   618  	for iter.Next() && withAddedRangesIter.Next() {
   619  		if !iter.Value().Equal(withAddedRangesIter.Value()) {
   620  			return fmt.Errorf("actual range %v does not match expected range %v",
   621  				ac, ex)
   622  		}
   623  	}
   624  
   625  	return nil
   626  }
   627  
   628  func validateShardTimeRanges(
   629  	r result.ShardTimeRanges,
   630  	ex result.ShardTimeRanges,
   631  ) error {
   632  	if ex.Len() != r.Len() {
   633  		return fmt.Errorf("expected %v and actual %v size mismatch", ex, r)
   634  	}
   635  
   636  	seen := make(map[uint32]struct{}, r.Len())
   637  	for k, val := range r.Iter() {
   638  		expectedVal, ok := ex.Get(k)
   639  		if !ok {
   640  			return fmt.Errorf("expected shard map %v does not have shard %d; "+
   641  				"actual: %v", ex, k, r)
   642  		}
   643  
   644  		if err := validateRanges(val, expectedVal); err != nil {
   645  			return err
   646  		}
   647  
   648  		seen[k] = struct{}{}
   649  	}
   650  
   651  	for k := range ex.Iter() {
   652  		if _, beenFound := seen[k]; !beenFound {
   653  			return fmt.Errorf("shard %d in actual not found in expected %v", k, ex)
   654  		}
   655  	}
   656  
   657  	return nil
   658  }
   659  
   660  // TestUnfulfilledForNamespace ensures the given namespace has the expected
   661  // range flagged as unfulfilled.
   662  func (nt *NamespacesTester) TestUnfulfilledForNamespace(
   663  	md namespace.Metadata,
   664  	ex result.ShardTimeRanges,
   665  	exIdx result.ShardTimeRanges,
   666  ) {
   667  	ns := nt.ResultForNamespace(md.ID())
   668  	actual := ns.DataResult.Unfulfilled()
   669  	require.NoError(nt.t, validateShardTimeRanges(actual, ex), "data")
   670  
   671  	if md.Options().IndexOptions().Enabled() {
   672  		actual := ns.IndexResult.Unfulfilled()
   673  		require.NoError(nt.t, validateShardTimeRanges(actual, exIdx), "index")
   674  	}
   675  }
   676  
   677  // TestUnfulfilledForNamespaceIsEmpty ensures the given namespace has an empty
   678  // unfulfilled range.
   679  func (nt *NamespacesTester) TestUnfulfilledForNamespaceIsEmpty(
   680  	md namespace.Metadata,
   681  ) {
   682  	nt.TestUnfulfilledForIDIsEmpty(md.ID(), md.Options().IndexOptions().Enabled())
   683  }
   684  
   685  // TestUnfulfilledForIDIsEmpty ensures the given id has an empty
   686  // unfulfilled range.
   687  func (nt *NamespacesTester) TestUnfulfilledForIDIsEmpty(
   688  	id ident.ID,
   689  	useIndex bool,
   690  ) {
   691  	ns := nt.ResultForNamespace(id)
   692  	actual := ns.DataResult.Unfulfilled()
   693  	assert.True(nt.t, actual.IsEmpty(), fmt.Sprintf("data: not empty %v", actual))
   694  
   695  	if useIndex {
   696  		actual := ns.DataResult.Unfulfilled()
   697  		assert.True(nt.t, actual.IsEmpty(),
   698  			fmt.Sprintf("index: not empty %v", actual))
   699  	}
   700  }
   701  
   702  // Finish closes the namespaceTester and tests mocks for completion.
   703  func (nt *NamespacesTester) Finish() {
   704  	nt.ctrl.Finish()
   705  }
   706  
   707  // NamespaceMatcher is a matcher for namespaces.
   708  type NamespaceMatcher struct {
   709  	// Namespaces are the expected namespaces.
   710  	Namespaces Namespaces
   711  }
   712  
   713  // String describes what the matcher matches.
   714  func (m NamespaceMatcher) String() string { return "namespace query" }
   715  
   716  // Matches returns whether x is a match.
   717  func (m NamespaceMatcher) Matches(x interface{}) bool {
   718  	ns, ok := x.(Namespaces)
   719  	if !ok {
   720  		return false
   721  	}
   722  
   723  	equalRange := func(a, b TargetRange) bool {
   724  		return a.Range.Start.Equal(b.Range.Start) &&
   725  			a.Range.End.Equal(b.Range.End)
   726  	}
   727  
   728  	for _, v := range ns.Namespaces.Iter() {
   729  		other, found := m.Namespaces.Namespaces.Get(v.Key())
   730  		if !found {
   731  			return false
   732  		}
   733  
   734  		val := v.Value()
   735  		if !other.Metadata.Equal(val.Metadata) {
   736  			return false
   737  		}
   738  
   739  		if !equalRange(val.DataTargetRange, other.DataTargetRange) {
   740  			return false
   741  		}
   742  
   743  		if !equalRange(val.IndexTargetRange, other.IndexTargetRange) {
   744  			return false
   745  		}
   746  	}
   747  
   748  	return true
   749  }
   750  
   751  // NB: assert NamespaceMatcher is a gomock.Matcher
   752  var _ gomock.Matcher = (*NamespaceMatcher)(nil)
   753  
   754  // ShardTimeRangesMatcher is a matcher for ShardTimeRanges.
   755  type ShardTimeRangesMatcher struct {
   756  	// Ranges are the expected ranges.
   757  	Ranges result.ShardTimeRanges
   758  }
   759  
   760  // Matches returns whether x is a match.
   761  func (m ShardTimeRangesMatcher) Matches(x interface{}) bool {
   762  	actual, ok := x.(result.ShardTimeRanges)
   763  	if !ok {
   764  		return false
   765  	}
   766  
   767  	if err := validateShardTimeRanges(m.Ranges, actual); err != nil {
   768  		fmt.Println("shard time ranges do not match:", err.Error())
   769  		return false
   770  	}
   771  
   772  	return true
   773  }
   774  
   775  // String describes what the matcher matches.
   776  func (m ShardTimeRangesMatcher) String() string {
   777  	return "shardTimeRangesMatcher"
   778  }
   779  
   780  // NB: assert ShardTimeRangesMatcher is a gomock.Matcher
   781  var _ gomock.Matcher = (*ShardTimeRangesMatcher)(nil)