From 142711f20db4abd793c9213c675b71e2d1305be4 Mon Sep 17 00:00:00 2001
From: Kasey Kirkham <kasey@users.noreply.github.com>
Date: Tue, 17 May 2022 21:59:15 -0500
Subject: [PATCH] cleaning up tests around big int

---
 api/client/builder/client_test.go | 6 ++++++
 api/client/builder/types_test.go  | 9 +++++----
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/api/client/builder/client_test.go b/api/client/builder/client_test.go
index 9821fba663da..612a22f13f77 100644
--- a/api/client/builder/client_test.go
+++ b/api/client/builder/client_test.go
@@ -29,6 +29,9 @@ func TestClient_Status(t *testing.T) {
 	hc := &http.Client{
 		Transport: roundtrip(func(r *http.Request) (*http.Response, error) {
 			defer func() {
+				if r.Body == nil {
+					return
+				}
 				require.NoError(t, r.Body.Close())
 			}()
 			require.Equal(t, statusPath, r.URL.Path)
@@ -47,6 +50,9 @@ func TestClient_Status(t *testing.T) {
 	hc = &http.Client{
 		Transport: roundtrip(func(r *http.Request) (*http.Response, error) {
 			defer func() {
+				if r.Body == nil {
+					return
+				}
 				require.NoError(t, r.Body.Close())
 			}()
 			require.Equal(t, statusPath, r.URL.Path)
diff --git a/api/client/builder/types_test.go b/api/client/builder/types_test.go
index af933a17acc9..034f4f31a000 100644
--- a/api/client/builder/types_test.go
+++ b/api/client/builder/types_test.go
@@ -76,8 +76,6 @@ var testExampleHeaderResponse = `{
 func TestExecutionHeaderResponseUnmarshal(t *testing.T) {
 	hr := &ExecHeaderResponse{}
 	require.NoError(t, json.Unmarshal([]byte(testExampleHeaderResponse), hr))
-	bidValue, err := hr.Data.Message.Value.MarshalText()
-	require.NoError(t, err)
 	cases := []struct {
 		expected string
 		actual   string
@@ -95,7 +93,7 @@ func TestExecutionHeaderResponseUnmarshal(t *testing.T) {
 		},
 		{
 			expected: "1",
-			actual:   string(bidValue),
+			actual:   hr.Data.Message.Value.String(),
 			name:     "ExecHeaderResponse.Value",
 		},
 		{
@@ -201,6 +199,9 @@ func TestExecutionHeaderResponseToProto(t *testing.T) {
 	require.NoError(t, err)
 	txRoot, err := hexutil.Decode("0xcf8e0d4e9587369b2301d0790347320302cc0943d5a1884560367e8208d920f2")
 	require.NoError(t, err)
+	value := big.NewInt(0)
+	value, ok := value.SetString("1", 10)
+	require.Equal(t, true, ok)
 	expected := &eth.SignedBuilderBid{
 		Message: &eth.BuilderBid{
 			Header: &eth.ExecutionPayloadHeader{
@@ -221,7 +222,7 @@ func TestExecutionHeaderResponseToProto(t *testing.T) {
 				TransactionsRoot: txRoot,
 			},
 			// TODO assumes weird byte slice field
-			Value:  []byte(strconv.FormatUint(uint64(1), 10)),
+			Value:  value.Bytes(),
 			Pubkey: pubkey,
 		},
 		Signature: signature,