Function TestChatGPTSearch_TemplateSearch

codersdk/toolsdk/chatgpt_test.go:19–101

(t *testing.T)

)

func TestChatGPTSearch_TemplateSearch(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name           string
		query          string
		setupTemplates int
		expectError    bool
		errorContains  string
	}{
		{
			name:           "ValidTemplatesQuery_MultipleTemplates",
			query:          "templates",
			setupTemplates: 3,
			expectError:    false,
		},
		{
			name:           "ValidTemplatesQuery_NoTemplates",
			query:          "templates",
			setupTemplates: 0,
			expectError:    false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			// Setup
			client, store := coderdtest.NewWithDatabase(t, nil)
			owner := coderdtest.CreateFirstUser(t, client)

			// Create templates as needed
			var expectedTemplates []database.Template
			for i := 0; i < tt.setupTemplates; i++ {
				template := dbfake.TemplateVersion(t, store).
					Seed(database.TemplateVersion{
						OrganizationID: owner.OrganizationID,
						CreatedBy:      owner.UserID,
					}).Do()
				expectedTemplates = append(expectedTemplates, template.Template)
			}

			// Create tool dependencies
			deps, err := toolsdk.NewDeps(client)
			require.NoError(t, err)

			// Execute tool
			args := toolsdk.SearchArgs{Query: tt.query}
			result, err := testTool(t, toolsdk.ChatGPTSearch, deps, args)

			// Verify results
			if tt.expectError {
				require.Error(t, err)
				if tt.errorContains != "" {
					require.Contains(t, err.Error(), tt.errorContains)
				}
				return