diff --git a/configmanage/server/service/configinstance.go b/configmanage/server/service/configinstance.go index 6fc40556cce6412ad07c9e183a0ab50d2b7bef74..771afbc142c1a33422ff2cfd8ef5d16cffc30925 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 e17570780bd387d10cec2fddb82ed4d9912e60d5..e6121591a81386f7e3a1fb54fd78a6f3a3ee226c 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 6d43c3366ca3e6c323c783077e9aa02280048c57..4f5d85eb8ba3928b828511fe2301e708b74441c8 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) + } +}