Add scale test

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

Author: ulyssessouza

Makefile

@@ -8,6 +8,7 @@ check: ## Checks the environment before running any command
 .PHONY: images
 images: ## Build the test images
 	docker build server -t test-server
+	docker build client -t test-client
 
 .PHONY: test
 test: check images ## Run tests

client/Dockerfile

@@ -0,0 +1,9 @@
+FROM golang:1.13.3 AS dev
+WORKDIR /
+COPY main.go /
+RUN go mod init github.com/compose-spec/compatibility-test-suite/client
+RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o main /main.go
+
+FROM scratch AS run
+COPY --from=dev main /
+ENTRYPOINT ["/main"]

client/main.go

@@ -0,0 +1,21 @@
+package main
+
+import (
+	"net/http"
+	"os"
+	"time"
+)
+
+const targetHost = "server"
+
+func main() {
+	time.Sleep(time.Second)
+	value := os.Getenv("HOSTNAME")
+	resp, err := http.Get("http://" + targetHost + ":8080/scalechecker?value=" + value)
+	if err != nil {
+		panic(err)
+	}
+	if resp.StatusCode != http.StatusOK {
+		panic(resp.Status)
+	}
+}

commands/docker-compose.yml

@@ -2,6 +2,7 @@ name: "docker-composeV1"
 command: "docker-compose"
 ps_command: "docker ps -aq"
 global_opts:
+  - name: "--compatibility"
   - name:  "-f"
     value: "compose.yaml"
 up:

compliance_test.go

@@ -6,6 +6,7 @@ import (
 	"testing"
 	"time"
 
+	"gopkg.in/yaml.v2"
 	"gotest.tools/v3/assert"
 )
 
@@ -19,6 +20,8 @@ const (
 	volumeUrl            = volumefileEntrypoint + "?filename="
 
 	udpEntrypoint = "http://" + localhost + ":8080/udp"
+
+	scaleEntrypoint = "http://" + localhost + ":8080/scalechecker"
 )
 
 func TestSimpleLifecycle(t *testing.T) {
@@ -136,3 +139,19 @@ func TestUdpPort(t *testing.T) {
 		h.Check(expected, actual)
 	})
 }
+
+func TestScaling(t *testing.T) {
+	h := TestHelper{
+		T:            t,
+		testDir:      "scaling",
+		skipCommands: []string{"compose-ref"},
+		specRef:      "Networks-top-level-element",
+	}
+	h.TestUpDown(func() {
+		actual := h.getHttpBody(scaleEntrypoint)
+		responseArray := Response{}
+		err := yaml.Unmarshal([]byte(actual), &responseArray)
+		assert.NilError(h.T, err)
+		h.Check(responseArray.Response, "3")
+	})
+}

server/Dockerfile

@@ -4,6 +4,6 @@ COPY main.go /
 RUN go mod init github.com/compose-spec/compatibility-test-suite/server
 RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o main /main.go
 
-FROM ubuntu AS run
+FROM scratch AS run
 COPY --from=dev main /
 ENTRYPOINT ["/main"]

server/main.go

@@ -107,6 +107,45 @@ func udpHandler(c echo.Context) error {
 	)
 }
 
+var scaleValues []string
+
+type ScaleValue struct {
+	Value string `json:"value"`
+}
+
+func scaleCheckHandler(c echo.Context) error {
+	s := new(ScaleValue)
+	if err := c.Bind(s); err != nil {
+		c.Error(err)
+		return err
+	}
+	if s.Value != "" && !containsString(scaleValues, s.Value) {
+		scaleValues = append(scaleValues, s.Value)
+	}
+	return c.JSON(
+		http.StatusOK,
+		getMapResponse(fmt.Sprintf("%d", len(scaleValues))),
+	)
+}
+
+func containsString(ss []string, s string) bool {
+	for _, v := range ss {
+		if v == s {
+			return true
+		}
+	}
+	return false
+}
+
+func checkError(err error, exitOnError bool) {
+	if err != nil {
+		fmt.Println("Error: ", err)
+		if exitOnError {
+			os.Exit(0)
+		}
+	}
+}
+
 func main() {
 	go udpServer()
 
@@ -125,14 +164,6 @@ func main() {
 	e.GET("/ping", pingHandler)
 	e.GET("/volumefile", fileHandler)
 	e.GET("/udp", udpHandler)
+	e.GET("/scalechecker", scaleCheckHandler)
 	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/scaling/compose.yaml

@@ -0,0 +1,13 @@
+version: '3.7'
+
+services:
+  server:
+    image: test-server
+    build: ../../server
+    ports:
+      - 8080:8080
+  client:
+    image: test-client
+    build: ../../client
+    deploy:
+      replicas: 3

tests_helper.go

@@ -45,6 +45,10 @@ type TestHelper struct {
 	specRef      string
 }
 
+type Response struct {
+	Response string `yaml:"response"`
+}
+
 func (h TestHelper) TestUpDown(fun func()) {
 	assert.Assert(h, fun != nil, "Test function cannot be `nil`")
 	for _, f := range h.listFiles(commandsDir) {