diff options
Diffstat (limited to 'libgo/go/cmd/go/internal/tlog/tlog_test.go')
-rw-r--r-- | libgo/go/cmd/go/internal/tlog/tlog_test.go | 269 |
1 files changed, 0 insertions, 269 deletions
diff --git a/libgo/go/cmd/go/internal/tlog/tlog_test.go b/libgo/go/cmd/go/internal/tlog/tlog_test.go deleted file mode 100644 index 584e728c1bb..00000000000 --- a/libgo/go/cmd/go/internal/tlog/tlog_test.go +++ /dev/null @@ -1,269 +0,0 @@ -// Copyright 2019 The Go Authors. All rights reserved. -// Use of this source code is governed by a BSD-style -// license that can be found in the LICENSE file. - -package tlog - -import ( - "bytes" - "fmt" - "testing" -) - -type testHashStorage []Hash - -func (t testHashStorage) ReadHash(level int, n int64) (Hash, error) { - return t[StoredHashIndex(level, n)], nil -} - -func (t testHashStorage) ReadHashes(index []int64) ([]Hash, error) { - // It's not required by HashReader that indexes be in increasing order, - // but check that the functions we are testing only ever ask for - // indexes in increasing order. - for i := 1; i < len(index); i++ { - if index[i-1] >= index[i] { - panic("indexes out of order") - } - } - - out := make([]Hash, len(index)) - for i, x := range index { - out[i] = t[x] - } - return out, nil -} - -type testTilesStorage struct { - unsaved int - m map[Tile][]byte -} - -func (t testTilesStorage) Height() int { - return 2 -} - -func (t *testTilesStorage) SaveTiles(tiles []Tile, data [][]byte) { - t.unsaved -= len(tiles) -} - -func (t *testTilesStorage) ReadTiles(tiles []Tile) ([][]byte, error) { - out := make([][]byte, len(tiles)) - for i, tile := range tiles { - out[i] = t.m[tile] - } - t.unsaved += len(tiles) - return out, nil -} - -func TestTree(t *testing.T) { - var trees []Hash - var leafhashes []Hash - var storage testHashStorage - tiles := make(map[Tile][]byte) - const testH = 2 - for i := int64(0); i < 100; i++ { - data := []byte(fmt.Sprintf("leaf %d", i)) - hashes, err := StoredHashes(i, data, storage) - if err != nil { - t.Fatal(err) - } - leafhashes = append(leafhashes, RecordHash(data)) - oldStorage := len(storage) - storage = append(storage, hashes...) - if count := StoredHashCount(i + 1); count != int64(len(storage)) { - t.Errorf("StoredHashCount(%d) = %d, have %d StoredHashes", i+1, count, len(storage)) - } - th, err := TreeHash(i+1, storage) - if err != nil { - t.Fatal(err) - } - - for _, tile := range NewTiles(testH, i, i+1) { - data, err := ReadTileData(tile, storage) - if err != nil { - t.Fatal(err) - } - old := Tile{H: tile.H, L: tile.L, N: tile.N, W: tile.W - 1} - oldData := tiles[old] - if len(oldData) != len(data)-HashSize || !bytes.Equal(oldData, data[:len(oldData)]) { - t.Fatalf("tile %v not extending earlier tile %v", tile.Path(), old.Path()) - } - tiles[tile] = data - } - for _, tile := range NewTiles(testH, 0, i+1) { - data, err := ReadTileData(tile, storage) - if err != nil { - t.Fatal(err) - } - if !bytes.Equal(tiles[tile], data) { - t.Fatalf("mismatch at %+v", tile) - } - } - for _, tile := range NewTiles(testH, i/2, i+1) { - data, err := ReadTileData(tile, storage) - if err != nil { - t.Fatal(err) - } - if !bytes.Equal(tiles[tile], data) { - t.Fatalf("mismatch at %+v", tile) - } - } - - // Check that all the new hashes are readable from their tiles. - for j := oldStorage; j < len(storage); j++ { - tile := TileForIndex(testH, int64(j)) - data, ok := tiles[tile] - if !ok { - t.Log(NewTiles(testH, 0, i+1)) - t.Fatalf("TileForIndex(%d, %d) = %v, not yet stored (i=%d, stored %d)", testH, j, tile.Path(), i, len(storage)) - continue - } - h, err := HashFromTile(tile, data, int64(j)) - if err != nil { - t.Fatal(err) - } - if h != storage[j] { - t.Errorf("HashFromTile(%v, %d) = %v, want %v", tile.Path(), int64(j), h, storage[j]) - } - } - - trees = append(trees, th) - - // Check that leaf proofs work, for all trees and leaves so far. - for j := int64(0); j <= i; j++ { - p, err := ProveRecord(i+1, j, storage) - if err != nil { - t.Fatalf("ProveRecord(%d, %d): %v", i+1, j, err) - } - if err := CheckRecord(p, i+1, th, j, leafhashes[j]); err != nil { - t.Fatalf("CheckRecord(%d, %d): %v", i+1, j, err) - } - for k := range p { - p[k][0] ^= 1 - if err := CheckRecord(p, i+1, th, j, leafhashes[j]); err == nil { - t.Fatalf("CheckRecord(%d, %d) succeeded with corrupt proof hash #%d!", i+1, j, k) - } - p[k][0] ^= 1 - } - } - - // Check that leaf proofs work using TileReader. - // To prove a leaf that way, all you have to do is read and verify its hash. - storage := &testTilesStorage{m: tiles} - thr := TileHashReader(Tree{i + 1, th}, storage) - for j := int64(0); j <= i; j++ { - h, err := thr.ReadHashes([]int64{StoredHashIndex(0, j)}) - if err != nil { - t.Fatalf("TileHashReader(%d).ReadHashes(%d): %v", i+1, j, err) - } - if h[0] != leafhashes[j] { - t.Fatalf("TileHashReader(%d).ReadHashes(%d) returned wrong hash", i+1, j) - } - - // Even though reading the hash suffices, - // check we can generate the proof too. - p, err := ProveRecord(i+1, j, thr) - if err != nil { - t.Fatalf("ProveRecord(%d, %d, TileHashReader(%d)): %v", i+1, j, i+1, err) - } - if err := CheckRecord(p, i+1, th, j, leafhashes[j]); err != nil { - t.Fatalf("CheckRecord(%d, %d, TileHashReader(%d)): %v", i+1, j, i+1, err) - } - } - if storage.unsaved != 0 { - t.Fatalf("TileHashReader(%d) did not save %d tiles", i+1, storage.unsaved) - } - - // Check that ReadHashes will give an error if the index is not in the tree. - if _, err := thr.ReadHashes([]int64{(i + 1) * 2}); err == nil { - t.Fatalf("TileHashReader(%d).ReadHashes(%d) for index not in tree <nil>, want err", i, i+1) - } - if storage.unsaved != 0 { - t.Fatalf("TileHashReader(%d) did not save %d tiles", i+1, storage.unsaved) - } - - // Check that tree proofs work, for all trees so far, using TileReader. - // To prove a tree that way, all you have to do is compute and verify its hash. - for j := int64(0); j <= i; j++ { - h, err := TreeHash(j+1, thr) - if err != nil { - t.Fatalf("TreeHash(%d, TileHashReader(%d)): %v", j, i+1, err) - } - if h != trees[j] { - t.Fatalf("TreeHash(%d, TileHashReader(%d)) = %x, want %x (%v)", j, i+1, h[:], trees[j][:], trees[j]) - } - - // Even though computing the subtree hash suffices, - // check that we can generate the proof too. - p, err := ProveTree(i+1, j+1, thr) - if err != nil { - t.Fatalf("ProveTree(%d, %d): %v", i+1, j+1, err) - } - if err := CheckTree(p, i+1, th, j+1, trees[j]); err != nil { - t.Fatalf("CheckTree(%d, %d): %v [%v]", i+1, j+1, err, p) - } - for k := range p { - p[k][0] ^= 1 - if err := CheckTree(p, i+1, th, j+1, trees[j]); err == nil { - t.Fatalf("CheckTree(%d, %d) succeeded with corrupt proof hash #%d!", i+1, j+1, k) - } - p[k][0] ^= 1 - } - } - if storage.unsaved != 0 { - t.Fatalf("TileHashReader(%d) did not save %d tiles", i+1, storage.unsaved) - } - } -} - -func TestSplitStoredHashIndex(t *testing.T) { - for l := 0; l < 10; l++ { - for n := int64(0); n < 100; n++ { - x := StoredHashIndex(l, n) - l1, n1 := SplitStoredHashIndex(x) - if l1 != l || n1 != n { - t.Fatalf("StoredHashIndex(%d, %d) = %d, but SplitStoredHashIndex(%d) = %d, %d", l, n, x, x, l1, n1) - } - } - } -} - -// TODO(rsc): Test invalid paths too, like "tile/3/5/123/456/078". -var tilePaths = []struct { - path string - tile Tile -}{ - {"tile/4/0/001", Tile{4, 0, 1, 16}}, - {"tile/4/0/001.p/5", Tile{4, 0, 1, 5}}, - {"tile/3/5/x123/x456/078", Tile{3, 5, 123456078, 8}}, - {"tile/3/5/x123/x456/078.p/2", Tile{3, 5, 123456078, 2}}, - {"tile/1/0/x003/x057/500", Tile{1, 0, 3057500, 2}}, - {"tile/3/5/123/456/078", Tile{}}, - {"tile/3/-1/123/456/078", Tile{}}, - {"tile/1/data/x003/x057/500", Tile{1, -1, 3057500, 2}}, -} - -func TestTilePath(t *testing.T) { - for _, tt := range tilePaths { - if tt.tile.H > 0 { - p := tt.tile.Path() - if p != tt.path { - t.Errorf("%+v.Path() = %q, want %q", tt.tile, p, tt.path) - } - } - tile, err := ParseTilePath(tt.path) - if err != nil { - if tt.tile.H == 0 { - // Expected error. - continue - } - t.Errorf("ParseTilePath(%q): %v", tt.path, err) - } else if tile != tt.tile { - if tt.tile.H == 0 { - t.Errorf("ParseTilePath(%q): expected error, got %+v", tt.path, tt.tile) - continue - } - t.Errorf("ParseTilePath(%q) = %+v, want %+v", tt.path, tile, tt.tile) - } - } -} |