Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commitec57659

Browse files
committed
chore: update tests
1 parent50bae9d commitec57659

File tree

1 file changed

+67
-5
lines changed

1 file changed

+67
-5
lines changed

‎codersdk/toolsdk/chatgpt_test.go‎

Lines changed: 67 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ import (
1111
"github.com/coder/coder/v2/coderd/coderdtest"
1212
"github.com/coder/coder/v2/coderd/database"
1313
"github.com/coder/coder/v2/coderd/database/dbfake"
14+
"github.com/coder/coder/v2/coderd/database/dbgen"
1415
"github.com/coder/coder/v2/codersdk"
1516
"github.com/coder/coder/v2/codersdk/toolsdk"
1617
)
@@ -99,6 +100,67 @@ func TestChatGPTSearch_TemplateSearch(t *testing.T) {
99100
}
100101
}
101102

103+
funcTestChatGPTSearch_TemplateMultipleFilters(t*testing.T) {
104+
t.Parallel()
105+
106+
// Setup
107+
client,store:=coderdtest.NewWithDatabase(t,nil)
108+
owner:=coderdtest.CreateFirstUser(t,client)
109+
110+
// Create templates directly with specific names for testing filters
111+
dockerTemplate1:=dbgen.Template(t,store, database.Template{
112+
OrganizationID:owner.OrganizationID,
113+
CreatedBy:owner.UserID,
114+
Name:"docker-development",// Name contains "docker"
115+
DisplayName:"Docker Development",
116+
Description:"A Docker-based development template",
117+
})
118+
119+
// Create another template that doesn't contain "docker"
120+
dbgen.Template(t,store, database.Template{
121+
OrganizationID:owner.OrganizationID,
122+
CreatedBy:owner.UserID,
123+
Name:"python-web",// Name doesn't contain "docker"
124+
DisplayName:"Python Web",
125+
Description:"A Python web development template",
126+
})
127+
128+
// Create third template with "docker" in name
129+
dockerTemplate2:=dbgen.Template(t,store, database.Template{
130+
OrganizationID:owner.OrganizationID,
131+
CreatedBy:owner.UserID,
132+
Name:"old-docker-template",// Name contains "docker"
133+
DisplayName:"Old Docker Template",
134+
Description:"An old Docker template",
135+
})
136+
137+
// Create tool dependencies
138+
deps,err:=toolsdk.NewDeps(client)
139+
require.NoError(t,err)
140+
141+
// Execute tool with name filter - should only return templates with "docker" in name
142+
args:= toolsdk.SearchArgs{Query:"templates/name:docker"}
143+
result,err:=testTool(t,toolsdk.ChatGPTSearch,deps,args)
144+
145+
// Verify results
146+
require.NoError(t,err)
147+
require.Len(t,result.Results,2,"Should match both docker templates")
148+
149+
// Validate the results contain both docker templates
150+
templateIDs:=make(map[string]bool)
151+
for_,item:=rangeresult.Results {
152+
require.NotEmpty(t,item.ID)
153+
require.Contains(t,item.ID,"template:")
154+
require.Contains(t,item.URL,"/templates/")
155+
templateIDs[item.ID]=true
156+
}
157+
158+
expectedID1:="template:"+dockerTemplate1.ID.String()
159+
expectedID2:="template:"+dockerTemplate2.ID.String()
160+
require.True(t,templateIDs[expectedID1],"Should contain first docker template")
161+
require.True(t,templateIDs[expectedID2],"Should contain second docker template")
162+
}
163+
102164
funcTestChatGPTSearch_WorkspaceSearch(t*testing.T) {
103165
t.Parallel()
104166

@@ -119,7 +181,7 @@ func TestChatGPTSearch_WorkspaceSearch(t *testing.T) {
119181
},
120182
{
121183
name:"ValidWorkspacesQuery_CurrentUserMe",
122-
query:"workspaces:me",
184+
query:"workspaces/owner:me",
123185
setupOwner:"self",
124186
setupWorkspace:true,
125187
expectError:false,
@@ -133,7 +195,7 @@ func TestChatGPTSearch_WorkspaceSearch(t *testing.T) {
133195
},
134196
{
135197
name:"ValidWorkspacesQuery_SpecificUser",
136-
query:"workspaces:otheruser",
198+
query:"workspaces/owner:otheruser",
137199
setupOwner:"other",
138200
setupWorkspace:true,
139201
expectError:false,
@@ -229,12 +291,12 @@ func TestChatGPTSearch_QueryParsing(t *testing.T) {
229291
},
230292
{
231293
name:"ValidWorkspacesMeQuery",
232-
query:"workspaces:me",
294+
query:"workspaces/owner:me",
233295
expectError:false,
234296
},
235297
{
236298
name:"ValidWorkspacesUserQuery",
237-
query:"workspaces:testuser",
299+
query:"workspaces/owner:testuser",
238300
expectError:false,
239301
},
240302
{
@@ -251,7 +313,7 @@ func TestChatGPTSearch_QueryParsing(t *testing.T) {
251313
},
252314
{
253315
name:"MalformedQuery",
254-
query:"workspaces:user:extra",
316+
query:"invalidtype/somequery",
255317
expectError:true,
256318
errorMsg:"invalid query",
257319
},

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp