Function newMockSubAgentAPIWithMaxPortShareLevel

enterprise/coderd/subagent_test.go:267–339

(
	t *testing.T,
	maxPortShareLevel database.AppSharingLevel,
	appCount int,
)

}

func newMockSubAgentAPIWithMaxPortShareLevel(
	t *testing.T,
	maxPortShareLevel database.AppSharingLevel,
	appCount int,
) (context.Context, *agentapi.SubAgentAPI, *[]database.UpsertWorkspaceAppParams) {
	t.Helper()

	ctx := testutil.Context(t, testutil.WaitShort)
	log := testutil.Logger(t)
	clock := quartz.NewMock(t)
	ownerID := uuid.New()
	organizationID := uuid.New()
	templateID := uuid.New()
	parentAgent := database.WorkspaceAgent{
		ID:         uuid.New(),
		ResourceID: uuid.New(),
	}
	workspace := database.Workspace{
		ID:             uuid.New(),
		OwnerID:        ownerID,
		OrganizationID: organizationID,
		TemplateID:     templateID,
	}
	template := database.Template{
		ID:                  templateID,
		MaxPortSharingLevel: maxPortShareLevel,
	}
	upsertedApps := []database.UpsertWorkspaceAppParams{}

	db := dbmock.NewMockStore(gomock.NewController(t))
	db.EXPECT().GetWorkspaceByAgentID(gomock.Any(), parentAgent.ID).Return(workspace, nil)
	db.EXPECT().GetTemplateByID(gomock.Any(), templateID).Return(template, nil)
	db.EXPECT().InsertWorkspaceAgent(gomock.Any(), gomock.Any()).DoAndReturn(
		func(_ context.Context, params database.InsertWorkspaceAgentParams) (database.WorkspaceAgent, error) {
			require.True(t, params.ParentID.Valid)
			require.Equal(t, parentAgent.ID, params.ParentID.UUID)

			return database.WorkspaceAgent{
				ID:        params.ID,
				Name:      params.Name,
				AuthToken: params.AuthToken,
			}, nil
		},
	)
	db.EXPECT().UpsertWorkspaceApp(gomock.Any(), gomock.Any()).DoAndReturn(
		func(_ context.Context, params database.UpsertWorkspaceAppParams) (database.WorkspaceApp, error) {
			upsertedApps = append(upsertedApps, params)
			return database.WorkspaceApp{
				ID:           params.ID,
				AgentID:      params.AgentID,
				Slug:         params.Slug,
				SharingLevel: params.SharingLevel,
			}, nil
		},
	).Times(appCount)

	portSharer := &atomic.Pointer[agplportsharing.PortSharer]{}
	var ps agplportsharing.PortSharer = entportsharing.NewEnterprisePortSharer()