func TestQueryCallsLast(t *testing.T) { var d1n, d2n, d3n int d1 := dscb.Wrap(ds.NewMapDatastore(), func() { d1n++ }) d2 := dscb.Wrap(ds.NewMapDatastore(), func() { d2n++ }) d3 := dscb.Wrap(ds.NewMapDatastore(), func() { d3n++ }) td := New(d1, d2, d3) td.Query(dsq.Query{}) if d3n < 1 { t.Error("should call last") } }
func setup() *mock { m := &mock{} mp := ds.NewMapDatastore() ts := dssync.MutexWrap(mp) cb1 := dscb.Wrap(ts, func() { m.Lock() m.inside++ m.Unlock() <-time.After(20 * time.Millisecond) }) cd := Wrap(cb1) cb2 := dscb.Wrap(cd, func() { m.Lock() m.outside++ m.Unlock() }) m.ds = cb2 return m }