add test

Author: PatStiles

operator/pkg/s3_test.go

@@ -1,6 +1,7 @@
 package operator
 
 import (
+	"context"
 	"fmt"
 	"net/http"
 	"net/http/httptest"
@@ -10,12 +11,8 @@ import (
 	"testing"
 	"time"
 
-	"github.com/ethereum/go-ethereum/accounts/abi/bind"
-	"github.com/ethereum/go-ethereum/common"
 	"github.com/stretchr/testify/assert"
 	retry "github.com/yetanotherco/aligned_layer/core"
-	"github.com/yetanotherco/aligned_layer/core/chainio"
-	s3 "github.com/yetanotherco/aligned_layer/operator/pkg/s3"
 )
 
 // Function wrapper around `make run_storage`
@@ -38,34 +35,38 @@ func RunStorage() (*exec.Cmd, error) {
 
 	return cmd, nil
 }
-
-func TestBatchersBalances(t *testing.T) {
-	cmd, err := RunStorage()
-	if err != nil {
-		t.Errorf("Error setting up Anvil: %s\n", err)
-	}
-
+func TestRequestBatch(t *testing.T) {
+	/*
+		cmd, err := RunStorage()
+		if err != nil {
+			t.Errorf("Error setting up Anvil: %s\n", err)
+		}
+	*/
 	// To Simulate Retrieving information from S3 we create a mock http server.
 	expected := "dummy data"
 	svr := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		fmt.Fprintf(w, expected)
+		t.Errorf("error %v %v", w, expected)
 	}))
 	defer svr.Close()
+
+	req, err := http.NewRequestWithContext(context.Background(), "GET", svr.URL, nil)
 	if err != nil {
-		return
+		t.Errorf("Error creating req: %s\n", err)
 	}
-	senderAddress := common.HexToAddress("0x0")
 
-	batcher_func := s3.RequestBatch(avsWriter, &bind.CallOpts{}, senderAddress)
+	batcher_func := RequestBatch(req, context.Background())
 	_, err = batcher_func()
 	assert.Nil(t, err)
 
-	if err := cmd.Process.Kill(); err != nil {
-		t.Errorf("Error killing process: %v\n", err)
-		return
-	}
+	svr.Close()
+	/*
+		if err := cmd.Process.Kill(); err != nil {
+			t.Errorf("Error killing process: %v\n", err)
+			return
+		}
+	*/
 
-	batcher_func = chainio.BatcherBalances(avsWriter, &bind.CallOpts{}, senderAddress)
+	batcher_func = RequestBatch(req, context.Background())
 	_, err = batcher_func()
 	assert.NotNil(t, err)
 	if _, ok := err.(retry.PermanentError); ok {
@@ -77,17 +78,22 @@ func TestBatchersBalances(t *testing.T) {
 		return
 	}
 
-	cmd, _, err = retry_test.SetupAnvil(8545)
-	if err != nil {
-		t.Errorf("Error setting up Anvil: %s\n", err)
-	}
+	svr.Start()
+	/*
+		cmd, err = RunStorage()
+		if err != nil {
+			t.Errorf("Error setting up Anvil: %s\n", err)
+		}
+	*/
 
-	batcher_func = chainio.BatcherBalances(avsWriter, &bind.CallOpts{}, senderAddress)
+	batcher_func = RequestBatch(req, context.Background())
 	_, err = batcher_func()
 	assert.Nil(t, err)
 
-	if err := cmd.Process.Kill(); err != nil {
-		t.Errorf("Error killing process: %v\n", err)
-		return
-	}
+	/*
+		if err := cmd.Process.Kill(); err != nil {
+			t.Errorf("Error killing process: %v\n", err)
+			return
+		}
+	*/
 }