Function TestChatGPTFetch_WorkspaceFetch

codersdk/toolsdk/chatgpt_test.go:417–492

(t *testing.T)

}

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

	tests := []struct {
		name           string
		setupWorkspace bool
		objectID       string // if empty, will use real workspace ID
		expectError    bool
		errorContains  string
	}{
		{
			name:           "ValidWorkspaceFetch",
			setupWorkspace: true,
			expectError:    false,
		},
		{
			name:           "NonExistentWorkspaceID",
			setupWorkspace: false,
			objectID:       "workspace:" + 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 workspaceID string
			var expectedWorkspace database.WorkspaceTable
			if tt.setupWorkspace {
				workspace := dbfake.WorkspaceBuild(t, store, database.WorkspaceTable{
					OrganizationID: owner.OrganizationID,
					OwnerID:        owner.UserID,
				}).Do()
				expectedWorkspace = workspace.Workspace
				workspaceID = "workspace:" + workspace.Workspace.ID.String()
			} else if tt.objectID != "" {
				workspaceID = tt.objectID
			}

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

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

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