From 04d5d3b9dc0b7f9f1330d9acd9a988552fc5157b Mon Sep 17 00:00:00 2001
From: Phodal Huang <h@phodal.com>
Date: Sat, 18 Jul 2020 13:07:49 +0800
Subject: [PATCH] feat: add simple test

---
 cmd/arch.go      | 7 ++++++-
 cmd/arch_test.go | 7 +++++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/cmd/arch.go b/cmd/arch.go
index fd21d16..8ae80c7 100644
--- a/cmd/arch.go
+++ b/cmd/arch.go
@@ -16,6 +16,7 @@ type ArchCmdConfig struct {
 	DependencePath string
 	IsMergePackage bool
 	FilterString   string
+	IsMergeHeader  bool
 }
 
 var (
@@ -44,6 +45,9 @@ var archCmd = &cobra.Command{
 			return false
 		}
 
+		if archCmdConfig.IsMergeHeader {
+			result = result.MergeHeaderFile(tequila.MergeHeaderFunc)
+		}
 
 		if archCmdConfig.IsMergePackage {
 			result = result.MergeHeaderFile(tequila.MergePackageFunc)
@@ -63,6 +67,7 @@ func init() {
 	rootCmd.AddCommand(archCmd)
 
 	archCmd.PersistentFlags().StringVarP(&archCmdConfig.DependencePath, "dependence", "d", config.CocaConfig.ReporterPath+"/deps.json", "get dependence file")
-	archCmd.PersistentFlags().BoolVarP(&archCmdConfig.IsMergePackage, "mergePackage", "P", false, "merge package/folder for include dependencies")
+	archCmd.PersistentFlags().BoolVarP(&archCmdConfig.IsMergePackage, "mergePackage", "P", false, "merge package")
+	archCmd.PersistentFlags().BoolVarP(&archCmdConfig.IsMergeHeader, "mergeHeader", "H", false, "merge header")
 	archCmd.PersistentFlags().StringVarP(&archCmdConfig.FilterString, "filter", "x", "", "filter -x com.phodal")
 }
diff --git a/cmd/arch_test.go b/cmd/arch_test.go
index c3013cc..5619fcc 100644
--- a/cmd/arch_test.go
+++ b/cmd/arch_test.go
@@ -21,4 +21,11 @@ func TestArch(t *testing.T) {
 		Golden: "",
 	}}
 	RunTestCmd(t, tests)
+
+	header := []testcase.CmdTestCase{{
+		Name:   "arch",
+		Cmd:    "arch -H -x \"com.phodal\"",
+		Golden: "",
+	}}
+	RunTestCmd(t, header)
 }
\ No newline at end of file
-- 
GitLab