Commitae06274

committed

fix toolsdk testmain

1 parent82803da commitae06274Copy full SHA for ae06274

File tree

+21

-11

lines changed

+21

-11

lines changed

Lines changed: 12 additions & 9 deletions

Original file line number	Diff line number	Diff line change
`@@ -32,22 +32,25 @@ func (t Tool[T]) Generic() Tool[any] {`
`32`	`32`	`}`
`33`	`33`
`34`	`34`	`var (`
	`35`	`+// All is a list of all tools that can be used in the Coder CLI.`
	`36`	`+// When you add a new tool, be sure to include it here!`
`35`	`37`	`All= []Tool[any]{`
`36`		`-ReportTask.Generic(),`
	`38`	`+CreateTemplateVersion.Generic(),`
	`39`	`+CreateTemplate.Generic(),`
`37`	`40`	`CreateWorkspace.Generic(),`
	`41`	`+CreateWorkspaceBuild.Generic(),`
	`42`	`+DeleteTemplate.Generic(),`
	`43`	`+GetAuthenticatedUser.Generic(),`
	`44`	`+GetTemplateVersionLogs.Generic(),`
	`45`	`+GetWorkspace.Generic(),`
	`46`	`+GetWorkspaceAgentLogs.Generic(),`
	`47`	`+GetWorkspaceBuildLogs.Generic(),`
`38`	`48`	`ListWorkspaces.Generic(),`
`39`	`49`	`ListTemplates.Generic(),`
`40`	`50`	`ListTemplateVersionParameters.Generic(),`
`41`		`-GetAuthenticatedUser.Generic(),`
`42`		`-CreateWorkspaceBuild.Generic(),`
	`51`	`+ReportTask.Generic(),`
`43`	`52`	`UploadTarFile.Generic(),`
`44`		`-CreateTemplateVersion.Generic(),`
`45`		`-CreateTemplate.Generic(),`
`46`		`-GetTemplateVersionLogs.Generic(),`
`47`	`53`	`UpdateTemplateActiveVersion.Generic(),`
`48`		`-DeleteTemplate.Generic(),`
`49`		`-GetWorkspaceAgentLogs.Generic(),`
`50`		`-GetWorkspaceBuildLogs.Generic(),`
`51`	`54`	`}`
`52`	`55`
`53`	`56`	`ReportTask=Tool[string]{`

Lines changed: 9 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -329,18 +329,25 @@ var testedTools sync.Map`
`329`	`329`	`// testTool is a helper function to test a tool and mark it as tested.`
`330`	`330`	`functestTool[Tany](ctx context.Context,t*testing.T,tool toolsdk.Tool[T],argsmap[string]any) (T,error) {`
`331`	`331`	`t.Helper()`
`332`		`-testedTools.Store(tool.Tool.Name,struct{}{})`
	`332`	`+testedTools.Store(tool.Tool.Name,true)`
`333`	`333`	`result,err:=tool.Handler(ctx,args)`
`334`	`334`	`returnresult,err`
`335`	`335`	`}`
`336`	`336`
`337`	`337`	`// TestMain runs after all tests to ensure that all tools in this package have`
`338`	`338`	`// been tested once.`
`339`	`339`	`funcTestMain(m*testing.M) {`
	`340`	`+// Initialize testedTools`
	`341`	`+for_,tool:=rangetoolsdk.All {`
	`342`	`+testedTools.Store(tool.Tool.Name,false)`
	`343`	`+}`
	`344`	`+`
`340`	`345`	`code:=m.Run()`
	`346`	`+`
	`347`	`+// Ensure all tools have been tested`
`341`	`348`	`varuntested []string`
`342`	`349`	`for_,tool:=rangetoolsdk.All {`
`343`		`-if_,ok:=testedTools.Load(tool.Tool.Name);!ok {`
	`350`	`+iftested,ok:=testedTools.Load(tool.Tool.Name);!ok\|\|!tested.(bool) {`
`344`	`351`	`untested=append(untested,tool.Tool.Name)`
`345`	`352`	`}`
`346`	`353`	`}`

Comments

(0)