api wip

Author: heiytor

tests/api_test.go

@@ -0,0 +1,103 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"strconv"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInfo(t *testing.T) {
+	type Expected struct {
+		res map[string]interface{}
+		err error
+	}
+
+	cases := []struct {
+		description string
+		compose     *Enviroment
+		expected    func(string, string) Expected
+	}{
+		{
+			description: "case 1",
+			compose: New().
+				WithEnv("SHELLHUB_HTTP_PORT", strconv.Itoa(getFreePort(t))).
+				WithEnv("SHELLHUB_SSH_PORT", strconv.Itoa(getFreePort(t))).
+				WithEnv("SHELLHUB_VERSION", "v0.14.2").
+				Build(),
+			expected: func(httpPort, sshPort string) Expected {
+				return Expected{
+					res: map[string]interface{}{
+						"endpoints": map[string]interface{}{
+							"api": fmt.Sprintf("localhost:%s", httpPort),
+							"ssh": fmt.Sprintf("localhost:%s", sshPort),
+						},
+						"version": "v0.14.2",
+					},
+					err: nil,
+				}
+			},
+		},
+		{
+			description: "case 2",
+			compose: New().
+				WithEnv("SHELLHUB_HTTP_PORT", strconv.Itoa(getFreePort(t))).
+				WithEnv("SHELLHUB_SSH_PORT", strconv.Itoa(getFreePort(t))).
+				WithEnv("SHELLHUB_VERSION", "v0.10.2").
+				Build(),
+			expected: func(httpPort, sshPort string) Expected {
+				return Expected{
+					res: map[string]interface{}{
+						"endpoints": map[string]interface{}{
+							"api": fmt.Sprintf("localhost:%s", httpPort),
+							"ssh": fmt.Sprintf("localhost:%s", sshPort),
+						},
+						"version": "v0.10.2",
+					},
+					err: nil,
+				}
+			},
+		},
+	}
+
+	for _, tt := range cases {
+		// Avoid "loop variable <> captured by func literal"
+		tc := tt
+
+		t.Run(tc.description, func(t *testing.T) {
+			t.Parallel()
+
+			tc.compose.Run(t, func() {
+				fmt.Printf("desciption %s\n", tc.description)
+
+				fmt.Printf("%s request to %s\n", tc.description, fmt.Sprintf("http://localhost:%s/info", tc.compose.GetEnv("SHELLHUB_HTTP_PORT")))
+
+				// res, _ := resty.
+				// 	New().
+				// 	R().
+				// 	Get(fmt.Sprintf("http://localhost:%s/info", tc.compose.GetEnv("SHELLHUB_HTTP_PORT")))
+
+				res, err := http.Get(fmt.Sprintf("http://localhost:%s/info", tc.compose.GetEnv("SHELLHUB_HTTP_PORT")))
+				assert.NoError(t, err)
+				defer res.Body.Close()
+
+				foo, err := io.ReadAll(res.Body)
+				assert.NoError(t, err)
+
+				body := map[string]interface{}{}
+				assert.NoError(t, json.Unmarshal(foo, &body))
+
+				fmt.Printf("%s http: %+v\n", tc.description, body)
+				fmt.Printf("%s env: %s %s %s\n", tc.description, tc.compose.GetEnv("SHELLHUB_HTTP_PORT"), tc.compose.GetEnv("SHELLHUB_SSH_PORT"), tc.compose.GetEnv("SHELLHUB_VERSION"))
+				fmt.Printf("%s expected: %+v\n", tc.description, tc.expected(tc.compose.GetEnv("SHELLHUB_HTTP_PORT"), tc.compose.GetEnv("SHELLHUB_SSH_PORT")).res)
+				fmt.Println("********************************************")
+				//
+				// assert.Equal(t, tc.expected(tc.compose.GetEnv("SHELLHUB_HTTP_PORT"), tc.compose.GetEnv("SHELLHUB_SSH_PORT")), Expected{body, err})
+			})
+		})
+	}
+}

tests/enviroment.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"context"
+	"testing"
+
+	"github.com/joho/godotenv"
+	"github.com/stretchr/testify/assert"
+	testcontainers "github.com/testcontainers/testcontainers-go/modules/compose"
+)
+
+type Enviroment struct {
+	env map[string]string
+}
+
+type EnviromentBuilder struct {
+	env map[string]string
+}
+
+func New() *EnviromentBuilder {
+	env, _ := godotenv.Read("../.env")
+	return &EnviromentBuilder{
+		env: env,
+	}
+}
+
+func (e *EnviromentBuilder) WithEnv(key, val string) *EnviromentBuilder {
+	e.env[key] = val
+
+	return e
+}
+
+func (e *EnviromentBuilder) Build() *Enviroment {
+	return &Enviroment{
+		env: e.env,
+	}
+}
+
+///////////////////////
+///////////////////////
+///////////////////////
+///////////////////////
+///////////////////////
+
+func (e *Enviroment) GetEnv(key string) string {
+	return e.env[key]
+}
+
+func (e *Enviroment) Run(t *testing.T, cb func()) {
+	compose, err := testcontainers.NewDockerCompose("../docker-compose.yml", "../docker-compose.dev.yml")
+	assert.NoError(t, err)
+
+	t.Cleanup(func() {
+		err := compose.Down(context.Background(), testcontainers.RemoveOrphans(true), testcontainers.RemoveImagesLocal)
+		assert.NoError(t, err)
+	})
+
+	ctx, cancel := context.WithCancel(context.Background())
+	t.Cleanup(cancel)
+
+	err = compose.WithEnv(e.env).Up(ctx, testcontainers.Wait(true))
+	assert.NoError(t, err)
+
+	cb()
+}