diff --git a/go.sum b/go.sum index 8d4fd7957c6db9f37113168100a8687712a7fb27..d1289721f2c72f9fbe4255542e01d698ae2b53c1 100644 --- a/go.sum +++ b/go.sum @@ -6,6 +6,7 @@ github.com/jbenet/goprocess v0.0.0-20160826012719-b497e2f366b8 h1:bspPhN+oKYFk5f github.com/jbenet/goprocess v0.0.0-20160826012719-b497e2f366b8/go.mod h1:Ly/wlsjFq/qrU3Rar62tu1gASgGw6chQbSh/XgIIXCY= github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI= github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= +github.com/kr/pty v1.1.1 h1:VkoXIwSboBpnk99O/KFauAEILuNHv5DVFKZMBN/gUgw= github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE= github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= diff --git a/mount/mount_test.go b/mount/mount_test.go index 7d88b0a75a04d77a63e0b893e3fab84851695850..a3059b46eed5e38f3588e4dcac04ba4b0cd9c1e6 100644 --- a/mount/mount_test.go +++ b/mount/mount_test.go @@ -841,9 +841,9 @@ func TestSuite(t *testing.T) { mapds2 := datastore.NewMapDatastore() mapds3 := datastore.NewMapDatastore() m := mount.New([]mount.Mount{ - {Prefix: datastore.NewKey("/foo"), Datastore: mapds1}, - {Prefix: datastore.NewKey("/bar"), Datastore: mapds2}, - {Prefix: datastore.NewKey("/baz"), Datastore: mapds3}, + {Prefix: datastore.NewKey("/prefix"), Datastore: mapds1}, + {Prefix: datastore.NewKey("/prefix/sub"), Datastore: mapds2}, + {Prefix: datastore.NewKey("/0"), Datastore: mapds3}, {Prefix: datastore.NewKey("/"), Datastore: mapds0}, }) dstest.SubtestAll(t, m) diff --git a/test/basic_tests.go b/test/basic_tests.go index 0506ee531c3d6d4931ce5c66aac3857d9aad3e66..6b8d10dba7b9c382af493345dfe29bc14a9d9a6c 100644 --- a/test/basic_tests.go +++ b/test/basic_tests.go @@ -177,27 +177,27 @@ func SubtestManyKeysAndQuery(t *testing.T, ds dstore.Datastore) { } func SubtestBasicSync(t *testing.T, ds dstore.Datastore) { - if err := ds.Sync(dstore.NewKey("foo")); err != nil { + if err := ds.Sync(dstore.NewKey("prefix")); err != nil { t.Fatal(err) } - if err := ds.Put(dstore.NewKey("/foo"), []byte("foo")); err != nil { + if err := ds.Put(dstore.NewKey("/prefix"), []byte("foo")); err != nil { t.Fatal(err) } - if err := ds.Sync(dstore.NewKey("/foo")); err != nil { + if err := ds.Sync(dstore.NewKey("/prefix")); err != nil { t.Fatal(err) } - if err := ds.Put(dstore.NewKey("/foo/bar"), []byte("bar")); err != nil { + if err := ds.Put(dstore.NewKey("/prefix/sub"), []byte("bar")); err != nil { t.Fatal(err) } - if err := ds.Sync(dstore.NewKey("/foo")); err != nil { + if err := ds.Sync(dstore.NewKey("/prefix")); err != nil { t.Fatal(err) } - if err := ds.Sync(dstore.NewKey("/foo/bar")); err != nil { + if err := ds.Sync(dstore.NewKey("/prefix/sub")); err != nil { t.Fatal(err) } @@ -237,6 +237,11 @@ func SubtestCombinations(t *testing.T, ds dstore.Datastore) { Key: "/2", }}, } + prefixes := []string{ + "", + "/prefix", + "/0", // keys exist under this prefix but they shouldn't match. + } orders := [][]dsq.Order{ {dsq.OrderByKey{}}, {dsq.OrderByKeyDescending{}}, @@ -255,8 +260,9 @@ func SubtestCombinations(t *testing.T, ds dstore.Datastore) { Limit: limits[perm[1]], Filters: filters[perm[2]], Orders: orders[perm[3]], + Prefix: prefixes[perm[4]], } - length := lengths[perm[4]] + length := lengths[perm[5]] t.Run(strings.ReplaceAll(fmt.Sprintf("%d/{%s}", length, q), " ", "ยท"), func(t *testing.T) { subtestQuery(t, ds, q, length) @@ -266,6 +272,7 @@ func SubtestCombinations(t *testing.T, ds dstore.Datastore) { len(limits), len(filters), len(orders), + len(prefixes), len(lengths), ) } @@ -335,6 +342,28 @@ func SubtestReturnSizes(t *testing.T, ds dstore.Datastore) { subtestQuery(t, ds, dsq.Query{ReturnsSizes: true}, 100) } +func SubtestPrefix(t *testing.T, ds dstore.Datastore) { + test := func(prefix string) { + t.Run(prefix, func(t *testing.T) { + subtestQuery(t, ds, dsq.Query{ + Prefix: prefix, + }, 100) + }) + } + test("") + test("/") + test("/./") + test("/.././/") + test("/prefix/../") + + test("/prefix") + test("/prefix/") + test("/prefix/sub/") + + test("/0/") + test("/bad/") +} + func randValue() []byte { value := make([]byte, 64) rand.Read(value) @@ -354,6 +383,28 @@ func subtestQuery(t *testing.T, ds dstore.Datastore, q dsq.Query, count int) { }) } + for i := 0; i < count; i++ { + s := fmt.Sprintf("/prefix/%dkey%d", i, i) + key := dstore.NewKey(s).String() + value := randValue() + input = append(input, dsq.Entry{ + Key: key, + Size: len(value), + Value: value, + }) + } + + for i := 0; i < count; i++ { + s := fmt.Sprintf("/prefix/sub/%dkey%d", i, i) + key := dstore.NewKey(s).String() + value := randValue() + input = append(input, dsq.Entry{ + Key: key, + Size: len(value), + Value: value, + }) + } + t.Logf("putting %d values", count) for i, e := range input { err := ds.Put(dstore.RawKey(e.Key), e.Value) diff --git a/test/suite.go b/test/suite.go index afebd1ca813e5cb46c5fd2439e08d0d2f8eef932..012f80217ab86cc5ac6ce1b9fe50e27f9b6f496b 100644 --- a/test/suite.go +++ b/test/suite.go @@ -14,6 +14,7 @@ var BasicSubtests = []func(t *testing.T, ds dstore.Datastore){ SubtestBasicPutGet, SubtestNotFounds, SubtestCombinations, + SubtestPrefix, SubtestOrder, SubtestLimit, SubtestFilter,