Merge pull request #2 from lambdaclass/get-receipts-msg-test

Get receipts msg test

cmd/devp2p/internal/ethtest/suite.go

@@ -72,6 +72,8 @@ func (s *Suite) EthTests() []utesting.Test {
 		{Name: "ZeroRequestID", Fn: s.TestZeroRequestID},
 		// get block bodies
 		{Name: "GetBlockBodies", Fn: s.TestGetBlockBodies},
+		// get block receipts
+		{Name: "GetBlockReceipts", Fn: s.TestGetBlockReceipts},
 		// // malicious handshakes + status
 		{Name: "MaliciousHandshake", Fn: s.TestMaliciousHandshake},
 		{Name: "MaliciousStatus", Fn: s.TestMaliciousStatus},
@@ -371,6 +373,46 @@ func (s *Suite) TestGetBlockBodies(t *utesting.T) {
 	}
 }
 
+// Test the get block receipts p2p message works
+// as expected, heavily based on the test for Block Bodies.
+func (s *Suite) TestGetBlockReceipts(t *utesting.T) {
+	t.Log(`This test sends block receipts requests to the node for known blocks in the test chain.`)
+
+	conn, err := s.dial()
+	if err != nil {
+		t.Fatalf("dial failed: %v", err)
+	}
+	defer conn.Close()
+	if err := conn.peer(s.chain, nil); err != nil {
+		t.Fatalf("peering failed: %v", err)
+	}
+	// Create block bodies request.
+	req := &eth.GetReceiptsPacket{
+		RequestId: 55,
+		GetReceiptsRequest: eth.GetReceiptsRequest{
+			s.chain.blocks[275].Hash(),
+			s.chain.blocks[276].Hash(),
+			s.chain.blocks[75].Hash(),
+		},
+	}
+
+	if err := conn.Write(ethProto, eth.GetReceiptsMsg, req); err != nil {
+		t.Fatalf("could not write to connection: %v", err)
+	}
+	// Wait for response.
+	resp := new(eth.ReceiptsPacket)
+	if err := conn.ReadMsg(ethProto, eth.ReceiptsMsg, &resp); err != nil {
+		t.Fatalf("error reading receipts msg: %v", err)
+	}
+	if got, want := resp.RequestId, req.RequestId; got != want {
+		t.Fatalf("unexpected request id in respond", got, want)
+	}
+	receipts := resp.ReceiptsResponse
+	if len(receipts) != len(req.GetReceiptsRequest) {
+		t.Fatalf("wrong receipts in response: expected %d receipts, got %d", len(req.GetReceiptsRequest), len(receipts))
+	}
+}
+
 // randBuf makes a random buffer size kilobytes large.
 func randBuf(size int) []byte {
 	buf := make([]byte, size*1024)