From 524f8d686893f66325a80f99262e9ecd293a1dc1 Mon Sep 17 00:00:00 2001
From: wubijie <wubijie@kylinos.cn>
Date: Tue, 5 Nov 2024 11:37:29 +0800
Subject: [PATCH] Newly added test and multiple occupy the harvest of the vault

---
 configmanage/server/service/configinstance.go |  4 +++
 .../server/service/configinstance_test.go     | 16 +++++++++++
 configmanage/server/service/ssh_test.go       | 28 +++++++++++++++++++
 3 files changed, 48 insertions(+)

diff --git a/configmanage/server/service/configinstance.go b/configmanage/server/service/configinstance.go
index 6fc40556..771afbc1 100644
--- a/configmanage/server/service/configinstance.go
+++ b/configmanage/server/service/configinstance.go
@@ -38,6 +38,10 @@ func Init() error {
 	if err != nil {
 		return err
 	}
+	err = db.MySQL().Set("gorm:table_options", "ENGINE=InnoDB CHARACTER SET utf8mb4").AutoMigrate(&internal.SSHFile{})
+	if err != nil {
+		return err
+	}
 	return nil
 }
 
diff --git a/configmanage/server/service/configinstance_test.go b/configmanage/server/service/configinstance_test.go
index e1757078..e6121591 100644
--- a/configmanage/server/service/configinstance_test.go
+++ b/configmanage/server/service/configinstance_test.go
@@ -87,3 +87,19 @@ func TestConfigInstanceTypeHost_Add(t *testing.T) {
 		os.Exit(-1)
 	}
 }
+
+func TestConfigInstanceTypeSSH_Add(t *testing.T) {
+	ci := &ConfigInstance{
+		UUID:        uuid.New().String(),
+		Type:        global.SSH,
+		Description: "test-SSH-description",
+		BatchIds:    []int{6},
+		DepartIds:   []int{60},
+		Nodes:       []string{"33333333-5f8e-42df-b2d0-49bf55cfeb56"},
+	}
+	err := ci.Add()
+	if err != nil {
+		fmt.Printf("Add() error = %v, want nil", err)
+		os.Exit(-1)
+	}
+}
diff --git a/configmanage/server/service/ssh_test.go b/configmanage/server/service/ssh_test.go
index 6d43c336..4f5d85eb 100644
--- a/configmanage/server/service/ssh_test.go
+++ b/configmanage/server/service/ssh_test.go
@@ -1 +1,29 @@
 package service
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+	"testing"
+
+	"github.com/google/uuid"
+)
+
+func TestSSHConfig_Record(t *testing.T) {
+	// 设置测试数据
+	sc := &SSHConfig{
+		UUID:           uuid.New().String(),
+		ConfigInfoUUID: "5973e993-6236-4b53-9eb6-0cc23c652460",
+		Content:        json.RawMessage(`{"test": "test"}`),
+		Path:           "/ssh",
+		Name:           "ssh.txt",
+		IsActive:       false,
+	}
+
+	// 调用被测试的函数
+	err := sc.Record()
+	if err != nil {
+		fmt.Printf("record error: %s\n", err)
+		os.Exit(-1)
+	}
+}
-- 
Gitee