Added relay message test

tools/integration-tests/tester/tests/example_test.go

-package tests
-
-import (
-	"fmt"
-	"os"
-	"testing"
-
-	"github.com/iotaledger/goshimmer/tools/integration-tests/tester/framework"
-	"github.com/stretchr/testify/assert"
-)
-
-var f *framework.Framework
-
-func TestMain(m *testing.M) {
-	f = framework.New()
-
-	// call the tests
-	os.Exit(m.Run())
-}
-
-func TestExample(t *testing.T) {
-	fmt.Println("This is an independent test.")
-
-	fmt.Printf("There are %d peers in the Docker network available.\n", len(f.Peers()))
-	assert.Equal(t, 6, len(f.Peers()))
-}

tools/integration-tests/tester/tests/relaymessage_test.go

+package tests
+
+import (
+	"os"
+	"testing"
+	"time"
+
+	"github.com/iotaledger/goshimmer/tools/integration-tests/tester/framework"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+var f *framework.Framework
+
+func TestMain(m *testing.M) {
+	f = framework.New()
+
+	// call the tests
+	os.Exit(m.Run())
+}
+
+func TestRelayMessages(t *testing.T) {
+	numMessages := 100
+	hashes := make([]string, numMessages)
+
+	// create messages on random peers
+	for i := 0; i < numMessages; i++ {
+		req := BroadcastDataRequest{Data: "Test"}
+		resp := new(BroadcastDataResponse)
+		err := f.HttpPost(f.RandomPeer(), "broadcastData", req, resp)
+		require.NoError(t, err)
+
+		hashes[i] = resp.Hash
+	}
+
+	// wait for messages to be gossiped
+	time.Sleep(5 * time.Second)
+
+	// check for messages on every peer
+	for _, peer := range f.Peers() {
+		req := GetMessageByHashRequest{Hashes: hashes}
+		resp := new(GetMessageByHashResponse)
+		err := f.HttpPost(peer, "getMessageByHash", req, resp)
+		require.NoError(t, err)
+
+		// check for the count of messages
+		assert.Equal(t, numMessages, len(resp.Messages))
+
+		// check that all messages are present in response
+	outer:
+		for _, hash := range hashes {
+			for _, msg := range resp.Messages {
+				// if message found skip to next
+				if msg.MessageId == hash {
+					continue outer
+				}
+			}
+
+			t.Errorf("MessageId=%s not found in peer %s.", hash, peer.String())
+		}
+	}
+}
+
+// TODO: there should be a nice way to handle all those API endpoints
+type BroadcastDataResponse struct {
+	Hash  string `json:"hash,omitempty"`
+	Error string `json:"error,omitempty"`
+}
+
+type BroadcastDataRequest struct {
+	Data string `json:"data"`
+}
+
+type GetMessageByHashResponse struct {
+	Messages []Message `json:"messages,omitempty"`
+	Error    string    `json:"error,omitempty"`
+}
+
+type GetMessageByHashRequest struct {
+	Hashes []string `json:"hashes"`
+}
+
+type Message struct {
+	MessageId           string `json:"messageId,omitempty"`
+	TrunkTransactionId  string `json:"trunkTransactionId,omitempty"`
+	BranchTransactionId string `json:"branchTransactionId,omitempty"`
+	IssuerPublicKey     string `json:"issuerPublicKey,omitempty"`
+	IssuingTime         string `json:"issuingTime,omitempty"`
+	SequenceNumber      uint64 `json:"sequenceNumber,omitempty"`
+	Payload             string `json:"payload,omitempty"`
+	Signature           string `json:"signature,omitempty"`
+}