From f01354b5bd70e5d9c3d2d80f72790f0ddd076740 Mon Sep 17 00:00:00 2001
From: Roman Khimov <roman@nspcc.ru>
Date: Mon, 16 Sep 2019 12:07:06 +0300
Subject: [PATCH] core: make TestHeaderEncodeDecode use assert for field
 comparisons

---
 pkg/core/header_test.go | 33 +++++++++------------------------
 1 file changed, 9 insertions(+), 24 deletions(-)

diff --git a/pkg/core/header_test.go b/pkg/core/header_test.go
index ef7dcbec0..dcd83d7c0 100644
--- a/pkg/core/header_test.go
+++ b/pkg/core/header_test.go
@@ -8,6 +8,7 @@ import (
 	"github.com/CityOfZion/neo-go/pkg/core/transaction"
 	"github.com/CityOfZion/neo-go/pkg/crypto/hash"
 	"github.com/CityOfZion/neo-go/pkg/util"
+	"github.com/stretchr/testify/assert"
 )
 
 func TestHeaderEncodeDecode(t *testing.T) {
@@ -34,28 +35,12 @@ func TestHeaderEncodeDecode(t *testing.T) {
 	if err := headerDecode.DecodeBinary(buf); err != nil {
 		t.Fatal(err)
 	}
-	if header.Version != headerDecode.Version {
-		t.Fatal("expected both versions to be equal")
-	}
-	if !header.PrevHash.Equals(headerDecode.PrevHash) {
-		t.Fatal("expected both prev hashes to be equal")
-	}
-	if !header.MerkleRoot.Equals(headerDecode.MerkleRoot) {
-		t.Fatal("expected both merkle roots to be equal")
-	}
-	if header.Index != headerDecode.Index {
-		t.Fatal("expected both indexes to be equal")
-	}
-	if header.ConsensusData != headerDecode.ConsensusData {
-		t.Fatal("expected both consensus data fields to be equal")
-	}
-	if !header.NextConsensus.Equals(headerDecode.NextConsensus) {
-		t.Fatalf("expected both next consensus fields to be equal")
-	}
-	if !bytes.Equal(header.Script.InvocationScript, headerDecode.Script.InvocationScript) {
-		t.Fatalf("expected equal invocation scripts %v and %v", header.Script.InvocationScript, headerDecode.Script.InvocationScript)
-	}
-	if !bytes.Equal(header.Script.VerificationScript, headerDecode.Script.VerificationScript) {
-		t.Fatalf("expected equal verification scripts %v and %v", header.Script.VerificationScript, headerDecode.Script.VerificationScript)
-	}
+	assert.Equal(t, header.Version, headerDecode.Version, "expected both versions to be equal")
+	assert.Equal(t, header.PrevHash, headerDecode.PrevHash, "expected both prev hashes to be equal")
+	assert.Equal(t, header.MerkleRoot, headerDecode.MerkleRoot, "expected both merkle roots to be equal")
+	assert.Equal(t, header.Index, headerDecode.Index, "expected both indexes to be equal")
+	assert.Equal(t, header.ConsensusData, headerDecode.ConsensusData, "expected both consensus data fields to be equal")
+	assert.Equal(t, header.NextConsensus, headerDecode.NextConsensus, "expected both next consensus fields to be equal")
+	assert.Equal(t, header.Script.InvocationScript, headerDecode.Script.InvocationScript, "expected equal invocation scripts")
+	assert.Equal(t, header.Script.VerificationScript, headerDecode.Script.VerificationScript, "expected equal verification scripts")
 }