prysm-pulse/tools/http-request-sink/main_test.go
Raul Jordan d077483577
Add V3 Suffix to All Prysm Packages (#11083)
* v3 import renamings

* tidy

* fmt

* rev

* Update beacon-chain/core/epoch/precompute/reward_penalty_test.go

* Update beacon-chain/core/helpers/validators_test.go

* Update beacon-chain/db/alias.go

* Update beacon-chain/db/alias.go

* Update beacon-chain/db/alias.go

* Update beacon-chain/db/iface/BUILD.bazel

* Update beacon-chain/db/kv/kv.go

* Update beacon-chain/db/kv/state.go

* Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go

* Update beacon-chain/rpc/prysm/v1alpha1/validator/attester_test.go

* Update beacon-chain/sync/initial-sync/service.go

* fix deps

* fix bad replacements

* fix bad replacements

* change back

* gohashtree version

* fix deps

Co-authored-by: Nishant Das <nishdas93@gmail.com>
Co-authored-by: Potuz <potuz@prysmaticlabs.com>
2022-08-16 12:20:13 +00:00

65 lines
1.5 KiB
Go

package main
import (
"bytes"
"encoding/json"
"io"
"net/http"
"os"
"path/filepath"
"testing"
"github.com/prysmaticlabs/prysm/v3/config/params"
"github.com/prysmaticlabs/prysm/v3/testing/require"
)
type sampleRPCRequest struct {
Name string `json:"name"`
ETHMethod string `json:"eth_method"`
Address string `json:"address"`
}
func Test_parseAndCaptureRequest(t *testing.T) {
tmpFile := filepath.Join(t.TempDir(), "faketest.log")
body := &sampleRPCRequest{
Name: "eth2",
ETHMethod: "eth2_produceBlock",
Address: "0x0923920930923",
}
enc, err := json.Marshal(body)
require.NoError(t, err)
httpReq, err := http.NewRequest("GET", "/", bytes.NewBuffer(enc))
require.NoError(t, err)
reqContent := map[string]interface{}{}
err = parseRequest(httpReq, &reqContent)
require.NoError(t, err)
// If the file doesn't exist, create it, or append to the file.
f, err := os.OpenFile(
tmpFile,
os.O_APPEND|os.O_CREATE|os.O_RDWR,
params.BeaconIoConfig().ReadWritePermissions,
)
require.NoError(t, err)
err = captureRequest(f, reqContent)
require.NoError(t, err)
require.NoError(t, f.Close())
f, err = os.Open(tmpFile)
require.NoError(t, err)
fileContents, err := io.ReadAll(f)
require.NoError(t, err)
receivedContent := map[string]interface{}{}
err = json.Unmarshal(fileContents, &receivedContent)
require.NoError(t, err)
for key, val := range reqContent {
receivedVal, ok := receivedContent[key]
require.Equal(t, true, ok)
require.DeepEqual(t, val, receivedVal)
}
}