Add udp test

Signed-off-by: Ulysses Souza <[email protected]>

Author: ulyssessouza

compliance_test.go

@@ -2,8 +2,11 @@ package main
 
 import (
 	"fmt"
+	"net"
 	"testing"
 	"time"
+
+	"gotest.tools/v3/assert"
 )
 
 const (
@@ -14,6 +17,8 @@ const (
 
 	volumefileEntrypoint = "http://" + localhost + ":8080/volumefile"
 	volumeUrl            = volumefileEntrypoint + "?filename="
+
+	udpEntrypoint = "http://" + localhost + ":8080/udp"
 )
 
 func TestSimpleLifecycle(t *testing.T) {
@@ -106,6 +111,28 @@ func TestConfigFile(t *testing.T) {
 	})
 }
 
-func jsonResponse(content string) string {
-	return fmt.Sprintf("{\"response\":\"%s\"}\n", content)
+func TestUdpPort(t *testing.T) {
+	h := TestHelper{
+		T:       t,
+		testDir: "udp_port",
+		specRef: "Networks-top-level-element",
+	}
+	h.TestUpDown(func() {
+		udpValue := "myUdpvalue"
+
+		ServerAddr, err := net.ResolveUDPAddr("udp", localhost+":10001")
+		assert.NilError(h.T, err)
+		LocalAddr, err := net.ResolveUDPAddr("udp", localhost+":0")
+		assert.NilError(h.T, err)
+		Conn, err := net.DialUDP("udp", LocalAddr, ServerAddr)
+		assert.NilError(h.T, err)
+		defer Conn.Close()
+		buf := []byte(fmt.Sprintf("{\"request\":%q}", udpValue))
+		_, err = Conn.Write(buf)
+		assert.NilError(h.T, err)
+
+		actual := h.getHttpBody(udpEntrypoint)
+		expected := jsonResponse(udpValue)
+		h.Check(expected, actual)
+	})
 }

server/main.go

@@ -1,8 +1,10 @@
 package main
 
 import (
+	"encoding/json"
 	"fmt"
 	"io/ioutil"
+	"net"
 	"net/http"
 	"os"
 	"path/filepath"
@@ -12,7 +14,10 @@ import (
 	"github.com/labstack/echo"
 )
 
-const defaultPort = 8080
+const (
+	defaultHttpPort = 8080
+	defaultUdpPort  = 10001
+)
 
 func getMapResponse(response string) map[string]string {
 	return map[string]string{
@@ -75,8 +80,37 @@ func fileHandler(c echo.Context) error {
 	)
 }
 
+var udpValue string
+
+func udpServer() {
+	fmt.Println("Running UDP server...")
+	ServerAddr, err := net.ResolveUDPAddr("udp", fmt.Sprintf(":%d", defaultUdpPort))
+	ServerConn, err := net.ListenUDP("udp", ServerAddr)
+	checkError(err, true)
+	defer ServerConn.Close()
+
+	var objmap map[string]string
+	buf := make([]byte, 1024)
+	for {
+		n, _, err := ServerConn.ReadFromUDP(buf)
+		checkError(err, true)
+		err = json.Unmarshal(buf[:n], &objmap)
+		checkError(err, true)
+		udpValue = objmap["request"]
+	}
+}
+
+func udpHandler(c echo.Context) error {
+	return c.JSON(
+		http.StatusOK,
+		getMapResponse(udpValue),
+	)
+}
+
 func main() {
-	port := defaultPort
+	go udpServer()
+
+	port := defaultHttpPort
 	httpPort := os.Getenv("HTTP_PORT")
 	if httpPort != "" {
 		port, _ = strconv.Atoi(httpPort)
@@ -90,5 +124,15 @@ func main() {
 	}
 	e.GET("/ping", pingHandler)
 	e.GET("/volumefile", fileHandler)
+	e.GET("/udp", udpHandler)
 	e.Logger.Fatal(e.StartServer(s))
 }
+
+func checkError(err error, exitOnError bool) {
+	if err != nil {
+		fmt.Println("Error: ", err)
+		if exitOnError {
+			os.Exit(0)
+		}
+	}
+}

tests/udp_port/compose.yaml

@@ -0,0 +1,11 @@
+version: '3.7'
+
+services:
+  entry:
+    image: test-server
+    ports:
+      - 8080:8080/tcp
+      - target: 10001
+        published: 10001
+        protocol: udp
+        mode: host

tests_helper.go

@@ -175,3 +175,7 @@ func (h TestHelper) getHttpBody(address string) string {
 	assert.NilError(h.T, err)
 	return string(body)
 }
+
+func jsonResponse(content string) string {
+	return fmt.Sprintf("{\"response\":\"%s\"}\n", content)
+}