github.com/angryronald/go-kit@v0.0.0-20240505173814-ff2bd9c79dbf/generic/http/client/integrationtest/main_test.go

package integrationtest

import (
	"fmt"
	"log"
	"os"
	"os/exec"
	"sync"
	"testing"
	"time"

	_ "github.com/lib/pq"
	"github.com/ory/dockertest"
	"github.com/redis/go-redis/v9"

	"github.com/angryronald/go-kit/test/docker/mockserver"
	redisInDocker "github.com/angryronald/go-kit/test/docker/redis"
)

var redisClient *redis.Client
var poolResourceMap map[*dockertest.Pool]*dockertest.Resource
var lock sync.Mutex
var port int

func init() {
	poolResourceMap = map[*dockertest.Pool]*dockertest.Resource{}
}

func preseed() {
	cmd := exec.Command("curl", "-X", "PUT",
		"-H", "Content-Type: application/json",
		"-d", `{
		  "httpRequest": {
			"method": "GET",
			"path": "/api/resource",
			"queryStringParameters": {
			  "param1": ["value1"]
			},
			"headers": {
			  "authorization": ["bearer token"],
			  "content-type": "application/json"
			}
		  },
		  "httpResponse": {
			"statusCode": 200,
			"body": "{\"message\": \"Mocked response\"}",
			"delay": {
			  "timeUnit": "MILLISECONDS",
			  "value": 500
			}
		  }
		}`,
		fmt.Sprintf("http://localhost:%d/mockserver/expectation", port),
	)

	// Capture and print the command's output
	output, err := cmd.CombinedOutput()
	if err != nil {
		fmt.Println("Error preseed mockserver endpoint (GET):", err)
		os.Exit(1)
	}

	fmt.Println("Command Output (GET):", string(output))

	cmd = exec.Command("curl", "-X", "PUT",
		"-H", "Content-Type: application/json",
		"-d", `{
		  "httpRequest": {
			"method": "POST",
			"path": "/api/resource",
			"body": {
			  "type": "STRING",
			  "value": "{\"username\":\"test\",\"password\":\"password\"}"
			},
			"headers": {
			  "authorization": ["bearer token"],
			  "content-type": "application/json"
			}
		  },
		  "httpResponse": {
			"statusCode": 200,
			"body": "{\"message\": \"Mocked response with request body\"}",
			"delay": {
			  "timeUnit": "MILLISECONDS",
			  "value": 500
			}
		  }
		}`,
		fmt.Sprintf("http://localhost:%d/mockserver/expectation", port),
	)

	// Capture and print the command's output
	output, err = cmd.CombinedOutput()
	if err != nil {
		fmt.Println("Error preseed mockserver endpoint (POST):", err)
		os.Exit(1)
	}

	fmt.Println("Command Output (POST):", string(output))
}

// Using docker command
func TestMain(m *testing.M) {
	// var err error

	port = mockserver.StartMockServer()

	time.Sleep(10 * time.Second)

	preseed()

	pool, err := dockertest.NewPool("")
	if err != nil {
		log.Fatalf("Could not connect to Docker: %s\n", err)
	}

	redisClientTemp, redisResource := redisInDocker.GenerateInstance(pool)
	if redisClientTemp != nil {
		redisClient = redisClientTemp
		fmt.Printf("Success generate redis instance\n")
	}

	code := m.Run() // execute all the tests

	// Delete the Docker container
	for pool, instance := range poolResourceMap {
		if err := pool.Purge(instance); err != nil {
			log.Fatalf("Could not purge postgres resource: %s\n", err)
		}
	}

	// Delete the Docker container
	if err := pool.Purge(redisResource); err != nil {
		log.Fatalf("Could not purge redis resource: %s\n", err)
	}

	defer os.Exit(code)
	defer mockserver.StopMockServer()
}