Function TestChatGPTFetch_TemplateFetch

codersdk/toolsdk/chatgpt_test.go:338–415

(t *testing.T)

}

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

	tests := []struct {
		name          string
		setupTemplate bool
		objectID      string // if empty, will use real template ID
		expectError   bool
		errorContains string
	}{
		{
			name:          "ValidTemplateFetch",
			setupTemplate: true,
			expectError:   false,
		},
		{
			name:          "NonExistentTemplateID",
			setupTemplate: false,
			objectID:      "template:" + uuid.NewString(),
			expectError:   true,
			errorContains: "Resource not found",
		},
	}

	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)

			var templateID string
			var expectedTemplate database.Template
			if tt.setupTemplate {
				template := dbfake.TemplateVersion(t, store).
					Seed(database.TemplateVersion{
						OrganizationID: owner.OrganizationID,
						CreatedBy:      owner.UserID,
					}).Do()
				expectedTemplate = template.Template
				templateID = "template:" + template.Template.ID.String()
			} else if tt.objectID != "" {
				templateID = tt.objectID
			}

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

			// Execute tool
			args := toolsdk.FetchArgs{ID: templateID}
			result, err := testTool(t, toolsdk.ChatGPTFetch, deps, args)

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