Skip to content

Commit 2bf6c54

Browse files
authored
test(mcp): remove side-effects from toolsets tests (#329)
Signed-off-by: Marc Nuri <[email protected]>
1 parent d6936f4 commit 2bf6c54

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

pkg/mcp/toolsets_test.go

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -30,13 +30,13 @@ type ToolsetsSuite struct {
3030

3131
func (s *ToolsetsSuite) SetupTest() {
3232
s.originalToolsets = toolsets.Toolsets()
33-
toolsets.Clear()
3433
s.MockServer = test.NewMockServer()
3534
s.Cfg = configuration.Default()
3635
s.Cfg.KubeConfig = s.MockServer.KubeconfigFile(s.T())
3736
}
3837

3938
func (s *ToolsetsSuite) TearDownTest() {
39+
toolsets.Clear()
4040
for _, toolset := range s.originalToolsets {
4141
toolsets.Register(toolset)
4242
}
@@ -54,6 +54,7 @@ func (s *ToolsetsSuite) TearDownSubTest() {
5454

5555
func (s *ToolsetsSuite) TestNoToolsets() {
5656
s.Run("No toolsets registered", func() {
57+
toolsets.Clear()
5758
s.Cfg.Toolsets = []string{}
5859
s.InitMcpClient()
5960
tools, err := s.ListTools(s.T().Context(), mcp.ListToolsRequest{})
@@ -67,10 +68,6 @@ func (s *ToolsetsSuite) TestNoToolsets() {
6768

6869
func (s *ToolsetsSuite) TestDefaultToolsetsTools() {
6970
s.Run("Default configuration toolsets", func() {
70-
s.Cfg.Toolsets = configuration.Default().Toolsets
71-
toolsets.Register(&core.Toolset{})
72-
toolsets.Register(&config.Toolset{})
73-
toolsets.Register(&helm.Toolset{})
7471
s.InitMcpClient()
7572
tools, err := s.ListTools(s.T().Context(), mcp.ListToolsRequest{})
7673
s.Run("ListTools returns tools", func() {
@@ -92,10 +89,6 @@ func (s *ToolsetsSuite) TestDefaultToolsetsTools() {
9289
func (s *ToolsetsSuite) TestDefaultToolsetsToolsInOpenShift() {
9390
s.Run("Default configuration toolsets in OpenShift", func() {
9491
s.Handle(&test.InOpenShiftHandler{})
95-
s.Cfg.Toolsets = configuration.Default().Toolsets
96-
toolsets.Register(&core.Toolset{})
97-
toolsets.Register(&config.Toolset{})
98-
toolsets.Register(&helm.Toolset{})
9992
s.InitMcpClient()
10093
tools, err := s.ListTools(s.T().Context(), mcp.ListToolsRequest{})
10194
s.Run("ListTools returns tools", func() {
@@ -122,6 +115,7 @@ func (s *ToolsetsSuite) TestGranularToolsetsTools() {
122115
}
123116
for _, testCase := range testCases {
124117
s.Run("Toolset "+testCase.GetName(), func() {
118+
toolsets.Clear()
125119
toolsets.Register(testCase)
126120
s.Cfg.Toolsets = []string{testCase.GetName()}
127121
s.InitMcpClient()

0 commit comments

Comments
 (0)