/ internal / mcp / plugins_test.go
plugins_test.go
 1  // Copyright (c) 2024-2026 Tencent Zhuque Lab. All rights reserved.
 2  //
 3  // Licensed under the Apache License, Version 2.0 (the "License");
 4  // you may not use this file except in compliance with the License.
 5  // You may obtain a copy of the License at
 6  //
 7  //     http://www.apache.org/licenses/LICENSE-2.0
 8  //
 9  // Unless required by applicable law or agreed to in writing, software
10  // distributed under the License is distributed on an "AS IS" BASIS,
11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  // See the License for the specific language governing permissions and
13  // limitations under the License.
14  //
15  // Requirement: Any integration or derivative work must explicitly attribute
16  // Tencent Zhuque Lab (https://github.com/Tencent/AI-Infra-Guard) in its
17  // documentation or user interface, as detailed in the NOTICE file.
18  
19  package mcp
20  
21  import (
22  	"bytes"
23  	"github.com/stretchr/testify/assert"
24  	"testing"
25  	"text/template"
26  )
27  
28  func TestRegisterPlugins(t *testing.T) {
29  	scanner := NewScanner(nil, nil)
30  	for _, plugin := range scanner.PluginConfigs {
31  		t.Logf("plugin: %s", plugin.Info.Name)
32  		tpl, err := template.New("template").Parse(plugin.PromptTemplate)
33  		assert.NoError(t, err)
34  		var buf bytes.Buffer
35  		err = tpl.Execute(&buf, McpTemplate{
36  			CodePath:              "",
37  			DirectoryStructure:    "",
38  			StaticAnalysisResults: "",
39  			OriginalReports:       "",
40  			McpStructure:          "",
41  		})
42  		assert.NoError(t, err)
43  		//t.Logf("template: %s", buf.String())
44  	}
45  }