Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-06-28 17:02:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

[DEV-14] Fix wire tests to work with new code changes

Browse Source
This commit is contained in:
Mike Zak 2018-06-17 17:29:57 +03:00
parent e6e9deef22
commit f4fc50f5d9
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
wire/bench_test.go
View File

@ -409,7 +409,8 @@ func BenchmarkDecodeGetHeaders(b *testing.B) {
} }
// BenchmarkDecodeHeaders performs a benchmark on how long it takes to // BenchmarkDecodeHeaders performs a benchmark on how long it takes to
// decode a headers message with the maximum number of headers. // decode a headers message with the maximum number of headers and maximum number of
// previous hashes per header.
func BenchmarkDecodeHeaders(b *testing.B) { func BenchmarkDecodeHeaders(b *testing.B) {
// Create a message with the maximum number of headers. // Create a message with the maximum number of headers.
pver := ProtocolVersion pver := ProtocolVersion
@ -419,7 +420,15 @@ func BenchmarkDecodeHeaders(b *testing.B) {
if err != nil { if err != nil {
b.Fatalf("NewHashFromStr: unexpected error: %v", err) b.Fatalf("NewHashFromStr: unexpected error: %v", err)
} }
m.AddBlockHeader(NewBlockHeader(1, hash, hash, 0, uint32(i))) prevBlocks := make([]chainhash.Hash, MaxNumPrevBlocks)
for j := byte(0); j < MaxNumPrevBlocks; j++ {
hash, err := chainhash.NewHashFromStr(fmt.Sprintf("%x%x", i, j))
if err != nil {
b.Fatalf("NewHashFromStr: unexpected error: %v", err)
}
prevBlocks[i] = *hash
}
m.AddBlockHeader(NewBlockHeader(1, prevBlocks, hash, 0, uint32(i)))
} }
// Serialize it so the bytes are available to test the decode below. // Serialize it so the bytes are available to test the decode below.
@ -565,7 +574,7 @@ func BenchmarkDecodeMerkleBlock(b *testing.B) {
if err != nil { if err != nil {
b.Fatalf("NewHashFromStr: unexpected error: %v", err) b.Fatalf("NewHashFromStr: unexpected error: %v", err)
} }
m.Header = *NewBlockHeader(1, hash, hash, 0, uint32(10000)) m.Header = *NewBlockHeader(1, []chainhash.Hash{*hash}, hash, 0, uint32(10000))
for i := 0; i < 105; i++ { for i := 0; i < 105; i++ {
hash, err := chainhash.NewHashFromStr(fmt.Sprintf("%x", i)) hash, err := chainhash.NewHashFromStr(fmt.Sprintf("%x", i))
if err != nil { if err != nil {