Skip to content

Commit

Permalink
common: improve test (ethereum#20354)
Browse files Browse the repository at this point in the history
  • Loading branch information
gzliudan committed Dec 26, 2024
1 parent a9a9a7c commit b201195
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 50 deletions.
52 changes: 27 additions & 25 deletions common/bytes_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -19,41 +19,43 @@ package common
import (
"bytes"
"testing"

checker "gopkg.in/check.v1"
)

type BytesSuite struct{}

var _ = checker.Suite(&BytesSuite{})
func TestCopyBytes(t *testing.T) {
input := []byte{1, 2, 3, 4}

func (s *BytesSuite) TestCopyBytes(c *checker.C) {
data1 := []byte{1, 2, 3, 4}
exp1 := []byte{1, 2, 3, 4}
res1 := CopyBytes(data1)
c.Assert(res1, checker.DeepEquals, exp1)
v := CopyBytes(input)
if !bytes.Equal(v, []byte{1, 2, 3, 4}) {
t.Fatal("not equal after copy")
}
v[0] = 99
if bytes.Equal(v, input) {
t.Fatal("result is not a copy")
}
}

func (s *BytesSuite) TestLeftPadBytes(c *checker.C) {
val1 := []byte{1, 2, 3, 4}
exp1 := []byte{0, 0, 0, 0, 1, 2, 3, 4}

res1 := LeftPadBytes(val1, 8)
res2 := LeftPadBytes(val1, 2)
func TestLeftPadBytes(t *testing.T) {
val := []byte{1, 2, 3, 4}
padded := []byte{0, 0, 0, 0, 1, 2, 3, 4}

c.Assert(res1, checker.DeepEquals, exp1)
c.Assert(res2, checker.DeepEquals, val1)
if r := LeftPadBytes(val, 8); !bytes.Equal(r, padded) {
t.Fatalf("LeftPadBytes(%v, 8) == %v", val, r)
}
if r := LeftPadBytes(val, 2); !bytes.Equal(r, val) {
t.Fatalf("LeftPadBytes(%v, 2) == %v", val, r)
}
}

func (s *BytesSuite) TestRightPadBytes(c *checker.C) {
func TestRightPadBytes(t *testing.T) {
val := []byte{1, 2, 3, 4}
exp := []byte{1, 2, 3, 4, 0, 0, 0, 0}

resstd := RightPadBytes(val, 8)
resshrt := RightPadBytes(val, 2)
padded := []byte{1, 2, 3, 4, 0, 0, 0, 0}

c.Assert(resstd, checker.DeepEquals, exp)
c.Assert(resshrt, checker.DeepEquals, val)
if r := RightPadBytes(val, 8); !bytes.Equal(r, padded) {
t.Fatalf("RightPadBytes(%v, 8) == %v", val, r)
}
if r := RightPadBytes(val, 2); !bytes.Equal(r, val) {
t.Fatalf("RightPadBytes(%v, 2) == %v", val, r)
}
}

func TestFromHex(t *testing.T) {
Expand Down
25 changes: 0 additions & 25 deletions common/main_test.go

This file was deleted.

0 comments on commit b201195

Please sign in to comment.