From 5b01f1c4315674a79526b024ecbc9a24c8c202d8 Mon Sep 17 00:00:00 2001
From: Kenta420 <poomipat002@gmail.com>
Date: Fri, 27 Oct 2023 14:32:37 +0700
Subject: [PATCH] update backend

---
 server/main_test.go | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 server/main_test.go

diff --git a/server/main_test.go b/server/main_test.go
new file mode 100644
index 0000000..44d9788
--- /dev/null
+++ b/server/main_test.go
@@ -0,0 +1,36 @@
+package main
+
+import (
+	"context"
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
+	"testing"
+	"time"
+)
+
+func TestServer(t *testing.T) {
+	s := NewServer()
+
+	serverCtx, serverStopCtx := context.WithCancel(context.Background())
+	sig := make(chan os.Signal, 1)
+	signal.Notify(sig, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
+
+	// start server 3 minute and stop
+	go func() {
+		time.Sleep(180 * time.Second)
+		serverStopCtx()
+	}()
+
+	// start server
+	go func() {
+		err := s.Run()
+		if err != nil {
+			log.Fatal(err)
+		}
+	}()
+
+	// stop server
+	<-serverCtx.Done()
+}