From a6fc3c5d055a75417de8e2e02b3ad4c2f17237ce Mon Sep 17 00:00:00 2001
From: Rogee <rogeecn@qq.com>
Date: Fri, 29 Nov 2024 17:46:47 +0800
Subject: [PATCH] feat: add main_test.go

---
 backend/common/service/model/gen.go | 10 +++++++---
 backend/main_test.go                | 16 ++++++++++++++++
 2 files changed, 23 insertions(+), 3 deletions(-)
 create mode 100755 backend/main_test.go

diff --git a/backend/common/service/model/gen.go b/backend/common/service/model/gen.go
index 1bbf82e..1c95477 100644
--- a/backend/common/service/model/gen.go
+++ b/backend/common/service/model/gen.go
@@ -26,8 +26,8 @@ func Default(providers ...container.ProviderContainer) container.Providers {
 	}, providers...)
 }
 
-func Command() atom.Option {
-	return atom.Command(
+func Options() []atom.Option {
+	return []atom.Option{
 		atom.Name("model"),
 		atom.Short("run model generator"),
 		atom.RunE(Serve),
@@ -36,7 +36,11 @@ func Command() atom.Option {
 			cmd.Flags().String("path", "./database/models", "generate to path")
 			cmd.Flags().String("transform", "./database/.transform.yaml", "transform config")
 		}),
-	)
+	}
+}
+
+func Command() atom.Option {
+	return atom.Command(Options()...)
 }
 
 type Migrate struct {
diff --git a/backend/main_test.go b/backend/main_test.go
new file mode 100755
index 0000000..a439a84
--- /dev/null
+++ b/backend/main_test.go
@@ -0,0 +1,16 @@
+package main
+
+import (
+	"testing"
+
+	"backend/common/service/model"
+
+	"git.ipao.vip/rogeecn/atom"
+)
+
+func Test_GenModel(t *testing.T) {
+	err := atom.Serve(model.Options()...)
+	if err != nil {
+		t.Fatal(err)
+	}
+}