main test

main.go

@@ -8,10 +8,14 @@ import (
 )
 
 func main() {
-	ctx, can := signal.NotifyContext(context.Background(), syscall.SIGINT)
-	defer can()
-
-	if err := cmd.Main(ctx); err != nil {
+	if err := Main(context.Background()); err != nil {
 		panic(err)
 	}
 }
+
+func Main(ctx context.Context) error {
+	ctx, can := signal.NotifyContext(ctx, syscall.SIGINT)
+	defer can()
+
+	return cmd.Main(ctx)
+}

main_test.go

@@ -2,21 +2,16 @@ package main_test
 
 import (
 	"context"
-	"os/signal"
-	"show-rss/src/cmd"
-	"syscall"
+	main "show-rss"
 	"testing"
 	"time"
 )
 
-func TestCmdMain(t *testing.T) {
-	ctx, can := signal.NotifyContext(context.Background(), syscall.SIGINT)
+func TestMain(t *testing.T) {
+	ctx, can := context.WithTimeout(context.Background(), 2*time.Second)
 	defer can()
 
-	ctx, can = context.WithTimeout(ctx, 5*time.Second)
-	defer can()
-
-	if err := cmd.Main(ctx); err != nil && ctx.Err() == nil {
+	if err := main.Main(ctx); err != nil && ctx.Err() == nil {
 		t.Fatal(err)
 	}
 }