From 6764208d5adc006c4dc06cb710b1017f74b168db Mon Sep 17 00:00:00 2001
From: Mostafa Moradian <mostafa@gatewayd.io>
Date: Mon, 18 Sep 2023 20:58:52 +0200
Subject: [PATCH] Add test for run command

---
 cmd/run.go      |  2 +-
 cmd/run_test.go | 72 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 73 insertions(+), 1 deletion(-)
 create mode 100644 cmd/run_test.go

diff --git a/cmd/run.go b/cmd/run.go
index 3aead864..57536d0c 100644
--- a/cmd/run.go
+++ b/cmd/run.go
@@ -109,7 +109,7 @@ func StopGracefully(
 	}
 	for name, server := range servers {
 		logger.Info().Str("name", name).Msg("Stopping server")
-		server.Shutdown()
+		server.Shutdown() //nolint:contextcheck
 		span.AddEvent("Stopped server")
 	}
 	logger.Info().Msg("Stopped all servers")
diff --git a/cmd/run_test.go b/cmd/run_test.go
new file mode 100644
index 00000000..4f8b6dee
--- /dev/null
+++ b/cmd/run_test.go
@@ -0,0 +1,72 @@
+package cmd
+
+import (
+	"os"
+	"sync"
+	"testing"
+	"time"
+
+	"github.com/gatewayd-io/gatewayd/config"
+	"github.com/stretchr/testify/assert"
+	"github.com/zenizh/go-capturer"
+)
+
+func Test_runCmd(t *testing.T) {
+	// Create a test plugins config file.
+	_, err := executeCommandC(rootCmd, "plugin", "init", "--force", "-p", pluginTestConfigFile)
+	assert.NoError(t, err, "plugin init command should not have returned an error")
+	assert.FileExists(t, pluginTestConfigFile, "plugin init command should have created a config file")
+
+	// Create a test config file.
+	_, err = executeCommandC(rootCmd, "config", "init", "--force", "-c", globalTestConfigFile)
+	assert.NoError(t, err, "configInitCmd should not return an error")
+	// Check that the config file was created.
+	assert.FileExists(t, globalTestConfigFile, "configInitCmd should create a config file")
+
+	var waitGroup sync.WaitGroup
+	waitGroup.Add(1)
+	go func(waitGroup *sync.WaitGroup) {
+		time.Sleep(100 * time.Millisecond)
+
+		StopGracefully(
+			runCmd.Context(),
+			runCmd.Context(),
+			nil,
+			nil,
+			nil,
+			loggers[config.Default],
+			servers,
+			stopChan,
+		)
+
+		waitGroup.Done()
+	}(&waitGroup)
+
+	waitGroup.Add(1)
+	go func(waitGroup *sync.WaitGroup) {
+		// Test run command.
+		output := capturer.CaptureOutput(func() {
+			_, err := executeCommandC(rootCmd, "run", "-c", globalTestConfigFile, "-p", pluginTestConfigFile)
+			assert.NoError(t, err, "run command should not have returned an error")
+		})
+		// Print the output for debugging purposes.
+		runCmd.Print(output)
+		// Check if GatewayD started and stopped correctly.
+		assert.Contains(t,
+			output,
+			"GatewayD is running",
+			"run command should have returned the correct output")
+		assert.Contains(t,
+			output,
+			"Stopped all servers\n",
+			"run command should have returned the correct output")
+
+		waitGroup.Done()
+	}(&waitGroup)
+
+	waitGroup.Wait()
+
+	// Clean up.
+	assert.NoError(t, os.Remove(pluginTestConfigFile))
+	assert.NoError(t, os.Remove(globalTestConfigFile))
+}