From a20be9f58e75bc5c9d69cc9d3cae67f802dde264 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Tue, 12 Nov 2024 11:48:32 +0000 Subject: [PATCH 01/19] feat: add registered users insights --- coderd/coderd.go | 1 + coderd/database/dbauthz/dbauthz.go | 7 +++ coderd/database/dbmem/dbmem.go | 42 +++++++++++++++ coderd/database/dbmetrics/querymetrics.go | 7 +++ coderd/database/dbmock/dbmock.go | 15 ++++++ coderd/database/querier.go | 3 ++ coderd/database/queries.sql.go | 64 ++++++++++++++++++++--- coderd/database/queries/insights.sql | 15 ++++++ coderd/insights.go | 48 +++++++++++++++++ coderd/insights_test.go | 60 +++++++++++++++++++++ coderd/rbac/object_gen.go | 1 + coderd/rbac/policy/policy.go | 2 + codersdk/insights.go | 29 ++++++++++ codersdk/rbacresources_gen.go | 2 +- site/src/api/rbacresourcesGenerated.ts | 1 + site/src/api/typesGenerated.ts | 15 ++++++ 16 files changed, 304 insertions(+), 8 deletions(-) diff --git a/coderd/coderd.go b/coderd/coderd.go index 75e588e89624b..85cce4701d2fe 100644 --- a/coderd/coderd.go +++ b/coderd/coderd.go @@ -1260,6 +1260,7 @@ func New(options *Options) *API { r.Get("/user-activity", api.insightsUserActivity) r.Get("/user-latency", api.insightsUserLatency) r.Get("/templates", api.insightsTemplates) + r.Get("/total-users", api.insightsTotalUsers) }) r.Route("/debug", func(r chi.Router) { r.Use( diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go index c855d5a1984df..322f3ee43ac3f 100644 --- a/coderd/database/dbauthz/dbauthz.go +++ b/coderd/database/dbauthz/dbauthz.go @@ -2283,6 +2283,13 @@ func (q *querier) GetTemplatesWithFilter(ctx context.Context, arg database.GetTe return q.db.GetAuthorizedTemplates(ctx, arg, prep) } +func (q *querier) GetAccumulatedUsersInsights(ctx context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { + if err := q.authorizeContext(ctx, policy.ActionViewInsights, rbac.ResourceUser); err != nil { + return nil, err + } + return q.db.GetAccumulatedUsersInsights(ctx, arg) +} + func (q *querier) GetUnexpiredLicenses(ctx context.Context) ([]database.License, error) { if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { return nil, err diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go index 9a306db09785e..5aacc5a06e96e 100644 --- a/coderd/database/dbmem/dbmem.go +++ b/coderd/database/dbmem/dbmem.go @@ -5169,6 +5169,48 @@ func (q *FakeQuerier) GetTemplatesWithFilter(ctx context.Context, arg database.G return q.GetAuthorizedTemplates(ctx, arg, nil) } +func (q *FakeQuerier) GetAccumulatedUsersInsights(_ context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { + q.mutex.RLock() + defer q.mutex.RUnlock() + + newUsersByDate := make(map[time.Time]int64) + for _, user := range q.users { + if user.CreatedAt.Before(arg.StartTime) || user.CreatedAt.After(arg.EndTime) { + continue + } + date := user.CreatedAt.Truncate(24 * time.Hour) + newUsersByDate[date]++ + } + + accumulatedByDate := make(map[time.Time]int64) + var accumulatedCount int64 + sortedDates := make([]time.Time, 0, len(newUsersByDate)) + + for date := range newUsersByDate { + sortedDates = append(sortedDates, date) + } + + // Sort dates in ascending order + sort.Slice(sortedDates, func(i, j int) bool { + return sortedDates[i].Before(sortedDates[j]) + }) + + // Calculate accumulated count + for _, date := range sortedDates { + accumulatedCount += newUsersByDate[date] + accumulatedByDate[date] = accumulatedCount + } + + var rows []database.GetAccumulatedUsersInsightsRow + for _, date := range sortedDates { + rows = append(rows, database.GetAccumulatedUsersInsightsRow{ + Date: date, + Count: accumulatedByDate[date], + }) + } + return rows, nil +} + func (q *FakeQuerier) GetUnexpiredLicenses(_ context.Context) ([]database.License, error) { q.mutex.RLock() defer q.mutex.RUnlock() diff --git a/coderd/database/dbmetrics/querymetrics.go b/coderd/database/dbmetrics/querymetrics.go index cee25e482bbaa..f536a279f2282 100644 --- a/coderd/database/dbmetrics/querymetrics.go +++ b/coderd/database/dbmetrics/querymetrics.go @@ -1253,6 +1253,13 @@ func (m queryMetricsStore) GetTemplatesWithFilter(ctx context.Context, arg datab return templates, err } +func (m queryMetricsStore) GetAccumulatedUsersInsights(ctx context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { + start := time.Now() + r0, r1 := m.s.GetAccumulatedUsersInsights(ctx, arg) + m.queryLatencies.WithLabelValues("GetAccumulatedUsersInsights").Observe(time.Since(start).Seconds()) + return r0, r1 +} + func (m queryMetricsStore) GetUnexpiredLicenses(ctx context.Context) ([]database.License, error) { start := time.Now() licenses, err := m.s.GetUnexpiredLicenses(ctx) diff --git a/coderd/database/dbmock/dbmock.go b/coderd/database/dbmock/dbmock.go index d8721f56d3f4e..558fbf97f1673 100644 --- a/coderd/database/dbmock/dbmock.go +++ b/coderd/database/dbmock/dbmock.go @@ -832,6 +832,21 @@ func (mr *MockStoreMockRecorder) GetAPIKeysLastUsedAfter(arg0, arg1 any) *gomock return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeysLastUsedAfter", reflect.TypeOf((*MockStore)(nil).GetAPIKeysLastUsedAfter), arg0, arg1) } +// GetAccumulatedUsersInsights mocks base method. +func (m *MockStore) GetAccumulatedUsersInsights(arg0 context.Context, arg1 database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetAccumulatedUsersInsights", arg0, arg1) + ret0, _ := ret[0].([]database.GetAccumulatedUsersInsightsRow) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetAccumulatedUsersInsights indicates an expected call of GetAccumulatedUsersInsights. +func (mr *MockStoreMockRecorder) GetAccumulatedUsersInsights(arg0, arg1 any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAccumulatedUsersInsights", reflect.TypeOf((*MockStore)(nil).GetAccumulatedUsersInsights), arg0, arg1) +} + // GetActiveUserCount mocks base method. func (m *MockStore) GetActiveUserCount(arg0 context.Context) (int64, error) { m.ctrl.T.Helper() diff --git a/coderd/database/querier.go b/coderd/database/querier.go index 46d1b1ae5b322..cfba923d31a96 100644 --- a/coderd/database/querier.go +++ b/coderd/database/querier.go @@ -262,6 +262,9 @@ type sqlcQuerier interface { GetTemplateVersionsCreatedAfter(ctx context.Context, createdAt time.Time) ([]TemplateVersion, error) GetTemplates(ctx context.Context) ([]Template, error) GetTemplatesWithFilter(ctx context.Context, arg GetTemplatesWithFilterParams) ([]Template, error) + // GetTotalUsersByDate returns the number of registered users for each day + // in the given timeframe. + GetAccumulatedUsersInsights(ctx context.Context, arg GetAccumulatedUsersInsightsParams) ([]GetAccumulatedUsersInsightsRow, error) GetUnexpiredLicenses(ctx context.Context) ([]License, error) // GetUserActivityInsights returns the ranking with top active users. // The result can be filtered on template_ids, meaning only user data diff --git a/coderd/database/queries.sql.go b/coderd/database/queries.sql.go index 87d3c17f5400f..951216d1dc5df 100644 --- a/coderd/database/queries.sql.go +++ b/coderd/database/queries.sql.go @@ -2887,6 +2887,56 @@ func (q *sqlQuerier) GetTemplateUsageStats(ctx context.Context, arg GetTemplateU return items, nil } +const getTotalUsersByDateInsights = `-- name: GetAccumulatedUsersInsights :many +SELECT + date_trunc('day', created_at)::date AS date, + COUNT(*) AS count +FROM + users +WHERE + created_at >= $1::timestamptz + AND created_at < $2::timestamptz +GROUP BY + date +ORDER BY + date ASC +` + +type GetAccumulatedUsersInsightsParams struct { + StartTime time.Time `db:"start_time" json:"start_time"` + EndTime time.Time `db:"end_time" json:"end_time"` +} + +type GetAccumulatedUsersInsightsRow struct { + Date time.Time `db:"date" json:"date"` + Count int64 `db:"count" json:"count"` +} + +// GetTotalUsersByDate returns the number of registered users for each day +// in the given timeframe. +func (q *sqlQuerier) GetAccumulatedUsersInsights(ctx context.Context, arg GetAccumulatedUsersInsightsParams) ([]GetAccumulatedUsersInsightsRow, error) { + rows, err := q.db.QueryContext(ctx, getTotalUsersByDateInsights, arg.StartTime, arg.EndTime) + if err != nil { + return nil, err + } + defer rows.Close() + var items []GetAccumulatedUsersInsightsRow + for rows.Next() { + var i GetAccumulatedUsersInsightsRow + if err := rows.Scan(&i.Date, &i.Count); err != nil { + return nil, err + } + items = append(items, i) + } + if err := rows.Close(); err != nil { + return nil, err + } + if err := rows.Err(); err != nil { + return nil, err + } + return items, nil +} + const getUserActivityInsights = `-- name: GetUserActivityInsights :many WITH deployment_stats AS ( @@ -3356,7 +3406,7 @@ func (q *sqlQuerier) GetJFrogXrayScanByWorkspaceAndAgentID(ctx context.Context, } const upsertJFrogXrayScanByWorkspaceAndAgentID = `-- name: UpsertJFrogXrayScanByWorkspaceAndAgentID :exec -INSERT INTO +INSERT INTO jfrog_xray_scans ( agent_id, workspace_id, @@ -3365,7 +3415,7 @@ INSERT INTO medium, results_url ) -VALUES +VALUES ($1, $2, $3, $4, $5, $6) ON CONFLICT (agent_id, workspace_id) DO UPDATE SET critical = $3, high = $4, medium = $5, results_url = $6 @@ -6175,7 +6225,7 @@ FROM provisioner_keys WHERE organization_id = $1 -AND +AND lower(name) = lower($2) ` @@ -6291,10 +6341,10 @@ WHERE AND -- exclude reserved built-in key id != '00000000-0000-0000-0000-000000000001'::uuid -AND +AND -- exclude reserved user-auth key id != '00000000-0000-0000-0000-000000000002'::uuid -AND +AND -- exclude reserved psk key id != '00000000-0000-0000-0000-000000000003'::uuid ` @@ -7980,7 +8030,7 @@ func (q *sqlQuerier) GetTailnetTunnelPeerIDs(ctx context.Context, srcID uuid.UUI } const updateTailnetPeerStatusByCoordinator = `-- name: UpdateTailnetPeerStatusByCoordinator :exec -UPDATE +UPDATE tailnet_peers SET status = $2 @@ -12436,7 +12486,7 @@ WITH agent_stats AS ( coalesce((PERCENTILE_CONT(0.95) WITHIN GROUP (ORDER BY connection_median_latency_ms)), -1)::FLOAT AS workspace_connection_latency_95 FROM workspace_agent_stats -- The greater than 0 is to support legacy agents that don't report connection_median_latency_ms. - WHERE workspace_agent_stats.created_at > $1 AND connection_median_latency_ms > 0 + WHERE workspace_agent_stats.created_at > $1 AND connection_median_latency_ms > 0 GROUP BY user_id, agent_id, workspace_id, template_id ), latest_agent_stats AS ( SELECT diff --git a/coderd/database/queries/insights.sql b/coderd/database/queries/insights.sql index de107bc0e80c7..813c0936be511 100644 --- a/coderd/database/queries/insights.sql +++ b/coderd/database/queries/insights.sql @@ -771,3 +771,18 @@ SELECT FROM unique_template_params utp JOIN workspace_build_parameters wbp ON (utp.workspace_build_ids @> ARRAY[wbp.workspace_build_id] AND utp.name = wbp.name) GROUP BY utp.num, utp.template_ids, utp.name, utp.type, utp.display_name, utp.description, utp.options, wbp.value; + +-- name: GetAccumulatedUsersInsights :many +-- GetAccumulatedUsersInsights returns the accumulated number of users created +-- in the given timeframe. It returns the accumulated number of users for each date +-- within the specified timeframe, providing a running total of user sign-ups. +SELECT + date_trunc('day', created_at)::date AS date, + COUNT(*) OVER (ORDER BY date_trunc('day', created_at)::date) AS accumulated_users +FROM + users +WHERE + created_at >= @start_time::timestamptz + AND created_at < @end_time::timestamptz +ORDER BY + date_trunc('day', created_at)::date; diff --git a/coderd/insights.go b/coderd/insights.go index 7234a88d44fe9..f6c4e093d98fc 100644 --- a/coderd/insights.go +++ b/coderd/insights.go @@ -470,6 +470,54 @@ func (api *API) insightsTemplates(rw http.ResponseWriter, r *http.Request) { httpapi.Write(ctx, rw, http.StatusOK, resp) } +func (api *API) insightsTotalUsers(rw http.ResponseWriter, r *http.Request) { + ctx := r.Context() + + p := httpapi.NewQueryParamParser(). + RequiredNotEmpty("start_time"). + RequiredNotEmpty("end_time") + vals := r.URL.Query() + var ( + // The QueryParamParser does not preserve timezone, so we need + // to parse the time ourselves. + startTimeString = p.String(vals, "", "start_time") + endTimeString = p.String(vals, "", "end_time") + ) + p.ErrorExcessParams(vals) + if len(p.Errors) > 0 { + httpapi.Write(ctx, rw, http.StatusBadRequest, codersdk.Response{ + Message: "Query parameters have invalid values.", + Validations: p.Errors, + }) + return + } + + startTime, endTime, ok := parseInsightsStartAndEndTime(ctx, rw, time.Now(), startTimeString, endTimeString) + if !ok { + return + } + + rows, err := api.Database.GetAccumulatedUsersInsights(ctx, database.GetAccumulatedUsersInsightsParams{ + StartTime: startTime, + EndTime: endTime, + }) + if err != nil { + httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{ + Message: "Internal error fetching total users.", + Detail: err.Error(), + }) + return + } + var res codersdk.TotalUsersInsightResponse + for _, row := range rows { + res = append(res, codersdk.TotalUserByDate{ + Date: row.Date.Format(time.DateOnly), + Total: uint64(row.Count), + }) + } + httpapi.Write(r.Context(), rw, http.StatusOK, res) +} + // convertTemplateInsightsApps builds the list of builtin apps and template apps // from the provided database rows, builtin apps are implicitly a part of all // templates. diff --git a/coderd/insights_test.go b/coderd/insights_test.go index bf8aa4bc44506..bd08baf264a03 100644 --- a/coderd/insights_test.go +++ b/coderd/insights_test.go @@ -2351,3 +2351,63 @@ func TestGenericInsights_RBAC(t *testing.T) { }) } } + +func TestTotalUsersInsight(t *testing.T) { + t.Parallel() + + t.Run("Success", func(t *testing.T) { + t.Parallel() + + client, db := coderdtest.NewWithDatabase(t, nil) + coderdtest.CreateFirstUser(t, client) + ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort) + t.Cleanup(cancel) + + // Given: a deployment with many users + dbgen.User(t, db, database.User{ + Email: "user1@coder.com", + Username: "user1", + CreatedAt: time.Now().UTC().AddDate(0, 0, -2), + }) + dbgen.User(t, db, database.User{ + Email: "user2@coder.com", + Username: "user2", + CreatedAt: time.Now().UTC().AddDate(0, 0, -2), + }) + dbgen.User(t, db, database.User{ + Email: "user3@coder.com", + Username: "user3", + CreatedAt: time.Now().UTC().AddDate(0, 0, -1), + }) + + // When: requesting the total users by day + threeDaysAgo := time.Now().UTC().AddDate(0, 0, -2).Truncate(24 * time.Hour) + today := time.Now().UTC().Truncate(time.Hour).Add(time.Hour) + res, err := client.TotalUsersInsight(ctx, codersdk.TotalUsersInsightRequest{ + StartTime: threeDaysAgo, + EndTime: today, + }) + require.NoError(t, err) + + // Then: expect to have the accumulated users count growing by day + require.Len(t, res, 4) + }) + + t.Run("EmptyTimes", func(t *testing.T) { + t.Parallel() + + client, _ := coderdtest.NewWithDatabase(t, nil) + coderdtest.CreateFirstUser(t, client) + ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort) + t.Cleanup(cancel) + + // When: requesting total users without a time range + _, err := client.TotalUsersInsight(ctx, codersdk.TotalUsersInsightRequest{}) + + // Then: expect a bad request error + require.Error(t, err) + var apiErr *codersdk.Error + require.ErrorAs(t, err, &apiErr) + require.Equal(t, http.StatusBadRequest, apiErr.StatusCode()) + }) +} diff --git a/coderd/rbac/object_gen.go b/coderd/rbac/object_gen.go index 12ac8a86a7dd9..e0730b97a08c1 100644 --- a/coderd/rbac/object_gen.go +++ b/coderd/rbac/object_gen.go @@ -269,6 +269,7 @@ var ( // - "ActionReadPersonal" :: read personal user data like user settings and auth links // - "ActionUpdate" :: update an existing user // - "ActionUpdatePersonal" :: update personal data + // - "ActionViewInsights" :: view insights ResourceUser = Object{ Type: "user", } diff --git a/coderd/rbac/policy/policy.go b/coderd/rbac/policy/policy.go index 2691eed9fe0a9..a678297abafa1 100644 --- a/coderd/rbac/policy/policy.go +++ b/coderd/rbac/policy/policy.go @@ -84,6 +84,8 @@ var RBACPermissions = map[string]PermissionDefinition{ ActionReadPersonal: actDef("read personal user data like user settings and auth links"), ActionUpdatePersonal: actDef("update personal data"), + + ActionViewInsights: actDef("view insights"), }, }, "workspace": { diff --git a/codersdk/insights.go b/codersdk/insights.go index c9e708de8f34a..cb8c81a720bf4 100644 --- a/codersdk/insights.go +++ b/codersdk/insights.go @@ -282,3 +282,32 @@ func (c *Client) TemplateInsights(ctx context.Context, req TemplateInsightsReque var result TemplateInsightsResponse return result, json.NewDecoder(resp.Body).Decode(&result) } + +type TotalUsersInsightRequest struct { + StartTime time.Time `json:"start_time" format:"date-time"` + EndTime time.Time `json:"end_time" format:"date-time"` +} +type TotalUserByDate struct { + Date string `json:"date"` + Total uint64 `json:"total"` +} +type TotalUsersInsightResponse []TotalUserByDate + +func (c *Client) TotalUsersInsight(ctx context.Context, req TotalUsersInsightRequest) (TotalUsersInsightResponse, error) { + qp := url.Values{} + qp.Add("start_time", req.StartTime.Format(insightsTimeLayout)) + qp.Add("end_time", req.EndTime.Format(insightsTimeLayout)) + + reqURL := fmt.Sprintf("/api/v2/insights/total-users?%s", qp.Encode()) + resp, err := c.Request(ctx, http.MethodGet, reqURL, nil) + if err != nil { + return nil, xerrors.Errorf("make request: %w", err) + } + defer resp.Body.Close() + + if resp.StatusCode != http.StatusOK { + return nil, ReadBodyAsError(resp) + } + var result TotalUsersInsightResponse + return result, json.NewDecoder(resp.Body).Decode(&result) +} diff --git a/codersdk/rbacresources_gen.go b/codersdk/rbacresources_gen.go index c903d5f8a02ff..0f78f013655a8 100644 --- a/codersdk/rbacresources_gen.go +++ b/codersdk/rbacresources_gen.go @@ -87,7 +87,7 @@ var RBACResourceActions = map[RBACResource][]RBACAction{ ResourceSystem: {ActionCreate, ActionDelete, ActionRead, ActionUpdate}, ResourceTailnetCoordinator: {ActionCreate, ActionDelete, ActionRead, ActionUpdate}, ResourceTemplate: {ActionCreate, ActionDelete, ActionRead, ActionUpdate, ActionViewInsights}, - ResourceUser: {ActionCreate, ActionDelete, ActionRead, ActionReadPersonal, ActionUpdate, ActionUpdatePersonal}, + ResourceUser: {ActionCreate, ActionDelete, ActionRead, ActionReadPersonal, ActionUpdate, ActionUpdatePersonal, ActionViewInsights}, ResourceWorkspace: {ActionApplicationConnect, ActionCreate, ActionDelete, ActionRead, ActionSSH, ActionWorkspaceStart, ActionWorkspaceStop, ActionUpdate}, ResourceWorkspaceDormant: {ActionApplicationConnect, ActionCreate, ActionDelete, ActionRead, ActionSSH, ActionWorkspaceStart, ActionWorkspaceStop, ActionUpdate}, ResourceWorkspaceProxy: {ActionCreate, ActionDelete, ActionRead, ActionUpdate}, diff --git a/site/src/api/rbacresourcesGenerated.ts b/site/src/api/rbacresourcesGenerated.ts index 4ce48121edbb3..a7f4033454471 100644 --- a/site/src/api/rbacresourcesGenerated.ts +++ b/site/src/api/rbacresourcesGenerated.ts @@ -153,6 +153,7 @@ export const RBACResourceActions: Partial< read_personal: "read personal user data like user settings and auth links", update: "update an existing user", update_personal: "update personal data", + view_insights: "view insights", }, workspace: { application_connect: "connect to workspace apps via browser", diff --git a/site/src/api/typesGenerated.ts b/site/src/api/typesGenerated.ts index f89d9eada822c..b5113a0e637de 100644 --- a/site/src/api/typesGenerated.ts +++ b/site/src/api/typesGenerated.ts @@ -1524,6 +1524,21 @@ export interface TokensFilter { readonly include_all: boolean; } +// From codersdk/insights.go +export interface TotalUserByDate { + readonly date: string; + readonly total: number; +} + +// From codersdk/insights.go +export interface TotalUsersInsightRequest { + readonly start_time: string; + readonly end_time: string; +} + +// From codersdk/insights.go +export type TotalUsersInsightResponse = Readonly> + // From codersdk/deployment.go export interface TraceConfig { readonly enable: boolean; From 24baca9989c2dbe535dd74a222141dce956f47c3 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Tue, 12 Nov 2024 17:33:40 +0000 Subject: [PATCH 02/19] Add tests --- coderd/insights.go | 13 +++++-- coderd/insights_test.go | 80 ++++++++++++++++++++++++++++++----------- 2 files changed, 69 insertions(+), 24 deletions(-) diff --git a/coderd/insights.go b/coderd/insights.go index f6c4e093d98fc..168e55ab9921f 100644 --- a/coderd/insights.go +++ b/coderd/insights.go @@ -508,11 +508,18 @@ func (api *API) insightsTotalUsers(rw http.ResponseWriter, r *http.Request) { }) return } + var res codersdk.TotalUsersInsightResponse - for _, row := range rows { + currentTotal := uint64(0) + for d := startTime; d.Before(endTime) || d.Equal(endTime); d = d.AddDate(0, 0, 1) { + for _, row := range rows { + if row.Date.Equal(d) { + currentTotal = uint64(row.Count) + } + } res = append(res, codersdk.TotalUserByDate{ - Date: row.Date.Format(time.DateOnly), - Total: uint64(row.Count), + Date: d.Format(time.DateOnly), + Total: currentTotal, }) } httpapi.Write(r.Context(), rw, http.StatusOK, res) diff --git a/coderd/insights_test.go b/coderd/insights_test.go index bd08baf264a03..96d1900d595c4 100644 --- a/coderd/insights_test.go +++ b/coderd/insights_test.go @@ -1212,7 +1212,7 @@ func TestTemplateInsights_Golden(t *testing.T) { }, }, { - name: "two days ago, no data", + name: "two dates ago, no data", ignoreTimes: true, makeRequest: func(_ []*testTemplate) codersdk.TemplateInsightsRequest { twoDaysAgo := time.Now().UTC().Truncate(24*time.Hour).AddDate(0, 0, -2) @@ -2110,7 +2110,7 @@ func TestTemplateInsights_BadRequest(t *testing.T) { EndTime: today, Interval: codersdk.InsightsReportIntervalWeek, }) - assert.Error(t, err, "last report interval must have at least 6 days") + assert.Error(t, err, "last report interval must have at least 6 dates") _, err = client.TemplateInsights(ctx, codersdk.TemplateInsightsRequest{ StartTime: today.AddDate(0, 0, -1), @@ -2364,50 +2364,88 @@ func TestTotalUsersInsight(t *testing.T) { t.Cleanup(cancel) // Given: a deployment with many users + dbgen.User(t, db, database.User{ Email: "user1@coder.com", Username: "user1", - CreatedAt: time.Now().UTC().AddDate(0, 0, -2), + CreatedAt: daysAgo(2), }) dbgen.User(t, db, database.User{ Email: "user2@coder.com", Username: "user2", - CreatedAt: time.Now().UTC().AddDate(0, 0, -2), + CreatedAt: daysAgo(2), }) dbgen.User(t, db, database.User{ Email: "user3@coder.com", Username: "user3", - CreatedAt: time.Now().UTC().AddDate(0, 0, -1), + CreatedAt: daysAgo(1), }) - // When: requesting the total users by day - threeDaysAgo := time.Now().UTC().AddDate(0, 0, -2).Truncate(24 * time.Hour) - today := time.Now().UTC().Truncate(time.Hour).Add(time.Hour) + // When: requesting the accumulated total of users by day res, err := client.TotalUsersInsight(ctx, codersdk.TotalUsersInsightRequest{ - StartTime: threeDaysAgo, - EndTime: today, + StartTime: daysAgo(2), + EndTime: today(), }) require.NoError(t, err) - // Then: expect to have the accumulated users count growing by day - require.Len(t, res, 4) + // Then: expect the correct number of dates and the growth of the accumulated total of users + require.Len(t, res, 3, "expect 3 dates of data") + + // First day + require.Equal(t, res[0].Date, daysAgo(2).Format(time.DateOnly)) + require.Equal(t, res[0].Total, uint64(2)) + + // Second day + require.Equal(t, res[1].Date, daysAgo(1).Format(time.DateOnly)) + require.Equal(t, res[1].Total, uint64(3)) + + // Third day + require.Equal(t, res[2].Date, today().Format(time.DateOnly)) + require.Equal(t, res[2].Total, uint64(4)) }) - t.Run("EmptyTimes", func(t *testing.T) { + t.Run("DatesWithNoUserRegistration", func(t *testing.T) { t.Parallel() - client, _ := coderdtest.NewWithDatabase(t, nil) + // Given: a deployment with no users getting created every day + client, db := coderdtest.NewWithDatabase(t, nil) coderdtest.CreateFirstUser(t, client) ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort) t.Cleanup(cancel) + dbgen.User(t, db, database.User{ + Email: "user1@coder.com", + Username: "user1", + CreatedAt: daysAgo(2), + }) - // When: requesting total users without a time range - _, err := client.TotalUsersInsight(ctx, codersdk.TotalUsersInsightRequest{}) + // When: requesting the accumulated total of users for dates with no users created + res, err := client.TotalUsersInsight(ctx, codersdk.TotalUsersInsightRequest{ + StartTime: daysAgo(2).Truncate(24 * time.Hour), + EndTime: today(), + }) + require.NoError(t, err) + + // Then: expect the correct number of dates. If there are no users created, the total should be 0. + require.Len(t, res, 3, "expect 3 dates of data") + + // First day + require.Equal(t, res[0].Date, daysAgo(2).Format(time.DateOnly)) + require.Equal(t, res[0].Total, uint64(1)) + + // Second day - no users created so should contain the same total as the previous day + require.Equal(t, res[1].Date, daysAgo(1).Format(time.DateOnly)) + require.Equal(t, res[1].Total, uint64(1)) - // Then: expect a bad request error - require.Error(t, err) - var apiErr *codersdk.Error - require.ErrorAs(t, err, &apiErr) - require.Equal(t, http.StatusBadRequest, apiErr.StatusCode()) + // Third day + require.Equal(t, res[2].Date, today().Format(time.DateOnly)) + require.Equal(t, res[2].Total, uint64(2)) }) } + +func daysAgo(days int) time.Time { + return time.Now().UTC().AddDate(0, 0, -days).Truncate(24 * time.Hour) +} + +func today() time.Time { + return time.Now().UTC().Truncate(time.Hour).Add(time.Hour) +} From da5f1b89d4dacb01cbeb0f609cb653e9528c7f38 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Tue, 12 Nov 2024 17:42:51 +0000 Subject: [PATCH 03/19] Fix make gen --- coderd/apidoc/docs.go | 57 +++++++++++ coderd/apidoc/swagger.json | 53 ++++++++++ coderd/database/dbauthz/dbauthz.go | 14 +-- coderd/database/dbmem/dbmem.go | 84 ++++++++-------- coderd/database/dbmetrics/querymetrics.go | 14 +-- coderd/database/querier.go | 7 +- coderd/database/queries.sql.go | 113 +++++++++++----------- coderd/database/queries/insights.sql | 2 +- coderd/insights.go | 11 ++- docs/reference/api/insights.md | 51 ++++++++++ docs/reference/api/schemas.md | 16 +++ 11 files changed, 304 insertions(+), 118 deletions(-) diff --git a/coderd/apidoc/docs.go b/coderd/apidoc/docs.go index 983abb61169c9..88ff237a59899 100644 --- a/coderd/apidoc/docs.go +++ b/coderd/apidoc/docs.go @@ -1292,6 +1292,52 @@ const docTemplate = `{ } } }, + "/insights/total-users": { + "get": { + "security": [ + { + "CoderSessionToken": [] + } + ], + "produces": [ + "application/json" + ], + "tags": [ + "Insights" + ], + "summary": "Get insights about total users", + "operationId": "get-insights-about-total-users", + "parameters": [ + { + "type": "string", + "format": "date-time", + "description": "Start time", + "name": "start_time", + "in": "query", + "required": true + }, + { + "type": "string", + "format": "date-time", + "description": "End time", + "name": "end_time", + "in": "query", + "required": true + } + ], + "responses": { + "200": { + "description": "OK", + "schema": { + "type": "array", + "items": { + "$ref": "#/definitions/codersdk.TotalUserByDate" + } + } + } + } + } + }, "/insights/user-activity": { "get": { "security": [ @@ -13680,6 +13726,17 @@ const docTemplate = `{ } } }, + "codersdk.TotalUserByDate": { + "type": "object", + "properties": { + "date": { + "type": "string" + }, + "total": { + "type": "integer" + } + } + }, "codersdk.TraceConfig": { "type": "object", "properties": { diff --git a/coderd/apidoc/swagger.json b/coderd/apidoc/swagger.json index 67cc92c71331d..d5a9bc5cfe726 100644 --- a/coderd/apidoc/swagger.json +++ b/coderd/apidoc/swagger.json @@ -1121,6 +1121,48 @@ } } }, + "/insights/total-users": { + "get": { + "security": [ + { + "CoderSessionToken": [] + } + ], + "produces": ["application/json"], + "tags": ["Insights"], + "summary": "Get insights about total users", + "operationId": "get-insights-about-total-users", + "parameters": [ + { + "type": "string", + "format": "date-time", + "description": "Start time", + "name": "start_time", + "in": "query", + "required": true + }, + { + "type": "string", + "format": "date-time", + "description": "End time", + "name": "end_time", + "in": "query", + "required": true + } + ], + "responses": { + "200": { + "description": "OK", + "schema": { + "type": "array", + "items": { + "$ref": "#/definitions/codersdk.TotalUserByDate" + } + } + } + } + } + }, "/insights/user-activity": { "get": { "security": [ @@ -12415,6 +12457,17 @@ } } }, + "codersdk.TotalUserByDate": { + "type": "object", + "properties": { + "date": { + "type": "string" + }, + "total": { + "type": "integer" + } + } + }, "codersdk.TraceConfig": { "type": "object", "properties": { diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go index 312361302375d..49bd59c461eba 100644 --- a/coderd/database/dbauthz/dbauthz.go +++ b/coderd/database/dbauthz/dbauthz.go @@ -1377,6 +1377,13 @@ func (q *querier) GetAPIKeysLastUsedAfter(ctx context.Context, lastUsed time.Tim return fetchWithPostFilter(q.auth, policy.ActionRead, q.db.GetAPIKeysLastUsedAfter)(ctx, lastUsed) } +func (q *querier) GetAccumulatedUsersInsights(ctx context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { + if err := q.authorizeContext(ctx, policy.ActionViewInsights, rbac.ResourceUser); err != nil { + return nil, err + } + return q.db.GetAccumulatedUsersInsights(ctx, arg) +} + func (q *querier) GetActiveUserCount(ctx context.Context) (int64, error) { if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { return 0, err @@ -2285,13 +2292,6 @@ func (q *querier) GetTemplatesWithFilter(ctx context.Context, arg database.GetTe return q.db.GetAuthorizedTemplates(ctx, arg, prep) } -func (q *querier) GetAccumulatedUsersInsights(ctx context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { - if err := q.authorizeContext(ctx, policy.ActionViewInsights, rbac.ResourceUser); err != nil { - return nil, err - } - return q.db.GetAccumulatedUsersInsights(ctx, arg) -} - func (q *querier) GetUnexpiredLicenses(ctx context.Context) ([]database.License, error) { if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { return nil, err diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go index 68fcf7683f231..5da5dcfb79826 100644 --- a/coderd/database/dbmem/dbmem.go +++ b/coderd/database/dbmem/dbmem.go @@ -2320,6 +2320,48 @@ func (q *FakeQuerier) GetAPIKeysLastUsedAfter(_ context.Context, after time.Time return apiKeys, nil } +func (q *FakeQuerier) GetAccumulatedUsersInsights(_ context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { + q.mutex.RLock() + defer q.mutex.RUnlock() + + newUsersByDate := make(map[time.Time]int64) + for _, user := range q.users { + if user.CreatedAt.Before(arg.StartTime) || user.CreatedAt.After(arg.EndTime) { + continue + } + date := user.CreatedAt.Truncate(24 * time.Hour) + newUsersByDate[date]++ + } + + accumulatedByDate := make(map[time.Time]int64) + var accumulatedCount int64 + sortedDates := make([]time.Time, 0, len(newUsersByDate)) + + for date := range newUsersByDate { + sortedDates = append(sortedDates, date) + } + + // Sort dates in ascending order + sort.Slice(sortedDates, func(i, j int) bool { + return sortedDates[i].Before(sortedDates[j]) + }) + + // Calculate accumulated count + for _, date := range sortedDates { + accumulatedCount += newUsersByDate[date] + accumulatedByDate[date] = accumulatedCount + } + + var rows []database.GetAccumulatedUsersInsightsRow + for _, date := range sortedDates { + rows = append(rows, database.GetAccumulatedUsersInsightsRow{ + Date: date, + Total: accumulatedByDate[date], + }) + } + return rows, nil +} + func (q *FakeQuerier) GetActiveUserCount(_ context.Context) (int64, error) { q.mutex.RLock() defer q.mutex.RUnlock() @@ -5169,48 +5211,6 @@ func (q *FakeQuerier) GetTemplatesWithFilter(ctx context.Context, arg database.G return q.GetAuthorizedTemplates(ctx, arg, nil) } -func (q *FakeQuerier) GetAccumulatedUsersInsights(_ context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { - q.mutex.RLock() - defer q.mutex.RUnlock() - - newUsersByDate := make(map[time.Time]int64) - for _, user := range q.users { - if user.CreatedAt.Before(arg.StartTime) || user.CreatedAt.After(arg.EndTime) { - continue - } - date := user.CreatedAt.Truncate(24 * time.Hour) - newUsersByDate[date]++ - } - - accumulatedByDate := make(map[time.Time]int64) - var accumulatedCount int64 - sortedDates := make([]time.Time, 0, len(newUsersByDate)) - - for date := range newUsersByDate { - sortedDates = append(sortedDates, date) - } - - // Sort dates in ascending order - sort.Slice(sortedDates, func(i, j int) bool { - return sortedDates[i].Before(sortedDates[j]) - }) - - // Calculate accumulated count - for _, date := range sortedDates { - accumulatedCount += newUsersByDate[date] - accumulatedByDate[date] = accumulatedCount - } - - var rows []database.GetAccumulatedUsersInsightsRow - for _, date := range sortedDates { - rows = append(rows, database.GetAccumulatedUsersInsightsRow{ - Date: date, - Count: accumulatedByDate[date], - }) - } - return rows, nil -} - func (q *FakeQuerier) GetUnexpiredLicenses(_ context.Context) ([]database.License, error) { q.mutex.RLock() defer q.mutex.RUnlock() diff --git a/coderd/database/dbmetrics/querymetrics.go b/coderd/database/dbmetrics/querymetrics.go index f536a279f2282..68a6014f60a9f 100644 --- a/coderd/database/dbmetrics/querymetrics.go +++ b/coderd/database/dbmetrics/querymetrics.go @@ -462,6 +462,13 @@ func (m queryMetricsStore) GetAPIKeysLastUsedAfter(ctx context.Context, lastUsed return apiKeys, err } +func (m queryMetricsStore) GetAccumulatedUsersInsights(ctx context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { + start := time.Now() + r0, r1 := m.s.GetAccumulatedUsersInsights(ctx, arg) + m.queryLatencies.WithLabelValues("GetAccumulatedUsersInsights").Observe(time.Since(start).Seconds()) + return r0, r1 +} + func (m queryMetricsStore) GetActiveUserCount(ctx context.Context) (int64, error) { start := time.Now() count, err := m.s.GetActiveUserCount(ctx) @@ -1253,13 +1260,6 @@ func (m queryMetricsStore) GetTemplatesWithFilter(ctx context.Context, arg datab return templates, err } -func (m queryMetricsStore) GetAccumulatedUsersInsights(ctx context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { - start := time.Now() - r0, r1 := m.s.GetAccumulatedUsersInsights(ctx, arg) - m.queryLatencies.WithLabelValues("GetAccumulatedUsersInsights").Observe(time.Since(start).Seconds()) - return r0, r1 -} - func (m queryMetricsStore) GetUnexpiredLicenses(ctx context.Context) ([]database.License, error) { start := time.Now() licenses, err := m.s.GetUnexpiredLicenses(ctx) diff --git a/coderd/database/querier.go b/coderd/database/querier.go index cfba923d31a96..ef04615d39e6d 100644 --- a/coderd/database/querier.go +++ b/coderd/database/querier.go @@ -115,6 +115,10 @@ type sqlcQuerier interface { GetAPIKeysByLoginType(ctx context.Context, loginType LoginType) ([]APIKey, error) GetAPIKeysByUserID(ctx context.Context, arg GetAPIKeysByUserIDParams) ([]APIKey, error) GetAPIKeysLastUsedAfter(ctx context.Context, lastUsed time.Time) ([]APIKey, error) + // GetAccumulatedUsersInsights returns the accumulated number of users created + // in the given timeframe. It returns the accumulated number of users for each date + // within the specified timeframe, providing a running total of user sign-ups. + GetAccumulatedUsersInsights(ctx context.Context, arg GetAccumulatedUsersInsightsParams) ([]GetAccumulatedUsersInsightsRow, error) GetActiveUserCount(ctx context.Context) (int64, error) GetActiveWorkspaceBuildsByTemplateID(ctx context.Context, templateID uuid.UUID) ([]WorkspaceBuild, error) GetAllTailnetAgents(ctx context.Context) ([]TailnetAgent, error) @@ -262,9 +266,6 @@ type sqlcQuerier interface { GetTemplateVersionsCreatedAfter(ctx context.Context, createdAt time.Time) ([]TemplateVersion, error) GetTemplates(ctx context.Context) ([]Template, error) GetTemplatesWithFilter(ctx context.Context, arg GetTemplatesWithFilterParams) ([]Template, error) - // GetTotalUsersByDate returns the number of registered users for each day - // in the given timeframe. - GetAccumulatedUsersInsights(ctx context.Context, arg GetAccumulatedUsersInsightsParams) ([]GetAccumulatedUsersInsightsRow, error) GetUnexpiredLicenses(ctx context.Context) ([]License, error) // GetUserActivityInsights returns the ranking with top active users. // The result can be filtered on template_ids, meaning only user data diff --git a/coderd/database/queries.sql.go b/coderd/database/queries.sql.go index 951216d1dc5df..67a2323d8e8aa 100644 --- a/coderd/database/queries.sql.go +++ b/coderd/database/queries.sql.go @@ -2131,6 +2131,55 @@ func (q *sqlQuerier) UpdateGroupByID(ctx context.Context, arg UpdateGroupByIDPar return i, err } +const getAccumulatedUsersInsights = `-- name: GetAccumulatedUsersInsights :many +SELECT + date_trunc('day', created_at)::date AS date, + COUNT(*) OVER (ORDER BY date_trunc('day', created_at)::date) AS total +FROM + users +WHERE + created_at >= $1::timestamptz + AND created_at < $2::timestamptz +ORDER BY + date_trunc('day', created_at)::date +` + +type GetAccumulatedUsersInsightsParams struct { + StartTime time.Time `db:"start_time" json:"start_time"` + EndTime time.Time `db:"end_time" json:"end_time"` +} + +type GetAccumulatedUsersInsightsRow struct { + Date time.Time `db:"date" json:"date"` + Total int64 `db:"total" json:"total"` +} + +// GetAccumulatedUsersInsights returns the accumulated number of users created +// in the given timeframe. It returns the accumulated number of users for each date +// within the specified timeframe, providing a running total of user sign-ups. +func (q *sqlQuerier) GetAccumulatedUsersInsights(ctx context.Context, arg GetAccumulatedUsersInsightsParams) ([]GetAccumulatedUsersInsightsRow, error) { + rows, err := q.db.QueryContext(ctx, getAccumulatedUsersInsights, arg.StartTime, arg.EndTime) + if err != nil { + return nil, err + } + defer rows.Close() + var items []GetAccumulatedUsersInsightsRow + for rows.Next() { + var i GetAccumulatedUsersInsightsRow + if err := rows.Scan(&i.Date, &i.Total); err != nil { + return nil, err + } + items = append(items, i) + } + if err := rows.Close(); err != nil { + return nil, err + } + if err := rows.Err(); err != nil { + return nil, err + } + return items, nil +} + const getTemplateAppInsights = `-- name: GetTemplateAppInsights :many WITH -- Create a list of all unique apps by template, this is used to @@ -2887,56 +2936,6 @@ func (q *sqlQuerier) GetTemplateUsageStats(ctx context.Context, arg GetTemplateU return items, nil } -const getTotalUsersByDateInsights = `-- name: GetAccumulatedUsersInsights :many -SELECT - date_trunc('day', created_at)::date AS date, - COUNT(*) AS count -FROM - users -WHERE - created_at >= $1::timestamptz - AND created_at < $2::timestamptz -GROUP BY - date -ORDER BY - date ASC -` - -type GetAccumulatedUsersInsightsParams struct { - StartTime time.Time `db:"start_time" json:"start_time"` - EndTime time.Time `db:"end_time" json:"end_time"` -} - -type GetAccumulatedUsersInsightsRow struct { - Date time.Time `db:"date" json:"date"` - Count int64 `db:"count" json:"count"` -} - -// GetTotalUsersByDate returns the number of registered users for each day -// in the given timeframe. -func (q *sqlQuerier) GetAccumulatedUsersInsights(ctx context.Context, arg GetAccumulatedUsersInsightsParams) ([]GetAccumulatedUsersInsightsRow, error) { - rows, err := q.db.QueryContext(ctx, getTotalUsersByDateInsights, arg.StartTime, arg.EndTime) - if err != nil { - return nil, err - } - defer rows.Close() - var items []GetAccumulatedUsersInsightsRow - for rows.Next() { - var i GetAccumulatedUsersInsightsRow - if err := rows.Scan(&i.Date, &i.Count); err != nil { - return nil, err - } - items = append(items, i) - } - if err := rows.Close(); err != nil { - return nil, err - } - if err := rows.Err(); err != nil { - return nil, err - } - return items, nil -} - const getUserActivityInsights = `-- name: GetUserActivityInsights :many WITH deployment_stats AS ( @@ -3406,7 +3405,7 @@ func (q *sqlQuerier) GetJFrogXrayScanByWorkspaceAndAgentID(ctx context.Context, } const upsertJFrogXrayScanByWorkspaceAndAgentID = `-- name: UpsertJFrogXrayScanByWorkspaceAndAgentID :exec -INSERT INTO +INSERT INTO jfrog_xray_scans ( agent_id, workspace_id, @@ -3415,7 +3414,7 @@ INSERT INTO medium, results_url ) -VALUES +VALUES ($1, $2, $3, $4, $5, $6) ON CONFLICT (agent_id, workspace_id) DO UPDATE SET critical = $3, high = $4, medium = $5, results_url = $6 @@ -6225,7 +6224,7 @@ FROM provisioner_keys WHERE organization_id = $1 -AND +AND lower(name) = lower($2) ` @@ -6341,10 +6340,10 @@ WHERE AND -- exclude reserved built-in key id != '00000000-0000-0000-0000-000000000001'::uuid -AND +AND -- exclude reserved user-auth key id != '00000000-0000-0000-0000-000000000002'::uuid -AND +AND -- exclude reserved psk key id != '00000000-0000-0000-0000-000000000003'::uuid ` @@ -8030,7 +8029,7 @@ func (q *sqlQuerier) GetTailnetTunnelPeerIDs(ctx context.Context, srcID uuid.UUI } const updateTailnetPeerStatusByCoordinator = `-- name: UpdateTailnetPeerStatusByCoordinator :exec -UPDATE +UPDATE tailnet_peers SET status = $2 @@ -12486,7 +12485,7 @@ WITH agent_stats AS ( coalesce((PERCENTILE_CONT(0.95) WITHIN GROUP (ORDER BY connection_median_latency_ms)), -1)::FLOAT AS workspace_connection_latency_95 FROM workspace_agent_stats -- The greater than 0 is to support legacy agents that don't report connection_median_latency_ms. - WHERE workspace_agent_stats.created_at > $1 AND connection_median_latency_ms > 0 + WHERE workspace_agent_stats.created_at > $1 AND connection_median_latency_ms > 0 GROUP BY user_id, agent_id, workspace_id, template_id ), latest_agent_stats AS ( SELECT diff --git a/coderd/database/queries/insights.sql b/coderd/database/queries/insights.sql index 813c0936be511..b120006829c2b 100644 --- a/coderd/database/queries/insights.sql +++ b/coderd/database/queries/insights.sql @@ -778,7 +778,7 @@ GROUP BY utp.num, utp.template_ids, utp.name, utp.type, utp.display_name, utp.de -- within the specified timeframe, providing a running total of user sign-ups. SELECT date_trunc('day', created_at)::date AS date, - COUNT(*) OVER (ORDER BY date_trunc('day', created_at)::date) AS accumulated_users + COUNT(*) OVER (ORDER BY date_trunc('day', created_at)::date) AS total FROM users WHERE diff --git a/coderd/insights.go b/coderd/insights.go index 168e55ab9921f..1847900ee3472 100644 --- a/coderd/insights.go +++ b/coderd/insights.go @@ -470,6 +470,15 @@ func (api *API) insightsTemplates(rw http.ResponseWriter, r *http.Request) { httpapi.Write(ctx, rw, http.StatusOK, resp) } +// @Summary Get insights about total users +// @ID get-insights-about-total-users +// @Security CoderSessionToken +// @Produce json +// @Tags Insights +// @Param start_time query string true "Start time" format(date-time) +// @Param end_time query string true "End time" format(date-time) +// @Success 200 {object} codersdk.TotalUsersInsightResponse +// @Router /insights/total-users [get] func (api *API) insightsTotalUsers(rw http.ResponseWriter, r *http.Request) { ctx := r.Context() @@ -514,7 +523,7 @@ func (api *API) insightsTotalUsers(rw http.ResponseWriter, r *http.Request) { for d := startTime; d.Before(endTime) || d.Equal(endTime); d = d.AddDate(0, 0, 1) { for _, row := range rows { if row.Date.Equal(d) { - currentTotal = uint64(row.Count) + currentTotal = uint64(row.Total) } } res = append(res, codersdk.TotalUserByDate{ diff --git a/docs/reference/api/insights.md b/docs/reference/api/insights.md index d9bb2327a9517..6c34ab97f612b 100644 --- a/docs/reference/api/insights.md +++ b/docs/reference/api/insights.md @@ -138,6 +138,57 @@ curl -X GET http://coder-server:8080/api/v2/insights/templates?start_time=2019-0 To perform this operation, you must be authenticated. [Learn more](authentication.md). +## Get insights about total users + +### Code samples + +```shell +# Example request using curl +curl -X GET http://coder-server:8080/api/v2/insights/total-users?start_time=2019-08-24T14%3A15%3A22Z&end_time=2019-08-24T14%3A15%3A22Z \ + -H 'Accept: application/json' \ + -H 'Coder-Session-Token: API_KEY' +``` + +`GET /insights/total-users` + +### Parameters + +| Name | In | Type | Required | Description | +| ------------ | ----- | ----------------- | -------- | ----------- | +| `start_time` | query | string(date-time) | true | Start time | +| `end_time` | query | string(date-time) | true | End time | + +### Example responses + +> 200 Response + +```json +[ + { + "date": "string", + "total": 0 + } +] +``` + +### Responses + +| Status | Meaning | Description | Schema | +| ------ | ------------------------------------------------------- | ----------- | ----------------------------------------------------------------------- | +| 200 | [OK](https://tools.ietf.org/html/rfc7231#section-6.3.1) | OK | array of [codersdk.TotalUserByDate](schemas.md#codersdktotaluserbydate) | + +

Response Schema

+ +Status Code **200** + +| Name | Type | Required | Restrictions | Description | +| -------------- | ------- | -------- | ------------ | ----------- | +| `[array item]` | array | false | | | +| `» date` | string | false | | | +| `» total` | integer | false | | | + +To perform this operation, you must be authenticated. [Learn more](authentication.md). + ## Get insights about user activity ### Code samples diff --git a/docs/reference/api/schemas.md b/docs/reference/api/schemas.md index fe8db822aafb5..8bc003faf336f 100644 --- a/docs/reference/api/schemas.md +++ b/docs/reference/api/schemas.md @@ -5800,6 +5800,22 @@ CreateWorkspaceRequest provides options for creating a new workspace. Only one o | -------------------- | ------- | -------- | ------------ | ----------- | | `max_token_lifetime` | integer | false | | | +## codersdk.TotalUserByDate + +```json +{ + "date": "string", + "total": 0 +} +``` + +### Properties + +| Name | Type | Required | Restrictions | Description | +| ------- | ------- | -------- | ------------ | ----------- | +| `date` | string | false | | | +| `total` | integer | false | | | + ## codersdk.TraceConfig ```json From e77cf6726c5f65c29e0a01c79022beac285e35e9 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Wed, 13 Nov 2024 16:33:09 +0000 Subject: [PATCH 04/19] Fix query --- coderd/database/queries.sql.go | 18 +++++++++++------- coderd/database/queries/insights.sql | 18 +++++++++++------- 2 files changed, 22 insertions(+), 14 deletions(-) diff --git a/coderd/database/queries.sql.go b/coderd/database/queries.sql.go index 67a2323d8e8aa..b7db23bae5e75 100644 --- a/coderd/database/queries.sql.go +++ b/coderd/database/queries.sql.go @@ -2132,16 +2132,20 @@ func (q *sqlQuerier) UpdateGroupByID(ctx context.Context, arg UpdateGroupByIDPar } const getAccumulatedUsersInsights = `-- name: GetAccumulatedUsersInsights :many +WITH RECURSIVE date_series AS ( + SELECT $1::timestamptz AS date + UNION ALL + SELECT date + INTERVAL '1 day' + FROM date_series + WHERE date + INTERVAL '1 day' <= $2::timestamptz +) SELECT - date_trunc('day', created_at)::date AS date, - COUNT(*) OVER (ORDER BY date_trunc('day', created_at)::date) AS total + d.date::date AS date, + (SELECT COUNT(*) FROM users u WHERE u.created_at <= d.date) AS total FROM - users -WHERE - created_at >= $1::timestamptz - AND created_at < $2::timestamptz + date_series d ORDER BY - date_trunc('day', created_at)::date + d.date ` type GetAccumulatedUsersInsightsParams struct { diff --git a/coderd/database/queries/insights.sql b/coderd/database/queries/insights.sql index b120006829c2b..7495998b3723f 100644 --- a/coderd/database/queries/insights.sql +++ b/coderd/database/queries/insights.sql @@ -776,13 +776,17 @@ GROUP BY utp.num, utp.template_ids, utp.name, utp.type, utp.display_name, utp.de -- GetAccumulatedUsersInsights returns the accumulated number of users created -- in the given timeframe. It returns the accumulated number of users for each date -- within the specified timeframe, providing a running total of user sign-ups. +WITH RECURSIVE date_series AS ( + SELECT @start_time::timestamptz AS date + UNION ALL + SELECT date + INTERVAL '1 day' + FROM date_series + WHERE date + INTERVAL '1 day' <= @end_time::timestamptz +) SELECT - date_trunc('day', created_at)::date AS date, - COUNT(*) OVER (ORDER BY date_trunc('day', created_at)::date) AS total + d.date::date AS date, + (SELECT COUNT(*) FROM users u WHERE u.created_at <= d.date) AS total FROM - users -WHERE - created_at >= @start_time::timestamptz - AND created_at < @end_time::timestamptz + date_series d ORDER BY - date_trunc('day', created_at)::date; + d.date; From 7cd2d86fdb4e1ea069e72c33fb3fe3f553bfa936 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Wed, 13 Nov 2024 16:46:46 +0000 Subject: [PATCH 05/19] Fix in memory repro --- coderd/database/dbmem/dbmem.go | 21 +++++++-------------- coderd/insights.go | 13 ++++--------- 2 files changed, 11 insertions(+), 23 deletions(-) diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go index 5da5dcfb79826..557867e331fe9 100644 --- a/coderd/database/dbmem/dbmem.go +++ b/coderd/database/dbmem/dbmem.go @@ -2333,27 +2333,20 @@ func (q *FakeQuerier) GetAccumulatedUsersInsights(_ context.Context, arg databas newUsersByDate[date]++ } - accumulatedByDate := make(map[time.Time]int64) - var accumulatedCount int64 - sortedDates := make([]time.Time, 0, len(newUsersByDate)) - - for date := range newUsersByDate { - sortedDates = append(sortedDates, date) + dateSeries := make([]time.Time, 0) + for date := arg.StartTime.Truncate(24 * time.Hour); date.Before(arg.EndTime); date = date.AddDate(0, 0, 1) { + dateSeries = append(dateSeries, date) } - // Sort dates in ascending order - sort.Slice(sortedDates, func(i, j int) bool { - return sortedDates[i].Before(sortedDates[j]) - }) - - // Calculate accumulated count - for _, date := range sortedDates { + accumulatedByDate := make(map[time.Time]int64) + var accumulatedCount int64 + for _, date := range dateSeries { accumulatedCount += newUsersByDate[date] accumulatedByDate[date] = accumulatedCount } var rows []database.GetAccumulatedUsersInsightsRow - for _, date := range sortedDates { + for _, date := range dateSeries { rows = append(rows, database.GetAccumulatedUsersInsightsRow{ Date: date, Total: accumulatedByDate[date], diff --git a/coderd/insights.go b/coderd/insights.go index 1847900ee3472..5ca715dc77b45 100644 --- a/coderd/insights.go +++ b/coderd/insights.go @@ -519,18 +519,13 @@ func (api *API) insightsTotalUsers(rw http.ResponseWriter, r *http.Request) { } var res codersdk.TotalUsersInsightResponse - currentTotal := uint64(0) - for d := startTime; d.Before(endTime) || d.Equal(endTime); d = d.AddDate(0, 0, 1) { - for _, row := range rows { - if row.Date.Equal(d) { - currentTotal = uint64(row.Total) - } - } + for _, row := range rows { res = append(res, codersdk.TotalUserByDate{ - Date: d.Format(time.DateOnly), - Total: currentTotal, + Date: row.Date.Format(time.DateOnly), + Total: uint64(row.Total), }) } + httpapi.Write(r.Context(), rw, http.StatusOK, res) } From 442c75aca64c3220832d8bf55ef89838ce46a702 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Wed, 13 Nov 2024 17:05:50 +0000 Subject: [PATCH 06/19] Fix policy --- coderd/database/dbauthz/dbauthz.go | 2 +- coderd/rbac/object_gen.go | 1 - coderd/rbac/policy/policy.go | 2 -- coderd/rbac/roles_test.go | 2 +- codersdk/rbacresources_gen.go | 2 +- site/src/api/rbacresourcesGenerated.ts | 1 - 6 files changed, 3 insertions(+), 7 deletions(-) diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go index 49bd59c461eba..d0ad835a90074 100644 --- a/coderd/database/dbauthz/dbauthz.go +++ b/coderd/database/dbauthz/dbauthz.go @@ -1378,7 +1378,7 @@ func (q *querier) GetAPIKeysLastUsedAfter(ctx context.Context, lastUsed time.Tim } func (q *querier) GetAccumulatedUsersInsights(ctx context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { - if err := q.authorizeContext(ctx, policy.ActionViewInsights, rbac.ResourceUser); err != nil { + if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { return nil, err } return q.db.GetAccumulatedUsersInsights(ctx, arg) diff --git a/coderd/rbac/object_gen.go b/coderd/rbac/object_gen.go index e0730b97a08c1..12ac8a86a7dd9 100644 --- a/coderd/rbac/object_gen.go +++ b/coderd/rbac/object_gen.go @@ -269,7 +269,6 @@ var ( // - "ActionReadPersonal" :: read personal user data like user settings and auth links // - "ActionUpdate" :: update an existing user // - "ActionUpdatePersonal" :: update personal data - // - "ActionViewInsights" :: view insights ResourceUser = Object{ Type: "user", } diff --git a/coderd/rbac/policy/policy.go b/coderd/rbac/policy/policy.go index a678297abafa1..2691eed9fe0a9 100644 --- a/coderd/rbac/policy/policy.go +++ b/coderd/rbac/policy/policy.go @@ -84,8 +84,6 @@ var RBACPermissions = map[string]PermissionDefinition{ ActionReadPersonal: actDef("read personal user data like user settings and auth links"), ActionUpdatePersonal: actDef("update personal data"), - - ActionViewInsights: actDef("view insights"), }, }, "workspace": { diff --git a/coderd/rbac/roles_test.go b/coderd/rbac/roles_test.go index 0172439829063..7a4b0be5e229e 100644 --- a/coderd/rbac/roles_test.go +++ b/coderd/rbac/roles_test.go @@ -344,7 +344,7 @@ func TestRolePermissions(t *testing.T) { }, { Name: "UserData", - Actions: []policy.Action{policy.ActionReadPersonal, policy.ActionUpdatePersonal}, + Actions: []policy.Action{policy.ActionReadPersonal, policy.ActionUpdatePersonal, policy.ActionViewInsights}, Resource: rbac.ResourceUserObject(currentUser), AuthorizeMap: map[bool][]hasAuthSubjects{ true: {owner, orgMemberMe, memberMe, userAdmin}, diff --git a/codersdk/rbacresources_gen.go b/codersdk/rbacresources_gen.go index 0f78f013655a8..c903d5f8a02ff 100644 --- a/codersdk/rbacresources_gen.go +++ b/codersdk/rbacresources_gen.go @@ -87,7 +87,7 @@ var RBACResourceActions = map[RBACResource][]RBACAction{ ResourceSystem: {ActionCreate, ActionDelete, ActionRead, ActionUpdate}, ResourceTailnetCoordinator: {ActionCreate, ActionDelete, ActionRead, ActionUpdate}, ResourceTemplate: {ActionCreate, ActionDelete, ActionRead, ActionUpdate, ActionViewInsights}, - ResourceUser: {ActionCreate, ActionDelete, ActionRead, ActionReadPersonal, ActionUpdate, ActionUpdatePersonal, ActionViewInsights}, + ResourceUser: {ActionCreate, ActionDelete, ActionRead, ActionReadPersonal, ActionUpdate, ActionUpdatePersonal}, ResourceWorkspace: {ActionApplicationConnect, ActionCreate, ActionDelete, ActionRead, ActionSSH, ActionWorkspaceStart, ActionWorkspaceStop, ActionUpdate}, ResourceWorkspaceDormant: {ActionApplicationConnect, ActionCreate, ActionDelete, ActionRead, ActionSSH, ActionWorkspaceStart, ActionWorkspaceStop, ActionUpdate}, ResourceWorkspaceProxy: {ActionCreate, ActionDelete, ActionRead, ActionUpdate}, diff --git a/site/src/api/rbacresourcesGenerated.ts b/site/src/api/rbacresourcesGenerated.ts index a7f4033454471..4ce48121edbb3 100644 --- a/site/src/api/rbacresourcesGenerated.ts +++ b/site/src/api/rbacresourcesGenerated.ts @@ -153,7 +153,6 @@ export const RBACResourceActions: Partial< read_personal: "read personal user data like user settings and auth links", update: "update an existing user", update_personal: "update personal data", - view_insights: "view insights", }, workspace: { application_connect: "connect to workspace apps via browser", From bcf91f98bcd293dd9f48c0b387c3650b2ac8fd33 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Wed, 13 Nov 2024 18:01:29 +0000 Subject: [PATCH 07/19] Fix roles test --- coderd/rbac/roles_test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/coderd/rbac/roles_test.go b/coderd/rbac/roles_test.go index 7a4b0be5e229e..0172439829063 100644 --- a/coderd/rbac/roles_test.go +++ b/coderd/rbac/roles_test.go @@ -344,7 +344,7 @@ func TestRolePermissions(t *testing.T) { }, { Name: "UserData", - Actions: []policy.Action{policy.ActionReadPersonal, policy.ActionUpdatePersonal, policy.ActionViewInsights}, + Actions: []policy.Action{policy.ActionReadPersonal, policy.ActionUpdatePersonal}, Resource: rbac.ResourceUserObject(currentUser), AuthorizeMap: map[bool][]hasAuthSubjects{ true: {owner, orgMemberMe, memberMe, userAdmin}, From 70e66a877ccbfe857adba3c8a39dfc8dd32fa763 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Wed, 13 Nov 2024 18:30:16 +0000 Subject: [PATCH 08/19] Fix dbauthz test --- coderd/database/dbauthz/dbauthz_test.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go index b610efe0349f5..43ada9dbed069 100644 --- a/coderd/database/dbauthz/dbauthz_test.go +++ b/coderd/database/dbauthz/dbauthz_test.go @@ -2346,6 +2346,9 @@ func (s *MethodTestSuite) TestSystemFunctions() { dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{}) check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead) })) + s.Run("GetAccumulatedUsersInsights", s.Subtest(func(db database.Store, check *expects) { + check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Returns(int64(0)) + })) s.Run("GetActiveUserCount", s.Subtest(func(db database.Store, check *expects) { check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Returns(int64(0)) })) From 3da408eca04e66f9ec6eb6152d4c330f4a6715de Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Wed, 13 Nov 2024 18:37:35 +0000 Subject: [PATCH 09/19] Fix dbauthz test 2 --- coderd/database/dbauthz/dbauthz_test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go index 43ada9dbed069..48abf23fc741f 100644 --- a/coderd/database/dbauthz/dbauthz_test.go +++ b/coderd/database/dbauthz/dbauthz_test.go @@ -2347,7 +2347,7 @@ func (s *MethodTestSuite) TestSystemFunctions() { check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead) })) s.Run("GetAccumulatedUsersInsights", s.Subtest(func(db database.Store, check *expects) { - check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Returns(int64(0)) + check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead) })) s.Run("GetActiveUserCount", s.Subtest(func(db database.Store, check *expects) { check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Returns(int64(0)) From 639777f9d66cbbd696015d7d5c2f72bf1207f887 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Wed, 13 Nov 2024 18:45:39 +0000 Subject: [PATCH 10/19] Fix return at dbauthz --- coderd/database/dbauthz/dbauthz_test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go index 48abf23fc741f..eed6b5e4a219f 100644 --- a/coderd/database/dbauthz/dbauthz_test.go +++ b/coderd/database/dbauthz/dbauthz_test.go @@ -2347,7 +2347,7 @@ func (s *MethodTestSuite) TestSystemFunctions() { check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead) })) s.Run("GetAccumulatedUsersInsights", s.Subtest(func(db database.Store, check *expects) { - check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead) + check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Returns([]database.GetAccumulatedUsersInsightsRow{}, nil) })) s.Run("GetActiveUserCount", s.Subtest(func(db database.Store, check *expects) { check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Returns(int64(0)) From 520790d625e6620a3bc0fb774f7c49f0d09a1d2f Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Wed, 13 Nov 2024 18:54:12 +0000 Subject: [PATCH 11/19] Fix args --- coderd/database/dbauthz/dbauthz_test.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go index eed6b5e4a219f..b4419ae288340 100644 --- a/coderd/database/dbauthz/dbauthz_test.go +++ b/coderd/database/dbauthz/dbauthz_test.go @@ -2347,7 +2347,10 @@ func (s *MethodTestSuite) TestSystemFunctions() { check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead) })) s.Run("GetAccumulatedUsersInsights", s.Subtest(func(db database.Store, check *expects) { - check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Returns([]database.GetAccumulatedUsersInsightsRow{}, nil) + check.Args(database.GetAccumulatedUsersInsightsParams{ + StartTime: dbtime.Time(time.Now()), + EndTime: dbtime.Time(time.Now()), + }).Asserts(rbac.ResourceSystem, policy.ActionRead) })) s.Run("GetActiveUserCount", s.Subtest(func(db database.Store, check *expects) { check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Returns(int64(0)) From c011a051c9b6217c94e715ef6dd293281ab2fd7c Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Wed, 13 Nov 2024 19:02:42 +0000 Subject: [PATCH 12/19] Gen golden files --- ...ameters_two_dates_ago,_no_data.json.golden | 56 +++++++++++++++++++ 1 file changed, 56 insertions(+) create mode 100644 coderd/testdata/insights/template/parameters_two_dates_ago,_no_data.json.golden diff --git a/coderd/testdata/insights/template/parameters_two_dates_ago,_no_data.json.golden b/coderd/testdata/insights/template/parameters_two_dates_ago,_no_data.json.golden new file mode 100644 index 0000000000000..3d6328e3134a3 --- /dev/null +++ b/coderd/testdata/insights/template/parameters_two_dates_ago,_no_data.json.golden @@ -0,0 +1,56 @@ +{ + "report": { + "start_time": "0001-01-01T00:00:00Z", + "end_time": "0001-01-01T00:00:00Z", + "template_ids": [], + "active_users": 0, + "apps_usage": [ + { + "template_ids": [], + "type": "builtin", + "display_name": "Visual Studio Code", + "slug": "vscode", + "icon": "/icon/code.svg", + "seconds": 0, + "times_used": 0 + }, + { + "template_ids": [], + "type": "builtin", + "display_name": "JetBrains", + "slug": "jetbrains", + "icon": "/icon/intellij.svg", + "seconds": 0, + "times_used": 0 + }, + { + "template_ids": [], + "type": "builtin", + "display_name": "Web Terminal", + "slug": "reconnecting-pty", + "icon": "/icon/terminal.svg", + "seconds": 0, + "times_used": 0 + }, + { + "template_ids": [], + "type": "builtin", + "display_name": "SSH", + "slug": "ssh", + "icon": "/icon/terminal.svg", + "seconds": 0, + "times_used": 0 + }, + { + "template_ids": [], + "type": "builtin", + "display_name": "SFTP", + "slug": "sftp", + "icon": "/icon/terminal.svg", + "seconds": 0, + "times_used": 0 + } + ], + "parameters_usage": [] + } +} From 0e94c746329467b538c615fc3be3b8e3f5867268 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Thu, 14 Nov 2024 16:57:26 +0000 Subject: [PATCH 13/19] Pass the zone into the query --- coderd/database/dbauthz/dbauthz_test.go | 1 + coderd/database/queries.sql.go | 9 +++++---- coderd/database/queries/insights.sql | 2 +- coderd/insights.go | 2 ++ 4 files changed, 9 insertions(+), 5 deletions(-) diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go index 836a1fc780aca..cf37ab7b65823 100644 --- a/coderd/database/dbauthz/dbauthz_test.go +++ b/coderd/database/dbauthz/dbauthz_test.go @@ -2350,6 +2350,7 @@ func (s *MethodTestSuite) TestSystemFunctions() { check.Args(database.GetAccumulatedUsersInsightsParams{ StartTime: dbtime.Time(time.Now()), EndTime: dbtime.Time(time.Now()), + Timezone: "UTC", }).Asserts(rbac.ResourceSystem, policy.ActionRead) })) s.Run("GetActiveUserCount", s.Subtest(func(db database.Store, check *expects) { diff --git a/coderd/database/queries.sql.go b/coderd/database/queries.sql.go index 2b28b0af61c01..7dde2eed13ea5 100644 --- a/coderd/database/queries.sql.go +++ b/coderd/database/queries.sql.go @@ -2133,14 +2133,14 @@ func (q *sqlQuerier) UpdateGroupByID(ctx context.Context, arg UpdateGroupByIDPar const getAccumulatedUsersInsights = `-- name: GetAccumulatedUsersInsights :many WITH RECURSIVE date_series AS ( - SELECT $1::timestamptz AS date + SELECT $2::timestamptz AS date UNION ALL SELECT date + INTERVAL '1 day' FROM date_series - WHERE date + INTERVAL '1 day' <= $2::timestamptz + WHERE date + INTERVAL '1 day' <= $3::timestamptz ) SELECT - d.date::date AS date, + (d.date AT TIME ZONE $1::text)::date AS date, (SELECT COUNT(*) FROM users u WHERE u.created_at <= d.date) AS total FROM date_series d @@ -2149,6 +2149,7 @@ ORDER BY ` type GetAccumulatedUsersInsightsParams struct { + Timezone string `db:"timezone" json:"timezone"` StartTime time.Time `db:"start_time" json:"start_time"` EndTime time.Time `db:"end_time" json:"end_time"` } @@ -2162,7 +2163,7 @@ type GetAccumulatedUsersInsightsRow struct { // in the given timeframe. It returns the accumulated number of users for each date // within the specified timeframe, providing a running total of user sign-ups. func (q *sqlQuerier) GetAccumulatedUsersInsights(ctx context.Context, arg GetAccumulatedUsersInsightsParams) ([]GetAccumulatedUsersInsightsRow, error) { - rows, err := q.db.QueryContext(ctx, getAccumulatedUsersInsights, arg.StartTime, arg.EndTime) + rows, err := q.db.QueryContext(ctx, getAccumulatedUsersInsights, arg.Timezone, arg.StartTime, arg.EndTime) if err != nil { return nil, err } diff --git a/coderd/database/queries/insights.sql b/coderd/database/queries/insights.sql index 7495998b3723f..2256373071575 100644 --- a/coderd/database/queries/insights.sql +++ b/coderd/database/queries/insights.sql @@ -784,7 +784,7 @@ WITH RECURSIVE date_series AS ( WHERE date + INTERVAL '1 day' <= @end_time::timestamptz ) SELECT - d.date::date AS date, + (d.date AT TIME ZONE @timezone::text)::date AS date, (SELECT COUNT(*) FROM users u WHERE u.created_at <= d.date) AS total FROM date_series d diff --git a/coderd/insights.go b/coderd/insights.go index 5ca715dc77b45..7433d7ccdf732 100644 --- a/coderd/insights.go +++ b/coderd/insights.go @@ -506,9 +506,11 @@ func (api *API) insightsTotalUsers(rw http.ResponseWriter, r *http.Request) { return } + zone, _ := startTime.Zone() rows, err := api.Database.GetAccumulatedUsersInsights(ctx, database.GetAccumulatedUsersInsightsParams{ StartTime: startTime, EndTime: endTime, + Timezone: zone, }) if err != nil { httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{ From 186a95127dde0dfb97970c69bb8833d76ac89a96 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Thu, 14 Nov 2024 17:19:48 +0000 Subject: [PATCH 14/19] Fix tests --- coderd/insights_test.go | 26 +++++++++++++++++--------- 1 file changed, 17 insertions(+), 9 deletions(-) diff --git a/coderd/insights_test.go b/coderd/insights_test.go index 96d1900d595c4..6601d0066e6cb 100644 --- a/coderd/insights_test.go +++ b/coderd/insights_test.go @@ -2380,11 +2380,17 @@ func TestTotalUsersInsight(t *testing.T) { Username: "user3", CreatedAt: daysAgo(1), }) + dbgen.User(t, db, database.User{ + Email: "user4@coder.com", + Username: "user4", + CreatedAt: daysAgo(0), + }) // When: requesting the accumulated total of users by day + now := time.Now().UTC().Truncate(time.Hour).Add(time.Hour) res, err := client.TotalUsersInsight(ctx, codersdk.TotalUsersInsightRequest{ StartTime: daysAgo(2), - EndTime: today(), + EndTime: now, }) require.NoError(t, err) @@ -2400,7 +2406,7 @@ func TestTotalUsersInsight(t *testing.T) { require.Equal(t, res[1].Total, uint64(3)) // Third day - require.Equal(t, res[2].Date, today().Format(time.DateOnly)) + require.Equal(t, res[2].Date, daysAgo(0).Format(time.DateOnly)) require.Equal(t, res[2].Total, uint64(4)) }) @@ -2417,11 +2423,17 @@ func TestTotalUsersInsight(t *testing.T) { Username: "user1", CreatedAt: daysAgo(2), }) + dbgen.User(t, db, database.User{ + Email: "user2@coder.com", + Username: "user2", + CreatedAt: daysAgo(1), + }) // When: requesting the accumulated total of users for dates with no users created + now := time.Now().UTC().Truncate(time.Hour).Add(time.Hour) res, err := client.TotalUsersInsight(ctx, codersdk.TotalUsersInsightRequest{ StartTime: daysAgo(2).Truncate(24 * time.Hour), - EndTime: today(), + EndTime: now, }) require.NoError(t, err) @@ -2434,10 +2446,10 @@ func TestTotalUsersInsight(t *testing.T) { // Second day - no users created so should contain the same total as the previous day require.Equal(t, res[1].Date, daysAgo(1).Format(time.DateOnly)) - require.Equal(t, res[1].Total, uint64(1)) + require.Equal(t, res[1].Total, uint64(2)) // Third day - require.Equal(t, res[2].Date, today().Format(time.DateOnly)) + require.Equal(t, res[2].Date, daysAgo(0).Format(time.DateOnly)) require.Equal(t, res[2].Total, uint64(2)) }) } @@ -2445,7 +2457,3 @@ func TestTotalUsersInsight(t *testing.T) { func daysAgo(days int) time.Time { return time.Now().UTC().AddDate(0, 0, -days).Truncate(24 * time.Hour) } - -func today() time.Time { - return time.Now().UTC().Truncate(time.Hour).Add(time.Hour) -} From d340e2ddf0f8b20be49a4df97ccbedae0e2c384c Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Thu, 14 Nov 2024 17:23:22 +0000 Subject: [PATCH 15/19] Use offset instead of zone --- coderd/insights.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/coderd/insights.go b/coderd/insights.go index 7433d7ccdf732..a764b4149f60a 100644 --- a/coderd/insights.go +++ b/coderd/insights.go @@ -506,11 +506,11 @@ func (api *API) insightsTotalUsers(rw http.ResponseWriter, r *http.Request) { return } - zone, _ := startTime.Zone() + _, offset := startTime.Zone() rows, err := api.Database.GetAccumulatedUsersInsights(ctx, database.GetAccumulatedUsersInsightsParams{ StartTime: startTime, EndTime: endTime, - Timezone: zone, + Timezone: string(offset), }) if err != nil { httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{ From 82b6c36db09aae682b090f94499926a27cec46b9 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Thu, 14 Nov 2024 19:04:56 +0000 Subject: [PATCH 16/19] Fix offset to str --- coderd/insights.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/coderd/insights.go b/coderd/insights.go index a764b4149f60a..e70c362964e89 100644 --- a/coderd/insights.go +++ b/coderd/insights.go @@ -5,6 +5,7 @@ import ( "database/sql" "fmt" "net/http" + "strconv" "strings" "time" @@ -510,7 +511,7 @@ func (api *API) insightsTotalUsers(rw http.ResponseWriter, r *http.Request) { rows, err := api.Database.GetAccumulatedUsersInsights(ctx, database.GetAccumulatedUsersInsightsParams{ StartTime: startTime, EndTime: endTime, - Timezone: string(offset), + Timezone: strconv.Itoa(offset), }) if err != nil { httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{ From fac0c681e4ad442baaae1879ea1d854a93d74ef8 Mon Sep 17 00:00:00 2001 From: BrunoQuaresma Date: Fri, 22 Nov 2024 13:12:12 +0000 Subject: [PATCH 17/19] Fix tests with expected value correctly --- coderd/insights_test.go | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/coderd/insights_test.go b/coderd/insights_test.go index 66350b57938e5..caadb05d98381 100644 --- a/coderd/insights_test.go +++ b/coderd/insights_test.go @@ -2364,7 +2364,6 @@ func TestTotalUsersInsight(t *testing.T) { t.Cleanup(cancel) // Given: a deployment with many users - dbgen.User(t, db, database.User{ Email: "user1@coder.com", Username: "user1", @@ -2407,7 +2406,7 @@ func TestTotalUsersInsight(t *testing.T) { // Third day require.Equal(t, res[2].Date, daysAgo(0).Format(time.DateOnly)) - require.Equal(t, res[2].Total, uint64(4)) + require.Equal(t, res[2].Total, uint64(5)) }) t.Run("DatesWithNoUserRegistration", func(t *testing.T) { @@ -2450,7 +2449,7 @@ func TestTotalUsersInsight(t *testing.T) { // Third day require.Equal(t, res[2].Date, daysAgo(0).Format(time.DateOnly)) - require.Equal(t, res[2].Total, uint64(2)) + require.Equal(t, res[2].Total, uint64(3)) }) } From 9ba203c67001d45445241f0a8c70e1c63461dc10 Mon Sep 17 00:00:00 2001 From: Sas Swart Date: Tue, 26 Nov 2024 17:58:28 +0000 Subject: [PATCH 18/19] Remove GetAccumulatedUsersInsights for now --- coderd/apidoc/docs.go | 57 - coderd/apidoc/swagger.json | 53 - coderd/coderd.go | 1 - coderd/database/dbauthz/dbauthz.go | 7 - coderd/database/dbauthz/dbauthz_test.go | 7 - coderd/database/dbmem/dbmem.go | 35 - coderd/database/dbmetrics/querymetrics.go | 7 - coderd/database/dbmock/dbmock.go | 3112 ++++++++++----------- coderd/database/models.go | 2 +- coderd/database/querier.go | 6 +- coderd/database/queries.sql.go | 56 +- coderd/database/queries/insights.sql | 19 - coderd/insights.go | 62 - docs/reference/api/insights.md | 51 - docs/reference/api/schemas.md | 16 - 15 files changed, 1552 insertions(+), 1939 deletions(-) diff --git a/coderd/apidoc/docs.go b/coderd/apidoc/docs.go index 538098e41dec4..4c79534ef86ab 100644 --- a/coderd/apidoc/docs.go +++ b/coderd/apidoc/docs.go @@ -1292,52 +1292,6 @@ const docTemplate = `{ } } }, - "/insights/total-users": { - "get": { - "security": [ - { - "CoderSessionToken": [] - } - ], - "produces": [ - "application/json" - ], - "tags": [ - "Insights" - ], - "summary": "Get insights about total users", - "operationId": "get-insights-about-total-users", - "parameters": [ - { - "type": "string", - "format": "date-time", - "description": "Start time", - "name": "start_time", - "in": "query", - "required": true - }, - { - "type": "string", - "format": "date-time", - "description": "End time", - "name": "end_time", - "in": "query", - "required": true - } - ], - "responses": { - "200": { - "description": "OK", - "schema": { - "type": "array", - "items": { - "$ref": "#/definitions/codersdk.TotalUserByDate" - } - } - } - } - } - }, "/insights/user-activity": { "get": { "security": [ @@ -13844,17 +13798,6 @@ const docTemplate = `{ } } }, - "codersdk.TotalUserByDate": { - "type": "object", - "properties": { - "date": { - "type": "string" - }, - "total": { - "type": "integer" - } - } - }, "codersdk.TraceConfig": { "type": "object", "properties": { diff --git a/coderd/apidoc/swagger.json b/coderd/apidoc/swagger.json index 8b7cf19bfbde7..024d247fe5c83 100644 --- a/coderd/apidoc/swagger.json +++ b/coderd/apidoc/swagger.json @@ -1121,48 +1121,6 @@ } } }, - "/insights/total-users": { - "get": { - "security": [ - { - "CoderSessionToken": [] - } - ], - "produces": ["application/json"], - "tags": ["Insights"], - "summary": "Get insights about total users", - "operationId": "get-insights-about-total-users", - "parameters": [ - { - "type": "string", - "format": "date-time", - "description": "Start time", - "name": "start_time", - "in": "query", - "required": true - }, - { - "type": "string", - "format": "date-time", - "description": "End time", - "name": "end_time", - "in": "query", - "required": true - } - ], - "responses": { - "200": { - "description": "OK", - "schema": { - "type": "array", - "items": { - "$ref": "#/definitions/codersdk.TotalUserByDate" - } - } - } - } - } - }, "/insights/user-activity": { "get": { "security": [ @@ -12563,17 +12521,6 @@ } } }, - "codersdk.TotalUserByDate": { - "type": "object", - "properties": { - "date": { - "type": "string" - }, - "total": { - "type": "integer" - } - } - }, "codersdk.TraceConfig": { "type": "object", "properties": { diff --git a/coderd/coderd.go b/coderd/coderd.go index f42980d75157f..b5450b1387dfe 100644 --- a/coderd/coderd.go +++ b/coderd/coderd.go @@ -1281,7 +1281,6 @@ func New(options *Options) *API { r.Get("/user-activity", api.insightsUserActivity) r.Get("/user-latency", api.insightsUserLatency) r.Get("/templates", api.insightsTemplates) - r.Get("/total-users", api.insightsTotalUsers) }) r.Route("/debug", func(r chi.Router) { r.Use( diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go index db137f1e25b11..58c9179da5e4b 100644 --- a/coderd/database/dbauthz/dbauthz.go +++ b/coderd/database/dbauthz/dbauthz.go @@ -1377,13 +1377,6 @@ func (q *querier) GetAPIKeysLastUsedAfter(ctx context.Context, lastUsed time.Tim return fetchWithPostFilter(q.auth, policy.ActionRead, q.db.GetAPIKeysLastUsedAfter)(ctx, lastUsed) } -func (q *querier) GetAccumulatedUsersInsights(ctx context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { - if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { - return nil, err - } - return q.db.GetAccumulatedUsersInsights(ctx, arg) -} - func (q *querier) GetActiveUserCount(ctx context.Context) (int64, error) { if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil { return 0, err diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go index 0270d13c22f54..638829ae24ae5 100644 --- a/coderd/database/dbauthz/dbauthz_test.go +++ b/coderd/database/dbauthz/dbauthz_test.go @@ -2374,13 +2374,6 @@ func (s *MethodTestSuite) TestSystemFunctions() { dbgen.WorkspaceBuild(s.T(), db, database.WorkspaceBuild{}) check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead) })) - s.Run("GetAccumulatedUsersInsights", s.Subtest(func(db database.Store, check *expects) { - check.Args(database.GetAccumulatedUsersInsightsParams{ - StartTime: dbtime.Time(time.Now()), - EndTime: dbtime.Time(time.Now()), - Timezone: "UTC", - }).Asserts(rbac.ResourceSystem, policy.ActionRead) - })) s.Run("GetActiveUserCount", s.Subtest(func(db database.Store, check *expects) { check.Args().Asserts(rbac.ResourceSystem, policy.ActionRead).Returns(int64(0)) })) diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go index 1550019075f68..5583fff111648 100644 --- a/coderd/database/dbmem/dbmem.go +++ b/coderd/database/dbmem/dbmem.go @@ -2322,41 +2322,6 @@ func (q *FakeQuerier) GetAPIKeysLastUsedAfter(_ context.Context, after time.Time return apiKeys, nil } -func (q *FakeQuerier) GetAccumulatedUsersInsights(_ context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { - q.mutex.RLock() - defer q.mutex.RUnlock() - - newUsersByDate := make(map[time.Time]int64) - for _, user := range q.users { - if user.CreatedAt.Before(arg.StartTime) || user.CreatedAt.After(arg.EndTime) { - continue - } - date := user.CreatedAt.Truncate(24 * time.Hour) - newUsersByDate[date]++ - } - - dateSeries := make([]time.Time, 0) - for date := arg.StartTime.Truncate(24 * time.Hour); date.Before(arg.EndTime); date = date.AddDate(0, 0, 1) { - dateSeries = append(dateSeries, date) - } - - accumulatedByDate := make(map[time.Time]int64) - var accumulatedCount int64 - for _, date := range dateSeries { - accumulatedCount += newUsersByDate[date] - accumulatedByDate[date] = accumulatedCount - } - - var rows []database.GetAccumulatedUsersInsightsRow - for _, date := range dateSeries { - rows = append(rows, database.GetAccumulatedUsersInsightsRow{ - Date: date, - Total: accumulatedByDate[date], - }) - } - return rows, nil -} - func (q *FakeQuerier) GetActiveUserCount(_ context.Context) (int64, error) { q.mutex.RLock() defer q.mutex.RUnlock() diff --git a/coderd/database/dbmetrics/querymetrics.go b/coderd/database/dbmetrics/querymetrics.go index 2e6c2d1f82880..efde94488828f 100644 --- a/coderd/database/dbmetrics/querymetrics.go +++ b/coderd/database/dbmetrics/querymetrics.go @@ -462,13 +462,6 @@ func (m queryMetricsStore) GetAPIKeysLastUsedAfter(ctx context.Context, lastUsed return apiKeys, err } -func (m queryMetricsStore) GetAccumulatedUsersInsights(ctx context.Context, arg database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { - start := time.Now() - r0, r1 := m.s.GetAccumulatedUsersInsights(ctx, arg) - m.queryLatencies.WithLabelValues("GetAccumulatedUsersInsights").Observe(time.Since(start).Seconds()) - return r0, r1 -} - func (m queryMetricsStore) GetActiveUserCount(ctx context.Context) (int64, error) { start := time.Now() count, err := m.s.GetActiveUserCount(ctx) diff --git a/coderd/database/dbmock/dbmock.go b/coderd/database/dbmock/dbmock.go index 054ce7b721964..c78fc4fb6f384 100644 --- a/coderd/database/dbmock/dbmock.go +++ b/coderd/database/dbmock/dbmock.go @@ -24,6 +24,7 @@ import ( type MockStore struct { ctrl *gomock.Controller recorder *MockStoreMockRecorder + isgomock struct{} } // MockStoreMockRecorder is the mock recorder for MockStore. @@ -44,3522 +45,3507 @@ func (m *MockStore) EXPECT() *MockStoreMockRecorder { } // AcquireLock mocks base method. -func (m *MockStore) AcquireLock(arg0 context.Context, arg1 int64) error { +func (m *MockStore) AcquireLock(ctx context.Context, pgAdvisoryXactLock int64) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "AcquireLock", arg0, arg1) + ret := m.ctrl.Call(m, "AcquireLock", ctx, pgAdvisoryXactLock) ret0, _ := ret[0].(error) return ret0 } // AcquireLock indicates an expected call of AcquireLock. -func (mr *MockStoreMockRecorder) AcquireLock(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) AcquireLock(ctx, pgAdvisoryXactLock any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AcquireLock", reflect.TypeOf((*MockStore)(nil).AcquireLock), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AcquireLock", reflect.TypeOf((*MockStore)(nil).AcquireLock), ctx, pgAdvisoryXactLock) } // AcquireNotificationMessages mocks base method. -func (m *MockStore) AcquireNotificationMessages(arg0 context.Context, arg1 database.AcquireNotificationMessagesParams) ([]database.AcquireNotificationMessagesRow, error) { +func (m *MockStore) AcquireNotificationMessages(ctx context.Context, arg database.AcquireNotificationMessagesParams) ([]database.AcquireNotificationMessagesRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "AcquireNotificationMessages", arg0, arg1) + ret := m.ctrl.Call(m, "AcquireNotificationMessages", ctx, arg) ret0, _ := ret[0].([]database.AcquireNotificationMessagesRow) ret1, _ := ret[1].(error) return ret0, ret1 } // AcquireNotificationMessages indicates an expected call of AcquireNotificationMessages. -func (mr *MockStoreMockRecorder) AcquireNotificationMessages(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) AcquireNotificationMessages(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AcquireNotificationMessages", reflect.TypeOf((*MockStore)(nil).AcquireNotificationMessages), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AcquireNotificationMessages", reflect.TypeOf((*MockStore)(nil).AcquireNotificationMessages), ctx, arg) } // AcquireProvisionerJob mocks base method. -func (m *MockStore) AcquireProvisionerJob(arg0 context.Context, arg1 database.AcquireProvisionerJobParams) (database.ProvisionerJob, error) { +func (m *MockStore) AcquireProvisionerJob(ctx context.Context, arg database.AcquireProvisionerJobParams) (database.ProvisionerJob, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "AcquireProvisionerJob", arg0, arg1) + ret := m.ctrl.Call(m, "AcquireProvisionerJob", ctx, arg) ret0, _ := ret[0].(database.ProvisionerJob) ret1, _ := ret[1].(error) return ret0, ret1 } // AcquireProvisionerJob indicates an expected call of AcquireProvisionerJob. -func (mr *MockStoreMockRecorder) AcquireProvisionerJob(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) AcquireProvisionerJob(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AcquireProvisionerJob", reflect.TypeOf((*MockStore)(nil).AcquireProvisionerJob), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AcquireProvisionerJob", reflect.TypeOf((*MockStore)(nil).AcquireProvisionerJob), ctx, arg) } // ActivityBumpWorkspace mocks base method. -func (m *MockStore) ActivityBumpWorkspace(arg0 context.Context, arg1 database.ActivityBumpWorkspaceParams) error { +func (m *MockStore) ActivityBumpWorkspace(ctx context.Context, arg database.ActivityBumpWorkspaceParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ActivityBumpWorkspace", arg0, arg1) + ret := m.ctrl.Call(m, "ActivityBumpWorkspace", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // ActivityBumpWorkspace indicates an expected call of ActivityBumpWorkspace. -func (mr *MockStoreMockRecorder) ActivityBumpWorkspace(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) ActivityBumpWorkspace(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ActivityBumpWorkspace", reflect.TypeOf((*MockStore)(nil).ActivityBumpWorkspace), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ActivityBumpWorkspace", reflect.TypeOf((*MockStore)(nil).ActivityBumpWorkspace), ctx, arg) } // AllUserIDs mocks base method. -func (m *MockStore) AllUserIDs(arg0 context.Context) ([]uuid.UUID, error) { +func (m *MockStore) AllUserIDs(ctx context.Context) ([]uuid.UUID, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "AllUserIDs", arg0) + ret := m.ctrl.Call(m, "AllUserIDs", ctx) ret0, _ := ret[0].([]uuid.UUID) ret1, _ := ret[1].(error) return ret0, ret1 } // AllUserIDs indicates an expected call of AllUserIDs. -func (mr *MockStoreMockRecorder) AllUserIDs(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) AllUserIDs(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AllUserIDs", reflect.TypeOf((*MockStore)(nil).AllUserIDs), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AllUserIDs", reflect.TypeOf((*MockStore)(nil).AllUserIDs), ctx) } // ArchiveUnusedTemplateVersions mocks base method. -func (m *MockStore) ArchiveUnusedTemplateVersions(arg0 context.Context, arg1 database.ArchiveUnusedTemplateVersionsParams) ([]uuid.UUID, error) { +func (m *MockStore) ArchiveUnusedTemplateVersions(ctx context.Context, arg database.ArchiveUnusedTemplateVersionsParams) ([]uuid.UUID, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ArchiveUnusedTemplateVersions", arg0, arg1) + ret := m.ctrl.Call(m, "ArchiveUnusedTemplateVersions", ctx, arg) ret0, _ := ret[0].([]uuid.UUID) ret1, _ := ret[1].(error) return ret0, ret1 } // ArchiveUnusedTemplateVersions indicates an expected call of ArchiveUnusedTemplateVersions. -func (mr *MockStoreMockRecorder) ArchiveUnusedTemplateVersions(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) ArchiveUnusedTemplateVersions(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ArchiveUnusedTemplateVersions", reflect.TypeOf((*MockStore)(nil).ArchiveUnusedTemplateVersions), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ArchiveUnusedTemplateVersions", reflect.TypeOf((*MockStore)(nil).ArchiveUnusedTemplateVersions), ctx, arg) } // BatchUpdateWorkspaceLastUsedAt mocks base method. -func (m *MockStore) BatchUpdateWorkspaceLastUsedAt(arg0 context.Context, arg1 database.BatchUpdateWorkspaceLastUsedAtParams) error { +func (m *MockStore) BatchUpdateWorkspaceLastUsedAt(ctx context.Context, arg database.BatchUpdateWorkspaceLastUsedAtParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BatchUpdateWorkspaceLastUsedAt", arg0, arg1) + ret := m.ctrl.Call(m, "BatchUpdateWorkspaceLastUsedAt", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // BatchUpdateWorkspaceLastUsedAt indicates an expected call of BatchUpdateWorkspaceLastUsedAt. -func (mr *MockStoreMockRecorder) BatchUpdateWorkspaceLastUsedAt(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) BatchUpdateWorkspaceLastUsedAt(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchUpdateWorkspaceLastUsedAt", reflect.TypeOf((*MockStore)(nil).BatchUpdateWorkspaceLastUsedAt), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchUpdateWorkspaceLastUsedAt", reflect.TypeOf((*MockStore)(nil).BatchUpdateWorkspaceLastUsedAt), ctx, arg) } // BulkMarkNotificationMessagesFailed mocks base method. -func (m *MockStore) BulkMarkNotificationMessagesFailed(arg0 context.Context, arg1 database.BulkMarkNotificationMessagesFailedParams) (int64, error) { +func (m *MockStore) BulkMarkNotificationMessagesFailed(ctx context.Context, arg database.BulkMarkNotificationMessagesFailedParams) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BulkMarkNotificationMessagesFailed", arg0, arg1) + ret := m.ctrl.Call(m, "BulkMarkNotificationMessagesFailed", ctx, arg) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // BulkMarkNotificationMessagesFailed indicates an expected call of BulkMarkNotificationMessagesFailed. -func (mr *MockStoreMockRecorder) BulkMarkNotificationMessagesFailed(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) BulkMarkNotificationMessagesFailed(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkMarkNotificationMessagesFailed", reflect.TypeOf((*MockStore)(nil).BulkMarkNotificationMessagesFailed), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkMarkNotificationMessagesFailed", reflect.TypeOf((*MockStore)(nil).BulkMarkNotificationMessagesFailed), ctx, arg) } // BulkMarkNotificationMessagesSent mocks base method. -func (m *MockStore) BulkMarkNotificationMessagesSent(arg0 context.Context, arg1 database.BulkMarkNotificationMessagesSentParams) (int64, error) { +func (m *MockStore) BulkMarkNotificationMessagesSent(ctx context.Context, arg database.BulkMarkNotificationMessagesSentParams) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "BulkMarkNotificationMessagesSent", arg0, arg1) + ret := m.ctrl.Call(m, "BulkMarkNotificationMessagesSent", ctx, arg) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // BulkMarkNotificationMessagesSent indicates an expected call of BulkMarkNotificationMessagesSent. -func (mr *MockStoreMockRecorder) BulkMarkNotificationMessagesSent(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) BulkMarkNotificationMessagesSent(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkMarkNotificationMessagesSent", reflect.TypeOf((*MockStore)(nil).BulkMarkNotificationMessagesSent), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BulkMarkNotificationMessagesSent", reflect.TypeOf((*MockStore)(nil).BulkMarkNotificationMessagesSent), ctx, arg) } // CleanTailnetCoordinators mocks base method. -func (m *MockStore) CleanTailnetCoordinators(arg0 context.Context) error { +func (m *MockStore) CleanTailnetCoordinators(ctx context.Context) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CleanTailnetCoordinators", arg0) + ret := m.ctrl.Call(m, "CleanTailnetCoordinators", ctx) ret0, _ := ret[0].(error) return ret0 } // CleanTailnetCoordinators indicates an expected call of CleanTailnetCoordinators. -func (mr *MockStoreMockRecorder) CleanTailnetCoordinators(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) CleanTailnetCoordinators(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanTailnetCoordinators", reflect.TypeOf((*MockStore)(nil).CleanTailnetCoordinators), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanTailnetCoordinators", reflect.TypeOf((*MockStore)(nil).CleanTailnetCoordinators), ctx) } // CleanTailnetLostPeers mocks base method. -func (m *MockStore) CleanTailnetLostPeers(arg0 context.Context) error { +func (m *MockStore) CleanTailnetLostPeers(ctx context.Context) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CleanTailnetLostPeers", arg0) + ret := m.ctrl.Call(m, "CleanTailnetLostPeers", ctx) ret0, _ := ret[0].(error) return ret0 } // CleanTailnetLostPeers indicates an expected call of CleanTailnetLostPeers. -func (mr *MockStoreMockRecorder) CleanTailnetLostPeers(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) CleanTailnetLostPeers(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanTailnetLostPeers", reflect.TypeOf((*MockStore)(nil).CleanTailnetLostPeers), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanTailnetLostPeers", reflect.TypeOf((*MockStore)(nil).CleanTailnetLostPeers), ctx) } // CleanTailnetTunnels mocks base method. -func (m *MockStore) CleanTailnetTunnels(arg0 context.Context) error { +func (m *MockStore) CleanTailnetTunnels(ctx context.Context) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CleanTailnetTunnels", arg0) + ret := m.ctrl.Call(m, "CleanTailnetTunnels", ctx) ret0, _ := ret[0].(error) return ret0 } // CleanTailnetTunnels indicates an expected call of CleanTailnetTunnels. -func (mr *MockStoreMockRecorder) CleanTailnetTunnels(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) CleanTailnetTunnels(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanTailnetTunnels", reflect.TypeOf((*MockStore)(nil).CleanTailnetTunnels), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CleanTailnetTunnels", reflect.TypeOf((*MockStore)(nil).CleanTailnetTunnels), ctx) } // CustomRoles mocks base method. -func (m *MockStore) CustomRoles(arg0 context.Context, arg1 database.CustomRolesParams) ([]database.CustomRole, error) { +func (m *MockStore) CustomRoles(ctx context.Context, arg database.CustomRolesParams) ([]database.CustomRole, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "CustomRoles", arg0, arg1) + ret := m.ctrl.Call(m, "CustomRoles", ctx, arg) ret0, _ := ret[0].([]database.CustomRole) ret1, _ := ret[1].(error) return ret0, ret1 } // CustomRoles indicates an expected call of CustomRoles. -func (mr *MockStoreMockRecorder) CustomRoles(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) CustomRoles(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CustomRoles", reflect.TypeOf((*MockStore)(nil).CustomRoles), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CustomRoles", reflect.TypeOf((*MockStore)(nil).CustomRoles), ctx, arg) } // DeleteAPIKeyByID mocks base method. -func (m *MockStore) DeleteAPIKeyByID(arg0 context.Context, arg1 string) error { +func (m *MockStore) DeleteAPIKeyByID(ctx context.Context, id string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteAPIKeyByID", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteAPIKeyByID", ctx, id) ret0, _ := ret[0].(error) return ret0 } // DeleteAPIKeyByID indicates an expected call of DeleteAPIKeyByID. -func (mr *MockStoreMockRecorder) DeleteAPIKeyByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteAPIKeyByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAPIKeyByID", reflect.TypeOf((*MockStore)(nil).DeleteAPIKeyByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAPIKeyByID", reflect.TypeOf((*MockStore)(nil).DeleteAPIKeyByID), ctx, id) } // DeleteAPIKeysByUserID mocks base method. -func (m *MockStore) DeleteAPIKeysByUserID(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteAPIKeysByUserID(ctx context.Context, userID uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteAPIKeysByUserID", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteAPIKeysByUserID", ctx, userID) ret0, _ := ret[0].(error) return ret0 } // DeleteAPIKeysByUserID indicates an expected call of DeleteAPIKeysByUserID. -func (mr *MockStoreMockRecorder) DeleteAPIKeysByUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteAPIKeysByUserID(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAPIKeysByUserID", reflect.TypeOf((*MockStore)(nil).DeleteAPIKeysByUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAPIKeysByUserID", reflect.TypeOf((*MockStore)(nil).DeleteAPIKeysByUserID), ctx, userID) } // DeleteAllTailnetClientSubscriptions mocks base method. -func (m *MockStore) DeleteAllTailnetClientSubscriptions(arg0 context.Context, arg1 database.DeleteAllTailnetClientSubscriptionsParams) error { +func (m *MockStore) DeleteAllTailnetClientSubscriptions(ctx context.Context, arg database.DeleteAllTailnetClientSubscriptionsParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteAllTailnetClientSubscriptions", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteAllTailnetClientSubscriptions", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteAllTailnetClientSubscriptions indicates an expected call of DeleteAllTailnetClientSubscriptions. -func (mr *MockStoreMockRecorder) DeleteAllTailnetClientSubscriptions(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteAllTailnetClientSubscriptions(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAllTailnetClientSubscriptions", reflect.TypeOf((*MockStore)(nil).DeleteAllTailnetClientSubscriptions), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAllTailnetClientSubscriptions", reflect.TypeOf((*MockStore)(nil).DeleteAllTailnetClientSubscriptions), ctx, arg) } // DeleteAllTailnetTunnels mocks base method. -func (m *MockStore) DeleteAllTailnetTunnels(arg0 context.Context, arg1 database.DeleteAllTailnetTunnelsParams) error { +func (m *MockStore) DeleteAllTailnetTunnels(ctx context.Context, arg database.DeleteAllTailnetTunnelsParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteAllTailnetTunnels", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteAllTailnetTunnels", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteAllTailnetTunnels indicates an expected call of DeleteAllTailnetTunnels. -func (mr *MockStoreMockRecorder) DeleteAllTailnetTunnels(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteAllTailnetTunnels(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAllTailnetTunnels", reflect.TypeOf((*MockStore)(nil).DeleteAllTailnetTunnels), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteAllTailnetTunnels", reflect.TypeOf((*MockStore)(nil).DeleteAllTailnetTunnels), ctx, arg) } // DeleteApplicationConnectAPIKeysByUserID mocks base method. -func (m *MockStore) DeleteApplicationConnectAPIKeysByUserID(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteApplicationConnectAPIKeysByUserID(ctx context.Context, userID uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteApplicationConnectAPIKeysByUserID", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteApplicationConnectAPIKeysByUserID", ctx, userID) ret0, _ := ret[0].(error) return ret0 } // DeleteApplicationConnectAPIKeysByUserID indicates an expected call of DeleteApplicationConnectAPIKeysByUserID. -func (mr *MockStoreMockRecorder) DeleteApplicationConnectAPIKeysByUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteApplicationConnectAPIKeysByUserID(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteApplicationConnectAPIKeysByUserID", reflect.TypeOf((*MockStore)(nil).DeleteApplicationConnectAPIKeysByUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteApplicationConnectAPIKeysByUserID", reflect.TypeOf((*MockStore)(nil).DeleteApplicationConnectAPIKeysByUserID), ctx, userID) } // DeleteCoordinator mocks base method. -func (m *MockStore) DeleteCoordinator(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteCoordinator(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteCoordinator", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteCoordinator", ctx, id) ret0, _ := ret[0].(error) return ret0 } // DeleteCoordinator indicates an expected call of DeleteCoordinator. -func (mr *MockStoreMockRecorder) DeleteCoordinator(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteCoordinator(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCoordinator", reflect.TypeOf((*MockStore)(nil).DeleteCoordinator), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCoordinator", reflect.TypeOf((*MockStore)(nil).DeleteCoordinator), ctx, id) } // DeleteCryptoKey mocks base method. -func (m *MockStore) DeleteCryptoKey(arg0 context.Context, arg1 database.DeleteCryptoKeyParams) (database.CryptoKey, error) { +func (m *MockStore) DeleteCryptoKey(ctx context.Context, arg database.DeleteCryptoKeyParams) (database.CryptoKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteCryptoKey", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteCryptoKey", ctx, arg) ret0, _ := ret[0].(database.CryptoKey) ret1, _ := ret[1].(error) return ret0, ret1 } // DeleteCryptoKey indicates an expected call of DeleteCryptoKey. -func (mr *MockStoreMockRecorder) DeleteCryptoKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteCryptoKey(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCryptoKey", reflect.TypeOf((*MockStore)(nil).DeleteCryptoKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCryptoKey", reflect.TypeOf((*MockStore)(nil).DeleteCryptoKey), ctx, arg) } // DeleteCustomRole mocks base method. -func (m *MockStore) DeleteCustomRole(arg0 context.Context, arg1 database.DeleteCustomRoleParams) error { +func (m *MockStore) DeleteCustomRole(ctx context.Context, arg database.DeleteCustomRoleParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteCustomRole", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteCustomRole", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteCustomRole indicates an expected call of DeleteCustomRole. -func (mr *MockStoreMockRecorder) DeleteCustomRole(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteCustomRole(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCustomRole", reflect.TypeOf((*MockStore)(nil).DeleteCustomRole), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteCustomRole", reflect.TypeOf((*MockStore)(nil).DeleteCustomRole), ctx, arg) } // DeleteExternalAuthLink mocks base method. -func (m *MockStore) DeleteExternalAuthLink(arg0 context.Context, arg1 database.DeleteExternalAuthLinkParams) error { +func (m *MockStore) DeleteExternalAuthLink(ctx context.Context, arg database.DeleteExternalAuthLinkParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteExternalAuthLink", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteExternalAuthLink", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteExternalAuthLink indicates an expected call of DeleteExternalAuthLink. -func (mr *MockStoreMockRecorder) DeleteExternalAuthLink(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteExternalAuthLink(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteExternalAuthLink", reflect.TypeOf((*MockStore)(nil).DeleteExternalAuthLink), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteExternalAuthLink", reflect.TypeOf((*MockStore)(nil).DeleteExternalAuthLink), ctx, arg) } // DeleteGitSSHKey mocks base method. -func (m *MockStore) DeleteGitSSHKey(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteGitSSHKey(ctx context.Context, userID uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteGitSSHKey", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteGitSSHKey", ctx, userID) ret0, _ := ret[0].(error) return ret0 } // DeleteGitSSHKey indicates an expected call of DeleteGitSSHKey. -func (mr *MockStoreMockRecorder) DeleteGitSSHKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteGitSSHKey(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteGitSSHKey", reflect.TypeOf((*MockStore)(nil).DeleteGitSSHKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteGitSSHKey", reflect.TypeOf((*MockStore)(nil).DeleteGitSSHKey), ctx, userID) } // DeleteGroupByID mocks base method. -func (m *MockStore) DeleteGroupByID(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteGroupByID(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteGroupByID", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteGroupByID", ctx, id) ret0, _ := ret[0].(error) return ret0 } // DeleteGroupByID indicates an expected call of DeleteGroupByID. -func (mr *MockStoreMockRecorder) DeleteGroupByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteGroupByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteGroupByID", reflect.TypeOf((*MockStore)(nil).DeleteGroupByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteGroupByID", reflect.TypeOf((*MockStore)(nil).DeleteGroupByID), ctx, id) } // DeleteGroupMemberFromGroup mocks base method. -func (m *MockStore) DeleteGroupMemberFromGroup(arg0 context.Context, arg1 database.DeleteGroupMemberFromGroupParams) error { +func (m *MockStore) DeleteGroupMemberFromGroup(ctx context.Context, arg database.DeleteGroupMemberFromGroupParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteGroupMemberFromGroup", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteGroupMemberFromGroup", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteGroupMemberFromGroup indicates an expected call of DeleteGroupMemberFromGroup. -func (mr *MockStoreMockRecorder) DeleteGroupMemberFromGroup(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteGroupMemberFromGroup(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteGroupMemberFromGroup", reflect.TypeOf((*MockStore)(nil).DeleteGroupMemberFromGroup), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteGroupMemberFromGroup", reflect.TypeOf((*MockStore)(nil).DeleteGroupMemberFromGroup), ctx, arg) } // DeleteLicense mocks base method. -func (m *MockStore) DeleteLicense(arg0 context.Context, arg1 int32) (int32, error) { +func (m *MockStore) DeleteLicense(ctx context.Context, id int32) (int32, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteLicense", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteLicense", ctx, id) ret0, _ := ret[0].(int32) ret1, _ := ret[1].(error) return ret0, ret1 } // DeleteLicense indicates an expected call of DeleteLicense. -func (mr *MockStoreMockRecorder) DeleteLicense(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteLicense(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteLicense", reflect.TypeOf((*MockStore)(nil).DeleteLicense), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteLicense", reflect.TypeOf((*MockStore)(nil).DeleteLicense), ctx, id) } // DeleteOAuth2ProviderAppByID mocks base method. -func (m *MockStore) DeleteOAuth2ProviderAppByID(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteOAuth2ProviderAppByID(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppByID", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppByID", ctx, id) ret0, _ := ret[0].(error) return ret0 } // DeleteOAuth2ProviderAppByID indicates an expected call of DeleteOAuth2ProviderAppByID. -func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppByID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppByID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppByID), ctx, id) } // DeleteOAuth2ProviderAppCodeByID mocks base method. -func (m *MockStore) DeleteOAuth2ProviderAppCodeByID(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteOAuth2ProviderAppCodeByID(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppCodeByID", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppCodeByID", ctx, id) ret0, _ := ret[0].(error) return ret0 } // DeleteOAuth2ProviderAppCodeByID indicates an expected call of DeleteOAuth2ProviderAppCodeByID. -func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppCodeByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppCodeByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppCodeByID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppCodeByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppCodeByID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppCodeByID), ctx, id) } // DeleteOAuth2ProviderAppCodesByAppAndUserID mocks base method. -func (m *MockStore) DeleteOAuth2ProviderAppCodesByAppAndUserID(arg0 context.Context, arg1 database.DeleteOAuth2ProviderAppCodesByAppAndUserIDParams) error { +func (m *MockStore) DeleteOAuth2ProviderAppCodesByAppAndUserID(ctx context.Context, arg database.DeleteOAuth2ProviderAppCodesByAppAndUserIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppCodesByAppAndUserID", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppCodesByAppAndUserID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteOAuth2ProviderAppCodesByAppAndUserID indicates an expected call of DeleteOAuth2ProviderAppCodesByAppAndUserID. -func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppCodesByAppAndUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppCodesByAppAndUserID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppCodesByAppAndUserID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppCodesByAppAndUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppCodesByAppAndUserID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppCodesByAppAndUserID), ctx, arg) } // DeleteOAuth2ProviderAppSecretByID mocks base method. -func (m *MockStore) DeleteOAuth2ProviderAppSecretByID(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteOAuth2ProviderAppSecretByID(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppSecretByID", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppSecretByID", ctx, id) ret0, _ := ret[0].(error) return ret0 } // DeleteOAuth2ProviderAppSecretByID indicates an expected call of DeleteOAuth2ProviderAppSecretByID. -func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppSecretByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppSecretByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppSecretByID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppSecretByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppSecretByID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppSecretByID), ctx, id) } // DeleteOAuth2ProviderAppTokensByAppAndUserID mocks base method. -func (m *MockStore) DeleteOAuth2ProviderAppTokensByAppAndUserID(arg0 context.Context, arg1 database.DeleteOAuth2ProviderAppTokensByAppAndUserIDParams) error { +func (m *MockStore) DeleteOAuth2ProviderAppTokensByAppAndUserID(ctx context.Context, arg database.DeleteOAuth2ProviderAppTokensByAppAndUserIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppTokensByAppAndUserID", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteOAuth2ProviderAppTokensByAppAndUserID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteOAuth2ProviderAppTokensByAppAndUserID indicates an expected call of DeleteOAuth2ProviderAppTokensByAppAndUserID. -func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppTokensByAppAndUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOAuth2ProviderAppTokensByAppAndUserID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppTokensByAppAndUserID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppTokensByAppAndUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOAuth2ProviderAppTokensByAppAndUserID", reflect.TypeOf((*MockStore)(nil).DeleteOAuth2ProviderAppTokensByAppAndUserID), ctx, arg) } // DeleteOldNotificationMessages mocks base method. -func (m *MockStore) DeleteOldNotificationMessages(arg0 context.Context) error { +func (m *MockStore) DeleteOldNotificationMessages(ctx context.Context) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOldNotificationMessages", arg0) + ret := m.ctrl.Call(m, "DeleteOldNotificationMessages", ctx) ret0, _ := ret[0].(error) return ret0 } // DeleteOldNotificationMessages indicates an expected call of DeleteOldNotificationMessages. -func (mr *MockStoreMockRecorder) DeleteOldNotificationMessages(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOldNotificationMessages(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOldNotificationMessages", reflect.TypeOf((*MockStore)(nil).DeleteOldNotificationMessages), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOldNotificationMessages", reflect.TypeOf((*MockStore)(nil).DeleteOldNotificationMessages), ctx) } // DeleteOldProvisionerDaemons mocks base method. -func (m *MockStore) DeleteOldProvisionerDaemons(arg0 context.Context) error { +func (m *MockStore) DeleteOldProvisionerDaemons(ctx context.Context) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOldProvisionerDaemons", arg0) + ret := m.ctrl.Call(m, "DeleteOldProvisionerDaemons", ctx) ret0, _ := ret[0].(error) return ret0 } // DeleteOldProvisionerDaemons indicates an expected call of DeleteOldProvisionerDaemons. -func (mr *MockStoreMockRecorder) DeleteOldProvisionerDaemons(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOldProvisionerDaemons(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOldProvisionerDaemons", reflect.TypeOf((*MockStore)(nil).DeleteOldProvisionerDaemons), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOldProvisionerDaemons", reflect.TypeOf((*MockStore)(nil).DeleteOldProvisionerDaemons), ctx) } // DeleteOldWorkspaceAgentLogs mocks base method. -func (m *MockStore) DeleteOldWorkspaceAgentLogs(arg0 context.Context, arg1 time.Time) error { +func (m *MockStore) DeleteOldWorkspaceAgentLogs(ctx context.Context, threshold time.Time) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOldWorkspaceAgentLogs", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteOldWorkspaceAgentLogs", ctx, threshold) ret0, _ := ret[0].(error) return ret0 } // DeleteOldWorkspaceAgentLogs indicates an expected call of DeleteOldWorkspaceAgentLogs. -func (mr *MockStoreMockRecorder) DeleteOldWorkspaceAgentLogs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOldWorkspaceAgentLogs(ctx, threshold any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOldWorkspaceAgentLogs", reflect.TypeOf((*MockStore)(nil).DeleteOldWorkspaceAgentLogs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOldWorkspaceAgentLogs", reflect.TypeOf((*MockStore)(nil).DeleteOldWorkspaceAgentLogs), ctx, threshold) } // DeleteOldWorkspaceAgentStats mocks base method. -func (m *MockStore) DeleteOldWorkspaceAgentStats(arg0 context.Context) error { +func (m *MockStore) DeleteOldWorkspaceAgentStats(ctx context.Context) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOldWorkspaceAgentStats", arg0) + ret := m.ctrl.Call(m, "DeleteOldWorkspaceAgentStats", ctx) ret0, _ := ret[0].(error) return ret0 } // DeleteOldWorkspaceAgentStats indicates an expected call of DeleteOldWorkspaceAgentStats. -func (mr *MockStoreMockRecorder) DeleteOldWorkspaceAgentStats(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOldWorkspaceAgentStats(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOldWorkspaceAgentStats", reflect.TypeOf((*MockStore)(nil).DeleteOldWorkspaceAgentStats), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOldWorkspaceAgentStats", reflect.TypeOf((*MockStore)(nil).DeleteOldWorkspaceAgentStats), ctx) } // DeleteOrganization mocks base method. -func (m *MockStore) DeleteOrganization(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteOrganization(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOrganization", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteOrganization", ctx, id) ret0, _ := ret[0].(error) return ret0 } // DeleteOrganization indicates an expected call of DeleteOrganization. -func (mr *MockStoreMockRecorder) DeleteOrganization(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOrganization(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOrganization", reflect.TypeOf((*MockStore)(nil).DeleteOrganization), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOrganization", reflect.TypeOf((*MockStore)(nil).DeleteOrganization), ctx, id) } // DeleteOrganizationMember mocks base method. -func (m *MockStore) DeleteOrganizationMember(arg0 context.Context, arg1 database.DeleteOrganizationMemberParams) error { +func (m *MockStore) DeleteOrganizationMember(ctx context.Context, arg database.DeleteOrganizationMemberParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteOrganizationMember", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteOrganizationMember", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteOrganizationMember indicates an expected call of DeleteOrganizationMember. -func (mr *MockStoreMockRecorder) DeleteOrganizationMember(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteOrganizationMember(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOrganizationMember", reflect.TypeOf((*MockStore)(nil).DeleteOrganizationMember), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOrganizationMember", reflect.TypeOf((*MockStore)(nil).DeleteOrganizationMember), ctx, arg) } // DeleteProvisionerKey mocks base method. -func (m *MockStore) DeleteProvisionerKey(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteProvisionerKey(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteProvisionerKey", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteProvisionerKey", ctx, id) ret0, _ := ret[0].(error) return ret0 } // DeleteProvisionerKey indicates an expected call of DeleteProvisionerKey. -func (mr *MockStoreMockRecorder) DeleteProvisionerKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteProvisionerKey(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteProvisionerKey", reflect.TypeOf((*MockStore)(nil).DeleteProvisionerKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteProvisionerKey", reflect.TypeOf((*MockStore)(nil).DeleteProvisionerKey), ctx, id) } // DeleteReplicasUpdatedBefore mocks base method. -func (m *MockStore) DeleteReplicasUpdatedBefore(arg0 context.Context, arg1 time.Time) error { +func (m *MockStore) DeleteReplicasUpdatedBefore(ctx context.Context, updatedAt time.Time) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteReplicasUpdatedBefore", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteReplicasUpdatedBefore", ctx, updatedAt) ret0, _ := ret[0].(error) return ret0 } // DeleteReplicasUpdatedBefore indicates an expected call of DeleteReplicasUpdatedBefore. -func (mr *MockStoreMockRecorder) DeleteReplicasUpdatedBefore(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteReplicasUpdatedBefore(ctx, updatedAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteReplicasUpdatedBefore", reflect.TypeOf((*MockStore)(nil).DeleteReplicasUpdatedBefore), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteReplicasUpdatedBefore", reflect.TypeOf((*MockStore)(nil).DeleteReplicasUpdatedBefore), ctx, updatedAt) } // DeleteRuntimeConfig mocks base method. -func (m *MockStore) DeleteRuntimeConfig(arg0 context.Context, arg1 string) error { +func (m *MockStore) DeleteRuntimeConfig(ctx context.Context, key string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteRuntimeConfig", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteRuntimeConfig", ctx, key) ret0, _ := ret[0].(error) return ret0 } // DeleteRuntimeConfig indicates an expected call of DeleteRuntimeConfig. -func (mr *MockStoreMockRecorder) DeleteRuntimeConfig(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteRuntimeConfig(ctx, key any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRuntimeConfig", reflect.TypeOf((*MockStore)(nil).DeleteRuntimeConfig), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteRuntimeConfig", reflect.TypeOf((*MockStore)(nil).DeleteRuntimeConfig), ctx, key) } // DeleteTailnetAgent mocks base method. -func (m *MockStore) DeleteTailnetAgent(arg0 context.Context, arg1 database.DeleteTailnetAgentParams) (database.DeleteTailnetAgentRow, error) { +func (m *MockStore) DeleteTailnetAgent(ctx context.Context, arg database.DeleteTailnetAgentParams) (database.DeleteTailnetAgentRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteTailnetAgent", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteTailnetAgent", ctx, arg) ret0, _ := ret[0].(database.DeleteTailnetAgentRow) ret1, _ := ret[1].(error) return ret0, ret1 } // DeleteTailnetAgent indicates an expected call of DeleteTailnetAgent. -func (mr *MockStoreMockRecorder) DeleteTailnetAgent(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteTailnetAgent(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetAgent", reflect.TypeOf((*MockStore)(nil).DeleteTailnetAgent), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetAgent", reflect.TypeOf((*MockStore)(nil).DeleteTailnetAgent), ctx, arg) } // DeleteTailnetClient mocks base method. -func (m *MockStore) DeleteTailnetClient(arg0 context.Context, arg1 database.DeleteTailnetClientParams) (database.DeleteTailnetClientRow, error) { +func (m *MockStore) DeleteTailnetClient(ctx context.Context, arg database.DeleteTailnetClientParams) (database.DeleteTailnetClientRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteTailnetClient", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteTailnetClient", ctx, arg) ret0, _ := ret[0].(database.DeleteTailnetClientRow) ret1, _ := ret[1].(error) return ret0, ret1 } // DeleteTailnetClient indicates an expected call of DeleteTailnetClient. -func (mr *MockStoreMockRecorder) DeleteTailnetClient(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteTailnetClient(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetClient", reflect.TypeOf((*MockStore)(nil).DeleteTailnetClient), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetClient", reflect.TypeOf((*MockStore)(nil).DeleteTailnetClient), ctx, arg) } // DeleteTailnetClientSubscription mocks base method. -func (m *MockStore) DeleteTailnetClientSubscription(arg0 context.Context, arg1 database.DeleteTailnetClientSubscriptionParams) error { +func (m *MockStore) DeleteTailnetClientSubscription(ctx context.Context, arg database.DeleteTailnetClientSubscriptionParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteTailnetClientSubscription", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteTailnetClientSubscription", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteTailnetClientSubscription indicates an expected call of DeleteTailnetClientSubscription. -func (mr *MockStoreMockRecorder) DeleteTailnetClientSubscription(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteTailnetClientSubscription(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetClientSubscription", reflect.TypeOf((*MockStore)(nil).DeleteTailnetClientSubscription), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetClientSubscription", reflect.TypeOf((*MockStore)(nil).DeleteTailnetClientSubscription), ctx, arg) } // DeleteTailnetPeer mocks base method. -func (m *MockStore) DeleteTailnetPeer(arg0 context.Context, arg1 database.DeleteTailnetPeerParams) (database.DeleteTailnetPeerRow, error) { +func (m *MockStore) DeleteTailnetPeer(ctx context.Context, arg database.DeleteTailnetPeerParams) (database.DeleteTailnetPeerRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteTailnetPeer", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteTailnetPeer", ctx, arg) ret0, _ := ret[0].(database.DeleteTailnetPeerRow) ret1, _ := ret[1].(error) return ret0, ret1 } // DeleteTailnetPeer indicates an expected call of DeleteTailnetPeer. -func (mr *MockStoreMockRecorder) DeleteTailnetPeer(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteTailnetPeer(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetPeer", reflect.TypeOf((*MockStore)(nil).DeleteTailnetPeer), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetPeer", reflect.TypeOf((*MockStore)(nil).DeleteTailnetPeer), ctx, arg) } // DeleteTailnetTunnel mocks base method. -func (m *MockStore) DeleteTailnetTunnel(arg0 context.Context, arg1 database.DeleteTailnetTunnelParams) (database.DeleteTailnetTunnelRow, error) { +func (m *MockStore) DeleteTailnetTunnel(ctx context.Context, arg database.DeleteTailnetTunnelParams) (database.DeleteTailnetTunnelRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteTailnetTunnel", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteTailnetTunnel", ctx, arg) ret0, _ := ret[0].(database.DeleteTailnetTunnelRow) ret1, _ := ret[1].(error) return ret0, ret1 } // DeleteTailnetTunnel indicates an expected call of DeleteTailnetTunnel. -func (mr *MockStoreMockRecorder) DeleteTailnetTunnel(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteTailnetTunnel(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetTunnel", reflect.TypeOf((*MockStore)(nil).DeleteTailnetTunnel), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteTailnetTunnel", reflect.TypeOf((*MockStore)(nil).DeleteTailnetTunnel), ctx, arg) } // DeleteWorkspaceAgentPortShare mocks base method. -func (m *MockStore) DeleteWorkspaceAgentPortShare(arg0 context.Context, arg1 database.DeleteWorkspaceAgentPortShareParams) error { +func (m *MockStore) DeleteWorkspaceAgentPortShare(ctx context.Context, arg database.DeleteWorkspaceAgentPortShareParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteWorkspaceAgentPortShare", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteWorkspaceAgentPortShare", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // DeleteWorkspaceAgentPortShare indicates an expected call of DeleteWorkspaceAgentPortShare. -func (mr *MockStoreMockRecorder) DeleteWorkspaceAgentPortShare(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteWorkspaceAgentPortShare(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWorkspaceAgentPortShare", reflect.TypeOf((*MockStore)(nil).DeleteWorkspaceAgentPortShare), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWorkspaceAgentPortShare", reflect.TypeOf((*MockStore)(nil).DeleteWorkspaceAgentPortShare), ctx, arg) } // DeleteWorkspaceAgentPortSharesByTemplate mocks base method. -func (m *MockStore) DeleteWorkspaceAgentPortSharesByTemplate(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) DeleteWorkspaceAgentPortSharesByTemplate(ctx context.Context, templateID uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "DeleteWorkspaceAgentPortSharesByTemplate", arg0, arg1) + ret := m.ctrl.Call(m, "DeleteWorkspaceAgentPortSharesByTemplate", ctx, templateID) ret0, _ := ret[0].(error) return ret0 } // DeleteWorkspaceAgentPortSharesByTemplate indicates an expected call of DeleteWorkspaceAgentPortSharesByTemplate. -func (mr *MockStoreMockRecorder) DeleteWorkspaceAgentPortSharesByTemplate(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) DeleteWorkspaceAgentPortSharesByTemplate(ctx, templateID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWorkspaceAgentPortSharesByTemplate", reflect.TypeOf((*MockStore)(nil).DeleteWorkspaceAgentPortSharesByTemplate), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWorkspaceAgentPortSharesByTemplate", reflect.TypeOf((*MockStore)(nil).DeleteWorkspaceAgentPortSharesByTemplate), ctx, templateID) } // EnqueueNotificationMessage mocks base method. -func (m *MockStore) EnqueueNotificationMessage(arg0 context.Context, arg1 database.EnqueueNotificationMessageParams) error { +func (m *MockStore) EnqueueNotificationMessage(ctx context.Context, arg database.EnqueueNotificationMessageParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "EnqueueNotificationMessage", arg0, arg1) + ret := m.ctrl.Call(m, "EnqueueNotificationMessage", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // EnqueueNotificationMessage indicates an expected call of EnqueueNotificationMessage. -func (mr *MockStoreMockRecorder) EnqueueNotificationMessage(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) EnqueueNotificationMessage(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EnqueueNotificationMessage", reflect.TypeOf((*MockStore)(nil).EnqueueNotificationMessage), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EnqueueNotificationMessage", reflect.TypeOf((*MockStore)(nil).EnqueueNotificationMessage), ctx, arg) } // FavoriteWorkspace mocks base method. -func (m *MockStore) FavoriteWorkspace(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) FavoriteWorkspace(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "FavoriteWorkspace", arg0, arg1) + ret := m.ctrl.Call(m, "FavoriteWorkspace", ctx, id) ret0, _ := ret[0].(error) return ret0 } // FavoriteWorkspace indicates an expected call of FavoriteWorkspace. -func (mr *MockStoreMockRecorder) FavoriteWorkspace(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) FavoriteWorkspace(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FavoriteWorkspace", reflect.TypeOf((*MockStore)(nil).FavoriteWorkspace), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FavoriteWorkspace", reflect.TypeOf((*MockStore)(nil).FavoriteWorkspace), ctx, id) } // FetchNewMessageMetadata mocks base method. -func (m *MockStore) FetchNewMessageMetadata(arg0 context.Context, arg1 database.FetchNewMessageMetadataParams) (database.FetchNewMessageMetadataRow, error) { +func (m *MockStore) FetchNewMessageMetadata(ctx context.Context, arg database.FetchNewMessageMetadataParams) (database.FetchNewMessageMetadataRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "FetchNewMessageMetadata", arg0, arg1) + ret := m.ctrl.Call(m, "FetchNewMessageMetadata", ctx, arg) ret0, _ := ret[0].(database.FetchNewMessageMetadataRow) ret1, _ := ret[1].(error) return ret0, ret1 } // FetchNewMessageMetadata indicates an expected call of FetchNewMessageMetadata. -func (mr *MockStoreMockRecorder) FetchNewMessageMetadata(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) FetchNewMessageMetadata(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchNewMessageMetadata", reflect.TypeOf((*MockStore)(nil).FetchNewMessageMetadata), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchNewMessageMetadata", reflect.TypeOf((*MockStore)(nil).FetchNewMessageMetadata), ctx, arg) } // GetAPIKeyByID mocks base method. -func (m *MockStore) GetAPIKeyByID(arg0 context.Context, arg1 string) (database.APIKey, error) { +func (m *MockStore) GetAPIKeyByID(ctx context.Context, id string) (database.APIKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAPIKeyByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetAPIKeyByID", ctx, id) ret0, _ := ret[0].(database.APIKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAPIKeyByID indicates an expected call of GetAPIKeyByID. -func (mr *MockStoreMockRecorder) GetAPIKeyByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAPIKeyByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByID", reflect.TypeOf((*MockStore)(nil).GetAPIKeyByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByID", reflect.TypeOf((*MockStore)(nil).GetAPIKeyByID), ctx, id) } // GetAPIKeyByName mocks base method. -func (m *MockStore) GetAPIKeyByName(arg0 context.Context, arg1 database.GetAPIKeyByNameParams) (database.APIKey, error) { +func (m *MockStore) GetAPIKeyByName(ctx context.Context, arg database.GetAPIKeyByNameParams) (database.APIKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAPIKeyByName", arg0, arg1) + ret := m.ctrl.Call(m, "GetAPIKeyByName", ctx, arg) ret0, _ := ret[0].(database.APIKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAPIKeyByName indicates an expected call of GetAPIKeyByName. -func (mr *MockStoreMockRecorder) GetAPIKeyByName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAPIKeyByName(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByName", reflect.TypeOf((*MockStore)(nil).GetAPIKeyByName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeyByName", reflect.TypeOf((*MockStore)(nil).GetAPIKeyByName), ctx, arg) } // GetAPIKeysByLoginType mocks base method. -func (m *MockStore) GetAPIKeysByLoginType(arg0 context.Context, arg1 database.LoginType) ([]database.APIKey, error) { +func (m *MockStore) GetAPIKeysByLoginType(ctx context.Context, loginType database.LoginType) ([]database.APIKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAPIKeysByLoginType", arg0, arg1) + ret := m.ctrl.Call(m, "GetAPIKeysByLoginType", ctx, loginType) ret0, _ := ret[0].([]database.APIKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAPIKeysByLoginType indicates an expected call of GetAPIKeysByLoginType. -func (mr *MockStoreMockRecorder) GetAPIKeysByLoginType(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAPIKeysByLoginType(ctx, loginType any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeysByLoginType", reflect.TypeOf((*MockStore)(nil).GetAPIKeysByLoginType), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeysByLoginType", reflect.TypeOf((*MockStore)(nil).GetAPIKeysByLoginType), ctx, loginType) } // GetAPIKeysByUserID mocks base method. -func (m *MockStore) GetAPIKeysByUserID(arg0 context.Context, arg1 database.GetAPIKeysByUserIDParams) ([]database.APIKey, error) { +func (m *MockStore) GetAPIKeysByUserID(ctx context.Context, arg database.GetAPIKeysByUserIDParams) ([]database.APIKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAPIKeysByUserID", arg0, arg1) + ret := m.ctrl.Call(m, "GetAPIKeysByUserID", ctx, arg) ret0, _ := ret[0].([]database.APIKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAPIKeysByUserID indicates an expected call of GetAPIKeysByUserID. -func (mr *MockStoreMockRecorder) GetAPIKeysByUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAPIKeysByUserID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeysByUserID", reflect.TypeOf((*MockStore)(nil).GetAPIKeysByUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeysByUserID", reflect.TypeOf((*MockStore)(nil).GetAPIKeysByUserID), ctx, arg) } // GetAPIKeysLastUsedAfter mocks base method. -func (m *MockStore) GetAPIKeysLastUsedAfter(arg0 context.Context, arg1 time.Time) ([]database.APIKey, error) { +func (m *MockStore) GetAPIKeysLastUsedAfter(ctx context.Context, lastUsed time.Time) ([]database.APIKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAPIKeysLastUsedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetAPIKeysLastUsedAfter", ctx, lastUsed) ret0, _ := ret[0].([]database.APIKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAPIKeysLastUsedAfter indicates an expected call of GetAPIKeysLastUsedAfter. -func (mr *MockStoreMockRecorder) GetAPIKeysLastUsedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAPIKeysLastUsedAfter(ctx, lastUsed any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeysLastUsedAfter", reflect.TypeOf((*MockStore)(nil).GetAPIKeysLastUsedAfter), arg0, arg1) -} - -// GetAccumulatedUsersInsights mocks base method. -func (m *MockStore) GetAccumulatedUsersInsights(arg0 context.Context, arg1 database.GetAccumulatedUsersInsightsParams) ([]database.GetAccumulatedUsersInsightsRow, error) { - m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAccumulatedUsersInsights", arg0, arg1) - ret0, _ := ret[0].([]database.GetAccumulatedUsersInsightsRow) - ret1, _ := ret[1].(error) - return ret0, ret1 -} - -// GetAccumulatedUsersInsights indicates an expected call of GetAccumulatedUsersInsights. -func (mr *MockStoreMockRecorder) GetAccumulatedUsersInsights(arg0, arg1 any) *gomock.Call { - mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAccumulatedUsersInsights", reflect.TypeOf((*MockStore)(nil).GetAccumulatedUsersInsights), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAPIKeysLastUsedAfter", reflect.TypeOf((*MockStore)(nil).GetAPIKeysLastUsedAfter), ctx, lastUsed) } // GetActiveUserCount mocks base method. -func (m *MockStore) GetActiveUserCount(arg0 context.Context) (int64, error) { +func (m *MockStore) GetActiveUserCount(ctx context.Context) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetActiveUserCount", arg0) + ret := m.ctrl.Call(m, "GetActiveUserCount", ctx) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetActiveUserCount indicates an expected call of GetActiveUserCount. -func (mr *MockStoreMockRecorder) GetActiveUserCount(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetActiveUserCount(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetActiveUserCount", reflect.TypeOf((*MockStore)(nil).GetActiveUserCount), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetActiveUserCount", reflect.TypeOf((*MockStore)(nil).GetActiveUserCount), ctx) } // GetActiveWorkspaceBuildsByTemplateID mocks base method. -func (m *MockStore) GetActiveWorkspaceBuildsByTemplateID(arg0 context.Context, arg1 uuid.UUID) ([]database.WorkspaceBuild, error) { +func (m *MockStore) GetActiveWorkspaceBuildsByTemplateID(ctx context.Context, templateID uuid.UUID) ([]database.WorkspaceBuild, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetActiveWorkspaceBuildsByTemplateID", arg0, arg1) + ret := m.ctrl.Call(m, "GetActiveWorkspaceBuildsByTemplateID", ctx, templateID) ret0, _ := ret[0].([]database.WorkspaceBuild) ret1, _ := ret[1].(error) return ret0, ret1 } // GetActiveWorkspaceBuildsByTemplateID indicates an expected call of GetActiveWorkspaceBuildsByTemplateID. -func (mr *MockStoreMockRecorder) GetActiveWorkspaceBuildsByTemplateID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetActiveWorkspaceBuildsByTemplateID(ctx, templateID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetActiveWorkspaceBuildsByTemplateID", reflect.TypeOf((*MockStore)(nil).GetActiveWorkspaceBuildsByTemplateID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetActiveWorkspaceBuildsByTemplateID", reflect.TypeOf((*MockStore)(nil).GetActiveWorkspaceBuildsByTemplateID), ctx, templateID) } // GetAllTailnetAgents mocks base method. -func (m *MockStore) GetAllTailnetAgents(arg0 context.Context) ([]database.TailnetAgent, error) { +func (m *MockStore) GetAllTailnetAgents(ctx context.Context) ([]database.TailnetAgent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAllTailnetAgents", arg0) + ret := m.ctrl.Call(m, "GetAllTailnetAgents", ctx) ret0, _ := ret[0].([]database.TailnetAgent) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAllTailnetAgents indicates an expected call of GetAllTailnetAgents. -func (mr *MockStoreMockRecorder) GetAllTailnetAgents(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAllTailnetAgents(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllTailnetAgents", reflect.TypeOf((*MockStore)(nil).GetAllTailnetAgents), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllTailnetAgents", reflect.TypeOf((*MockStore)(nil).GetAllTailnetAgents), ctx) } // GetAllTailnetCoordinators mocks base method. -func (m *MockStore) GetAllTailnetCoordinators(arg0 context.Context) ([]database.TailnetCoordinator, error) { +func (m *MockStore) GetAllTailnetCoordinators(ctx context.Context) ([]database.TailnetCoordinator, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAllTailnetCoordinators", arg0) + ret := m.ctrl.Call(m, "GetAllTailnetCoordinators", ctx) ret0, _ := ret[0].([]database.TailnetCoordinator) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAllTailnetCoordinators indicates an expected call of GetAllTailnetCoordinators. -func (mr *MockStoreMockRecorder) GetAllTailnetCoordinators(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAllTailnetCoordinators(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllTailnetCoordinators", reflect.TypeOf((*MockStore)(nil).GetAllTailnetCoordinators), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllTailnetCoordinators", reflect.TypeOf((*MockStore)(nil).GetAllTailnetCoordinators), ctx) } // GetAllTailnetPeers mocks base method. -func (m *MockStore) GetAllTailnetPeers(arg0 context.Context) ([]database.TailnetPeer, error) { +func (m *MockStore) GetAllTailnetPeers(ctx context.Context) ([]database.TailnetPeer, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAllTailnetPeers", arg0) + ret := m.ctrl.Call(m, "GetAllTailnetPeers", ctx) ret0, _ := ret[0].([]database.TailnetPeer) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAllTailnetPeers indicates an expected call of GetAllTailnetPeers. -func (mr *MockStoreMockRecorder) GetAllTailnetPeers(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAllTailnetPeers(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllTailnetPeers", reflect.TypeOf((*MockStore)(nil).GetAllTailnetPeers), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllTailnetPeers", reflect.TypeOf((*MockStore)(nil).GetAllTailnetPeers), ctx) } // GetAllTailnetTunnels mocks base method. -func (m *MockStore) GetAllTailnetTunnels(arg0 context.Context) ([]database.TailnetTunnel, error) { +func (m *MockStore) GetAllTailnetTunnels(ctx context.Context) ([]database.TailnetTunnel, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAllTailnetTunnels", arg0) + ret := m.ctrl.Call(m, "GetAllTailnetTunnels", ctx) ret0, _ := ret[0].([]database.TailnetTunnel) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAllTailnetTunnels indicates an expected call of GetAllTailnetTunnels. -func (mr *MockStoreMockRecorder) GetAllTailnetTunnels(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAllTailnetTunnels(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllTailnetTunnels", reflect.TypeOf((*MockStore)(nil).GetAllTailnetTunnels), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllTailnetTunnels", reflect.TypeOf((*MockStore)(nil).GetAllTailnetTunnels), ctx) } // GetAnnouncementBanners mocks base method. -func (m *MockStore) GetAnnouncementBanners(arg0 context.Context) (string, error) { +func (m *MockStore) GetAnnouncementBanners(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAnnouncementBanners", arg0) + ret := m.ctrl.Call(m, "GetAnnouncementBanners", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAnnouncementBanners indicates an expected call of GetAnnouncementBanners. -func (mr *MockStoreMockRecorder) GetAnnouncementBanners(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAnnouncementBanners(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAnnouncementBanners", reflect.TypeOf((*MockStore)(nil).GetAnnouncementBanners), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAnnouncementBanners", reflect.TypeOf((*MockStore)(nil).GetAnnouncementBanners), ctx) } // GetAppSecurityKey mocks base method. -func (m *MockStore) GetAppSecurityKey(arg0 context.Context) (string, error) { +func (m *MockStore) GetAppSecurityKey(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAppSecurityKey", arg0) + ret := m.ctrl.Call(m, "GetAppSecurityKey", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAppSecurityKey indicates an expected call of GetAppSecurityKey. -func (mr *MockStoreMockRecorder) GetAppSecurityKey(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAppSecurityKey(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAppSecurityKey", reflect.TypeOf((*MockStore)(nil).GetAppSecurityKey), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAppSecurityKey", reflect.TypeOf((*MockStore)(nil).GetAppSecurityKey), ctx) } // GetApplicationName mocks base method. -func (m *MockStore) GetApplicationName(arg0 context.Context) (string, error) { +func (m *MockStore) GetApplicationName(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetApplicationName", arg0) + ret := m.ctrl.Call(m, "GetApplicationName", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetApplicationName indicates an expected call of GetApplicationName. -func (mr *MockStoreMockRecorder) GetApplicationName(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetApplicationName(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetApplicationName", reflect.TypeOf((*MockStore)(nil).GetApplicationName), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetApplicationName", reflect.TypeOf((*MockStore)(nil).GetApplicationName), ctx) } // GetAuditLogsOffset mocks base method. -func (m *MockStore) GetAuditLogsOffset(arg0 context.Context, arg1 database.GetAuditLogsOffsetParams) ([]database.GetAuditLogsOffsetRow, error) { +func (m *MockStore) GetAuditLogsOffset(ctx context.Context, arg database.GetAuditLogsOffsetParams) ([]database.GetAuditLogsOffsetRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAuditLogsOffset", arg0, arg1) + ret := m.ctrl.Call(m, "GetAuditLogsOffset", ctx, arg) ret0, _ := ret[0].([]database.GetAuditLogsOffsetRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAuditLogsOffset indicates an expected call of GetAuditLogsOffset. -func (mr *MockStoreMockRecorder) GetAuditLogsOffset(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAuditLogsOffset(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuditLogsOffset", reflect.TypeOf((*MockStore)(nil).GetAuditLogsOffset), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuditLogsOffset", reflect.TypeOf((*MockStore)(nil).GetAuditLogsOffset), ctx, arg) } // GetAuthorizationUserRoles mocks base method. -func (m *MockStore) GetAuthorizationUserRoles(arg0 context.Context, arg1 uuid.UUID) (database.GetAuthorizationUserRolesRow, error) { +func (m *MockStore) GetAuthorizationUserRoles(ctx context.Context, userID uuid.UUID) (database.GetAuthorizationUserRolesRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAuthorizationUserRoles", arg0, arg1) + ret := m.ctrl.Call(m, "GetAuthorizationUserRoles", ctx, userID) ret0, _ := ret[0].(database.GetAuthorizationUserRolesRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAuthorizationUserRoles indicates an expected call of GetAuthorizationUserRoles. -func (mr *MockStoreMockRecorder) GetAuthorizationUserRoles(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAuthorizationUserRoles(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizationUserRoles", reflect.TypeOf((*MockStore)(nil).GetAuthorizationUserRoles), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizationUserRoles", reflect.TypeOf((*MockStore)(nil).GetAuthorizationUserRoles), ctx, userID) } // GetAuthorizedAuditLogsOffset mocks base method. -func (m *MockStore) GetAuthorizedAuditLogsOffset(arg0 context.Context, arg1 database.GetAuditLogsOffsetParams, arg2 rbac.PreparedAuthorized) ([]database.GetAuditLogsOffsetRow, error) { +func (m *MockStore) GetAuthorizedAuditLogsOffset(ctx context.Context, arg database.GetAuditLogsOffsetParams, prepared rbac.PreparedAuthorized) ([]database.GetAuditLogsOffsetRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAuthorizedAuditLogsOffset", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetAuthorizedAuditLogsOffset", ctx, arg, prepared) ret0, _ := ret[0].([]database.GetAuditLogsOffsetRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAuthorizedAuditLogsOffset indicates an expected call of GetAuthorizedAuditLogsOffset. -func (mr *MockStoreMockRecorder) GetAuthorizedAuditLogsOffset(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAuthorizedAuditLogsOffset(ctx, arg, prepared any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedAuditLogsOffset", reflect.TypeOf((*MockStore)(nil).GetAuthorizedAuditLogsOffset), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedAuditLogsOffset", reflect.TypeOf((*MockStore)(nil).GetAuthorizedAuditLogsOffset), ctx, arg, prepared) } // GetAuthorizedTemplates mocks base method. -func (m *MockStore) GetAuthorizedTemplates(arg0 context.Context, arg1 database.GetTemplatesWithFilterParams, arg2 rbac.PreparedAuthorized) ([]database.Template, error) { +func (m *MockStore) GetAuthorizedTemplates(ctx context.Context, arg database.GetTemplatesWithFilterParams, prepared rbac.PreparedAuthorized) ([]database.Template, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAuthorizedTemplates", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetAuthorizedTemplates", ctx, arg, prepared) ret0, _ := ret[0].([]database.Template) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAuthorizedTemplates indicates an expected call of GetAuthorizedTemplates. -func (mr *MockStoreMockRecorder) GetAuthorizedTemplates(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAuthorizedTemplates(ctx, arg, prepared any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedTemplates", reflect.TypeOf((*MockStore)(nil).GetAuthorizedTemplates), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedTemplates", reflect.TypeOf((*MockStore)(nil).GetAuthorizedTemplates), ctx, arg, prepared) } // GetAuthorizedUsers mocks base method. -func (m *MockStore) GetAuthorizedUsers(arg0 context.Context, arg1 database.GetUsersParams, arg2 rbac.PreparedAuthorized) ([]database.GetUsersRow, error) { +func (m *MockStore) GetAuthorizedUsers(ctx context.Context, arg database.GetUsersParams, prepared rbac.PreparedAuthorized) ([]database.GetUsersRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAuthorizedUsers", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetAuthorizedUsers", ctx, arg, prepared) ret0, _ := ret[0].([]database.GetUsersRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAuthorizedUsers indicates an expected call of GetAuthorizedUsers. -func (mr *MockStoreMockRecorder) GetAuthorizedUsers(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAuthorizedUsers(ctx, arg, prepared any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedUsers", reflect.TypeOf((*MockStore)(nil).GetAuthorizedUsers), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedUsers", reflect.TypeOf((*MockStore)(nil).GetAuthorizedUsers), ctx, arg, prepared) } // GetAuthorizedWorkspaces mocks base method. -func (m *MockStore) GetAuthorizedWorkspaces(arg0 context.Context, arg1 database.GetWorkspacesParams, arg2 rbac.PreparedAuthorized) ([]database.GetWorkspacesRow, error) { +func (m *MockStore) GetAuthorizedWorkspaces(ctx context.Context, arg database.GetWorkspacesParams, prepared rbac.PreparedAuthorized) ([]database.GetWorkspacesRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAuthorizedWorkspaces", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetAuthorizedWorkspaces", ctx, arg, prepared) ret0, _ := ret[0].([]database.GetWorkspacesRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAuthorizedWorkspaces indicates an expected call of GetAuthorizedWorkspaces. -func (mr *MockStoreMockRecorder) GetAuthorizedWorkspaces(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAuthorizedWorkspaces(ctx, arg, prepared any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedWorkspaces", reflect.TypeOf((*MockStore)(nil).GetAuthorizedWorkspaces), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedWorkspaces", reflect.TypeOf((*MockStore)(nil).GetAuthorizedWorkspaces), ctx, arg, prepared) } // GetAuthorizedWorkspacesAndAgentsByOwnerID mocks base method. -func (m *MockStore) GetAuthorizedWorkspacesAndAgentsByOwnerID(arg0 context.Context, arg1 uuid.UUID, arg2 rbac.PreparedAuthorized) ([]database.GetWorkspacesAndAgentsByOwnerIDRow, error) { +func (m *MockStore) GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx context.Context, ownerID uuid.UUID, prepared rbac.PreparedAuthorized) ([]database.GetWorkspacesAndAgentsByOwnerIDRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetAuthorizedWorkspacesAndAgentsByOwnerID", arg0, arg1, arg2) + ret := m.ctrl.Call(m, "GetAuthorizedWorkspacesAndAgentsByOwnerID", ctx, ownerID, prepared) ret0, _ := ret[0].([]database.GetWorkspacesAndAgentsByOwnerIDRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetAuthorizedWorkspacesAndAgentsByOwnerID indicates an expected call of GetAuthorizedWorkspacesAndAgentsByOwnerID. -func (mr *MockStoreMockRecorder) GetAuthorizedWorkspacesAndAgentsByOwnerID(arg0, arg1, arg2 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetAuthorizedWorkspacesAndAgentsByOwnerID(ctx, ownerID, prepared any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedWorkspacesAndAgentsByOwnerID", reflect.TypeOf((*MockStore)(nil).GetAuthorizedWorkspacesAndAgentsByOwnerID), arg0, arg1, arg2) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAuthorizedWorkspacesAndAgentsByOwnerID", reflect.TypeOf((*MockStore)(nil).GetAuthorizedWorkspacesAndAgentsByOwnerID), ctx, ownerID, prepared) } // GetCoordinatorResumeTokenSigningKey mocks base method. -func (m *MockStore) GetCoordinatorResumeTokenSigningKey(arg0 context.Context) (string, error) { +func (m *MockStore) GetCoordinatorResumeTokenSigningKey(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetCoordinatorResumeTokenSigningKey", arg0) + ret := m.ctrl.Call(m, "GetCoordinatorResumeTokenSigningKey", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetCoordinatorResumeTokenSigningKey indicates an expected call of GetCoordinatorResumeTokenSigningKey. -func (mr *MockStoreMockRecorder) GetCoordinatorResumeTokenSigningKey(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetCoordinatorResumeTokenSigningKey(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCoordinatorResumeTokenSigningKey", reflect.TypeOf((*MockStore)(nil).GetCoordinatorResumeTokenSigningKey), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCoordinatorResumeTokenSigningKey", reflect.TypeOf((*MockStore)(nil).GetCoordinatorResumeTokenSigningKey), ctx) } // GetCryptoKeyByFeatureAndSequence mocks base method. -func (m *MockStore) GetCryptoKeyByFeatureAndSequence(arg0 context.Context, arg1 database.GetCryptoKeyByFeatureAndSequenceParams) (database.CryptoKey, error) { +func (m *MockStore) GetCryptoKeyByFeatureAndSequence(ctx context.Context, arg database.GetCryptoKeyByFeatureAndSequenceParams) (database.CryptoKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetCryptoKeyByFeatureAndSequence", arg0, arg1) + ret := m.ctrl.Call(m, "GetCryptoKeyByFeatureAndSequence", ctx, arg) ret0, _ := ret[0].(database.CryptoKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetCryptoKeyByFeatureAndSequence indicates an expected call of GetCryptoKeyByFeatureAndSequence. -func (mr *MockStoreMockRecorder) GetCryptoKeyByFeatureAndSequence(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetCryptoKeyByFeatureAndSequence(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCryptoKeyByFeatureAndSequence", reflect.TypeOf((*MockStore)(nil).GetCryptoKeyByFeatureAndSequence), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCryptoKeyByFeatureAndSequence", reflect.TypeOf((*MockStore)(nil).GetCryptoKeyByFeatureAndSequence), ctx, arg) } // GetCryptoKeys mocks base method. -func (m *MockStore) GetCryptoKeys(arg0 context.Context) ([]database.CryptoKey, error) { +func (m *MockStore) GetCryptoKeys(ctx context.Context) ([]database.CryptoKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetCryptoKeys", arg0) + ret := m.ctrl.Call(m, "GetCryptoKeys", ctx) ret0, _ := ret[0].([]database.CryptoKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetCryptoKeys indicates an expected call of GetCryptoKeys. -func (mr *MockStoreMockRecorder) GetCryptoKeys(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetCryptoKeys(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCryptoKeys", reflect.TypeOf((*MockStore)(nil).GetCryptoKeys), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCryptoKeys", reflect.TypeOf((*MockStore)(nil).GetCryptoKeys), ctx) } // GetCryptoKeysByFeature mocks base method. -func (m *MockStore) GetCryptoKeysByFeature(arg0 context.Context, arg1 database.CryptoKeyFeature) ([]database.CryptoKey, error) { +func (m *MockStore) GetCryptoKeysByFeature(ctx context.Context, feature database.CryptoKeyFeature) ([]database.CryptoKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetCryptoKeysByFeature", arg0, arg1) + ret := m.ctrl.Call(m, "GetCryptoKeysByFeature", ctx, feature) ret0, _ := ret[0].([]database.CryptoKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetCryptoKeysByFeature indicates an expected call of GetCryptoKeysByFeature. -func (mr *MockStoreMockRecorder) GetCryptoKeysByFeature(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetCryptoKeysByFeature(ctx, feature any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCryptoKeysByFeature", reflect.TypeOf((*MockStore)(nil).GetCryptoKeysByFeature), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCryptoKeysByFeature", reflect.TypeOf((*MockStore)(nil).GetCryptoKeysByFeature), ctx, feature) } // GetDBCryptKeys mocks base method. -func (m *MockStore) GetDBCryptKeys(arg0 context.Context) ([]database.DBCryptKey, error) { +func (m *MockStore) GetDBCryptKeys(ctx context.Context) ([]database.DBCryptKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDBCryptKeys", arg0) + ret := m.ctrl.Call(m, "GetDBCryptKeys", ctx) ret0, _ := ret[0].([]database.DBCryptKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDBCryptKeys indicates an expected call of GetDBCryptKeys. -func (mr *MockStoreMockRecorder) GetDBCryptKeys(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetDBCryptKeys(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDBCryptKeys", reflect.TypeOf((*MockStore)(nil).GetDBCryptKeys), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDBCryptKeys", reflect.TypeOf((*MockStore)(nil).GetDBCryptKeys), ctx) } // GetDERPMeshKey mocks base method. -func (m *MockStore) GetDERPMeshKey(arg0 context.Context) (string, error) { +func (m *MockStore) GetDERPMeshKey(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDERPMeshKey", arg0) + ret := m.ctrl.Call(m, "GetDERPMeshKey", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDERPMeshKey indicates an expected call of GetDERPMeshKey. -func (mr *MockStoreMockRecorder) GetDERPMeshKey(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetDERPMeshKey(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDERPMeshKey", reflect.TypeOf((*MockStore)(nil).GetDERPMeshKey), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDERPMeshKey", reflect.TypeOf((*MockStore)(nil).GetDERPMeshKey), ctx) } // GetDefaultOrganization mocks base method. -func (m *MockStore) GetDefaultOrganization(arg0 context.Context) (database.Organization, error) { +func (m *MockStore) GetDefaultOrganization(ctx context.Context) (database.Organization, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDefaultOrganization", arg0) + ret := m.ctrl.Call(m, "GetDefaultOrganization", ctx) ret0, _ := ret[0].(database.Organization) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDefaultOrganization indicates an expected call of GetDefaultOrganization. -func (mr *MockStoreMockRecorder) GetDefaultOrganization(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetDefaultOrganization(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDefaultOrganization", reflect.TypeOf((*MockStore)(nil).GetDefaultOrganization), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDefaultOrganization", reflect.TypeOf((*MockStore)(nil).GetDefaultOrganization), ctx) } // GetDefaultProxyConfig mocks base method. -func (m *MockStore) GetDefaultProxyConfig(arg0 context.Context) (database.GetDefaultProxyConfigRow, error) { +func (m *MockStore) GetDefaultProxyConfig(ctx context.Context) (database.GetDefaultProxyConfigRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDefaultProxyConfig", arg0) + ret := m.ctrl.Call(m, "GetDefaultProxyConfig", ctx) ret0, _ := ret[0].(database.GetDefaultProxyConfigRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDefaultProxyConfig indicates an expected call of GetDefaultProxyConfig. -func (mr *MockStoreMockRecorder) GetDefaultProxyConfig(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetDefaultProxyConfig(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDefaultProxyConfig", reflect.TypeOf((*MockStore)(nil).GetDefaultProxyConfig), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDefaultProxyConfig", reflect.TypeOf((*MockStore)(nil).GetDefaultProxyConfig), ctx) } // GetDeploymentDAUs mocks base method. -func (m *MockStore) GetDeploymentDAUs(arg0 context.Context, arg1 int32) ([]database.GetDeploymentDAUsRow, error) { +func (m *MockStore) GetDeploymentDAUs(ctx context.Context, tzOffset int32) ([]database.GetDeploymentDAUsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDeploymentDAUs", arg0, arg1) + ret := m.ctrl.Call(m, "GetDeploymentDAUs", ctx, tzOffset) ret0, _ := ret[0].([]database.GetDeploymentDAUsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDeploymentDAUs indicates an expected call of GetDeploymentDAUs. -func (mr *MockStoreMockRecorder) GetDeploymentDAUs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetDeploymentDAUs(ctx, tzOffset any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentDAUs", reflect.TypeOf((*MockStore)(nil).GetDeploymentDAUs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentDAUs", reflect.TypeOf((*MockStore)(nil).GetDeploymentDAUs), ctx, tzOffset) } // GetDeploymentID mocks base method. -func (m *MockStore) GetDeploymentID(arg0 context.Context) (string, error) { +func (m *MockStore) GetDeploymentID(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDeploymentID", arg0) + ret := m.ctrl.Call(m, "GetDeploymentID", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDeploymentID indicates an expected call of GetDeploymentID. -func (mr *MockStoreMockRecorder) GetDeploymentID(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetDeploymentID(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentID", reflect.TypeOf((*MockStore)(nil).GetDeploymentID), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentID", reflect.TypeOf((*MockStore)(nil).GetDeploymentID), ctx) } // GetDeploymentWorkspaceAgentStats mocks base method. -func (m *MockStore) GetDeploymentWorkspaceAgentStats(arg0 context.Context, arg1 time.Time) (database.GetDeploymentWorkspaceAgentStatsRow, error) { +func (m *MockStore) GetDeploymentWorkspaceAgentStats(ctx context.Context, createdAt time.Time) (database.GetDeploymentWorkspaceAgentStatsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDeploymentWorkspaceAgentStats", arg0, arg1) + ret := m.ctrl.Call(m, "GetDeploymentWorkspaceAgentStats", ctx, createdAt) ret0, _ := ret[0].(database.GetDeploymentWorkspaceAgentStatsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDeploymentWorkspaceAgentStats indicates an expected call of GetDeploymentWorkspaceAgentStats. -func (mr *MockStoreMockRecorder) GetDeploymentWorkspaceAgentStats(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetDeploymentWorkspaceAgentStats(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentWorkspaceAgentStats", reflect.TypeOf((*MockStore)(nil).GetDeploymentWorkspaceAgentStats), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentWorkspaceAgentStats", reflect.TypeOf((*MockStore)(nil).GetDeploymentWorkspaceAgentStats), ctx, createdAt) } // GetDeploymentWorkspaceAgentUsageStats mocks base method. -func (m *MockStore) GetDeploymentWorkspaceAgentUsageStats(arg0 context.Context, arg1 time.Time) (database.GetDeploymentWorkspaceAgentUsageStatsRow, error) { +func (m *MockStore) GetDeploymentWorkspaceAgentUsageStats(ctx context.Context, createdAt time.Time) (database.GetDeploymentWorkspaceAgentUsageStatsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDeploymentWorkspaceAgentUsageStats", arg0, arg1) + ret := m.ctrl.Call(m, "GetDeploymentWorkspaceAgentUsageStats", ctx, createdAt) ret0, _ := ret[0].(database.GetDeploymentWorkspaceAgentUsageStatsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDeploymentWorkspaceAgentUsageStats indicates an expected call of GetDeploymentWorkspaceAgentUsageStats. -func (mr *MockStoreMockRecorder) GetDeploymentWorkspaceAgentUsageStats(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetDeploymentWorkspaceAgentUsageStats(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentWorkspaceAgentUsageStats", reflect.TypeOf((*MockStore)(nil).GetDeploymentWorkspaceAgentUsageStats), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentWorkspaceAgentUsageStats", reflect.TypeOf((*MockStore)(nil).GetDeploymentWorkspaceAgentUsageStats), ctx, createdAt) } // GetDeploymentWorkspaceStats mocks base method. -func (m *MockStore) GetDeploymentWorkspaceStats(arg0 context.Context) (database.GetDeploymentWorkspaceStatsRow, error) { +func (m *MockStore) GetDeploymentWorkspaceStats(ctx context.Context) (database.GetDeploymentWorkspaceStatsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetDeploymentWorkspaceStats", arg0) + ret := m.ctrl.Call(m, "GetDeploymentWorkspaceStats", ctx) ret0, _ := ret[0].(database.GetDeploymentWorkspaceStatsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetDeploymentWorkspaceStats indicates an expected call of GetDeploymentWorkspaceStats. -func (mr *MockStoreMockRecorder) GetDeploymentWorkspaceStats(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetDeploymentWorkspaceStats(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentWorkspaceStats", reflect.TypeOf((*MockStore)(nil).GetDeploymentWorkspaceStats), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDeploymentWorkspaceStats", reflect.TypeOf((*MockStore)(nil).GetDeploymentWorkspaceStats), ctx) } // GetExternalAuthLink mocks base method. -func (m *MockStore) GetExternalAuthLink(arg0 context.Context, arg1 database.GetExternalAuthLinkParams) (database.ExternalAuthLink, error) { +func (m *MockStore) GetExternalAuthLink(ctx context.Context, arg database.GetExternalAuthLinkParams) (database.ExternalAuthLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetExternalAuthLink", arg0, arg1) + ret := m.ctrl.Call(m, "GetExternalAuthLink", ctx, arg) ret0, _ := ret[0].(database.ExternalAuthLink) ret1, _ := ret[1].(error) return ret0, ret1 } // GetExternalAuthLink indicates an expected call of GetExternalAuthLink. -func (mr *MockStoreMockRecorder) GetExternalAuthLink(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetExternalAuthLink(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExternalAuthLink", reflect.TypeOf((*MockStore)(nil).GetExternalAuthLink), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExternalAuthLink", reflect.TypeOf((*MockStore)(nil).GetExternalAuthLink), ctx, arg) } // GetExternalAuthLinksByUserID mocks base method. -func (m *MockStore) GetExternalAuthLinksByUserID(arg0 context.Context, arg1 uuid.UUID) ([]database.ExternalAuthLink, error) { +func (m *MockStore) GetExternalAuthLinksByUserID(ctx context.Context, userID uuid.UUID) ([]database.ExternalAuthLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetExternalAuthLinksByUserID", arg0, arg1) + ret := m.ctrl.Call(m, "GetExternalAuthLinksByUserID", ctx, userID) ret0, _ := ret[0].([]database.ExternalAuthLink) ret1, _ := ret[1].(error) return ret0, ret1 } // GetExternalAuthLinksByUserID indicates an expected call of GetExternalAuthLinksByUserID. -func (mr *MockStoreMockRecorder) GetExternalAuthLinksByUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetExternalAuthLinksByUserID(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExternalAuthLinksByUserID", reflect.TypeOf((*MockStore)(nil).GetExternalAuthLinksByUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetExternalAuthLinksByUserID", reflect.TypeOf((*MockStore)(nil).GetExternalAuthLinksByUserID), ctx, userID) } // GetFailedWorkspaceBuildsByTemplateID mocks base method. -func (m *MockStore) GetFailedWorkspaceBuildsByTemplateID(arg0 context.Context, arg1 database.GetFailedWorkspaceBuildsByTemplateIDParams) ([]database.GetFailedWorkspaceBuildsByTemplateIDRow, error) { +func (m *MockStore) GetFailedWorkspaceBuildsByTemplateID(ctx context.Context, arg database.GetFailedWorkspaceBuildsByTemplateIDParams) ([]database.GetFailedWorkspaceBuildsByTemplateIDRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetFailedWorkspaceBuildsByTemplateID", arg0, arg1) + ret := m.ctrl.Call(m, "GetFailedWorkspaceBuildsByTemplateID", ctx, arg) ret0, _ := ret[0].([]database.GetFailedWorkspaceBuildsByTemplateIDRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetFailedWorkspaceBuildsByTemplateID indicates an expected call of GetFailedWorkspaceBuildsByTemplateID. -func (mr *MockStoreMockRecorder) GetFailedWorkspaceBuildsByTemplateID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetFailedWorkspaceBuildsByTemplateID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFailedWorkspaceBuildsByTemplateID", reflect.TypeOf((*MockStore)(nil).GetFailedWorkspaceBuildsByTemplateID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFailedWorkspaceBuildsByTemplateID", reflect.TypeOf((*MockStore)(nil).GetFailedWorkspaceBuildsByTemplateID), ctx, arg) } // GetFileByHashAndCreator mocks base method. -func (m *MockStore) GetFileByHashAndCreator(arg0 context.Context, arg1 database.GetFileByHashAndCreatorParams) (database.File, error) { +func (m *MockStore) GetFileByHashAndCreator(ctx context.Context, arg database.GetFileByHashAndCreatorParams) (database.File, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetFileByHashAndCreator", arg0, arg1) + ret := m.ctrl.Call(m, "GetFileByHashAndCreator", ctx, arg) ret0, _ := ret[0].(database.File) ret1, _ := ret[1].(error) return ret0, ret1 } // GetFileByHashAndCreator indicates an expected call of GetFileByHashAndCreator. -func (mr *MockStoreMockRecorder) GetFileByHashAndCreator(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetFileByHashAndCreator(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFileByHashAndCreator", reflect.TypeOf((*MockStore)(nil).GetFileByHashAndCreator), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFileByHashAndCreator", reflect.TypeOf((*MockStore)(nil).GetFileByHashAndCreator), ctx, arg) } // GetFileByID mocks base method. -func (m *MockStore) GetFileByID(arg0 context.Context, arg1 uuid.UUID) (database.File, error) { +func (m *MockStore) GetFileByID(ctx context.Context, id uuid.UUID) (database.File, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetFileByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetFileByID", ctx, id) ret0, _ := ret[0].(database.File) ret1, _ := ret[1].(error) return ret0, ret1 } // GetFileByID indicates an expected call of GetFileByID. -func (mr *MockStoreMockRecorder) GetFileByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetFileByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFileByID", reflect.TypeOf((*MockStore)(nil).GetFileByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFileByID", reflect.TypeOf((*MockStore)(nil).GetFileByID), ctx, id) } // GetFileTemplates mocks base method. -func (m *MockStore) GetFileTemplates(arg0 context.Context, arg1 uuid.UUID) ([]database.GetFileTemplatesRow, error) { +func (m *MockStore) GetFileTemplates(ctx context.Context, fileID uuid.UUID) ([]database.GetFileTemplatesRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetFileTemplates", arg0, arg1) + ret := m.ctrl.Call(m, "GetFileTemplates", ctx, fileID) ret0, _ := ret[0].([]database.GetFileTemplatesRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetFileTemplates indicates an expected call of GetFileTemplates. -func (mr *MockStoreMockRecorder) GetFileTemplates(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetFileTemplates(ctx, fileID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFileTemplates", reflect.TypeOf((*MockStore)(nil).GetFileTemplates), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFileTemplates", reflect.TypeOf((*MockStore)(nil).GetFileTemplates), ctx, fileID) } // GetGitSSHKey mocks base method. -func (m *MockStore) GetGitSSHKey(arg0 context.Context, arg1 uuid.UUID) (database.GitSSHKey, error) { +func (m *MockStore) GetGitSSHKey(ctx context.Context, userID uuid.UUID) (database.GitSSHKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetGitSSHKey", arg0, arg1) + ret := m.ctrl.Call(m, "GetGitSSHKey", ctx, userID) ret0, _ := ret[0].(database.GitSSHKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetGitSSHKey indicates an expected call of GetGitSSHKey. -func (mr *MockStoreMockRecorder) GetGitSSHKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetGitSSHKey(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGitSSHKey", reflect.TypeOf((*MockStore)(nil).GetGitSSHKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGitSSHKey", reflect.TypeOf((*MockStore)(nil).GetGitSSHKey), ctx, userID) } // GetGroupByID mocks base method. -func (m *MockStore) GetGroupByID(arg0 context.Context, arg1 uuid.UUID) (database.Group, error) { +func (m *MockStore) GetGroupByID(ctx context.Context, id uuid.UUID) (database.Group, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetGroupByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetGroupByID", ctx, id) ret0, _ := ret[0].(database.Group) ret1, _ := ret[1].(error) return ret0, ret1 } // GetGroupByID indicates an expected call of GetGroupByID. -func (mr *MockStoreMockRecorder) GetGroupByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetGroupByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupByID", reflect.TypeOf((*MockStore)(nil).GetGroupByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupByID", reflect.TypeOf((*MockStore)(nil).GetGroupByID), ctx, id) } // GetGroupByOrgAndName mocks base method. -func (m *MockStore) GetGroupByOrgAndName(arg0 context.Context, arg1 database.GetGroupByOrgAndNameParams) (database.Group, error) { +func (m *MockStore) GetGroupByOrgAndName(ctx context.Context, arg database.GetGroupByOrgAndNameParams) (database.Group, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetGroupByOrgAndName", arg0, arg1) + ret := m.ctrl.Call(m, "GetGroupByOrgAndName", ctx, arg) ret0, _ := ret[0].(database.Group) ret1, _ := ret[1].(error) return ret0, ret1 } // GetGroupByOrgAndName indicates an expected call of GetGroupByOrgAndName. -func (mr *MockStoreMockRecorder) GetGroupByOrgAndName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetGroupByOrgAndName(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupByOrgAndName", reflect.TypeOf((*MockStore)(nil).GetGroupByOrgAndName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupByOrgAndName", reflect.TypeOf((*MockStore)(nil).GetGroupByOrgAndName), ctx, arg) } // GetGroupMembers mocks base method. -func (m *MockStore) GetGroupMembers(arg0 context.Context) ([]database.GroupMember, error) { +func (m *MockStore) GetGroupMembers(ctx context.Context) ([]database.GroupMember, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetGroupMembers", arg0) + ret := m.ctrl.Call(m, "GetGroupMembers", ctx) ret0, _ := ret[0].([]database.GroupMember) ret1, _ := ret[1].(error) return ret0, ret1 } // GetGroupMembers indicates an expected call of GetGroupMembers. -func (mr *MockStoreMockRecorder) GetGroupMembers(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetGroupMembers(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupMembers", reflect.TypeOf((*MockStore)(nil).GetGroupMembers), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupMembers", reflect.TypeOf((*MockStore)(nil).GetGroupMembers), ctx) } // GetGroupMembersByGroupID mocks base method. -func (m *MockStore) GetGroupMembersByGroupID(arg0 context.Context, arg1 uuid.UUID) ([]database.GroupMember, error) { +func (m *MockStore) GetGroupMembersByGroupID(ctx context.Context, groupID uuid.UUID) ([]database.GroupMember, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetGroupMembersByGroupID", arg0, arg1) + ret := m.ctrl.Call(m, "GetGroupMembersByGroupID", ctx, groupID) ret0, _ := ret[0].([]database.GroupMember) ret1, _ := ret[1].(error) return ret0, ret1 } // GetGroupMembersByGroupID indicates an expected call of GetGroupMembersByGroupID. -func (mr *MockStoreMockRecorder) GetGroupMembersByGroupID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetGroupMembersByGroupID(ctx, groupID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupMembersByGroupID", reflect.TypeOf((*MockStore)(nil).GetGroupMembersByGroupID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupMembersByGroupID", reflect.TypeOf((*MockStore)(nil).GetGroupMembersByGroupID), ctx, groupID) } // GetGroupMembersCountByGroupID mocks base method. -func (m *MockStore) GetGroupMembersCountByGroupID(arg0 context.Context, arg1 uuid.UUID) (int64, error) { +func (m *MockStore) GetGroupMembersCountByGroupID(ctx context.Context, groupID uuid.UUID) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetGroupMembersCountByGroupID", arg0, arg1) + ret := m.ctrl.Call(m, "GetGroupMembersCountByGroupID", ctx, groupID) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetGroupMembersCountByGroupID indicates an expected call of GetGroupMembersCountByGroupID. -func (mr *MockStoreMockRecorder) GetGroupMembersCountByGroupID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetGroupMembersCountByGroupID(ctx, groupID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupMembersCountByGroupID", reflect.TypeOf((*MockStore)(nil).GetGroupMembersCountByGroupID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroupMembersCountByGroupID", reflect.TypeOf((*MockStore)(nil).GetGroupMembersCountByGroupID), ctx, groupID) } // GetGroups mocks base method. -func (m *MockStore) GetGroups(arg0 context.Context, arg1 database.GetGroupsParams) ([]database.GetGroupsRow, error) { +func (m *MockStore) GetGroups(ctx context.Context, arg database.GetGroupsParams) ([]database.GetGroupsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetGroups", arg0, arg1) + ret := m.ctrl.Call(m, "GetGroups", ctx, arg) ret0, _ := ret[0].([]database.GetGroupsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetGroups indicates an expected call of GetGroups. -func (mr *MockStoreMockRecorder) GetGroups(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetGroups(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroups", reflect.TypeOf((*MockStore)(nil).GetGroups), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGroups", reflect.TypeOf((*MockStore)(nil).GetGroups), ctx, arg) } // GetHealthSettings mocks base method. -func (m *MockStore) GetHealthSettings(arg0 context.Context) (string, error) { +func (m *MockStore) GetHealthSettings(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetHealthSettings", arg0) + ret := m.ctrl.Call(m, "GetHealthSettings", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetHealthSettings indicates an expected call of GetHealthSettings. -func (mr *MockStoreMockRecorder) GetHealthSettings(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetHealthSettings(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHealthSettings", reflect.TypeOf((*MockStore)(nil).GetHealthSettings), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHealthSettings", reflect.TypeOf((*MockStore)(nil).GetHealthSettings), ctx) } // GetHungProvisionerJobs mocks base method. -func (m *MockStore) GetHungProvisionerJobs(arg0 context.Context, arg1 time.Time) ([]database.ProvisionerJob, error) { +func (m *MockStore) GetHungProvisionerJobs(ctx context.Context, updatedAt time.Time) ([]database.ProvisionerJob, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetHungProvisionerJobs", arg0, arg1) + ret := m.ctrl.Call(m, "GetHungProvisionerJobs", ctx, updatedAt) ret0, _ := ret[0].([]database.ProvisionerJob) ret1, _ := ret[1].(error) return ret0, ret1 } // GetHungProvisionerJobs indicates an expected call of GetHungProvisionerJobs. -func (mr *MockStoreMockRecorder) GetHungProvisionerJobs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetHungProvisionerJobs(ctx, updatedAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHungProvisionerJobs", reflect.TypeOf((*MockStore)(nil).GetHungProvisionerJobs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHungProvisionerJobs", reflect.TypeOf((*MockStore)(nil).GetHungProvisionerJobs), ctx, updatedAt) } // GetJFrogXrayScanByWorkspaceAndAgentID mocks base method. -func (m *MockStore) GetJFrogXrayScanByWorkspaceAndAgentID(arg0 context.Context, arg1 database.GetJFrogXrayScanByWorkspaceAndAgentIDParams) (database.JfrogXrayScan, error) { +func (m *MockStore) GetJFrogXrayScanByWorkspaceAndAgentID(ctx context.Context, arg database.GetJFrogXrayScanByWorkspaceAndAgentIDParams) (database.JfrogXrayScan, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetJFrogXrayScanByWorkspaceAndAgentID", arg0, arg1) + ret := m.ctrl.Call(m, "GetJFrogXrayScanByWorkspaceAndAgentID", ctx, arg) ret0, _ := ret[0].(database.JfrogXrayScan) ret1, _ := ret[1].(error) return ret0, ret1 } // GetJFrogXrayScanByWorkspaceAndAgentID indicates an expected call of GetJFrogXrayScanByWorkspaceAndAgentID. -func (mr *MockStoreMockRecorder) GetJFrogXrayScanByWorkspaceAndAgentID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetJFrogXrayScanByWorkspaceAndAgentID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetJFrogXrayScanByWorkspaceAndAgentID", reflect.TypeOf((*MockStore)(nil).GetJFrogXrayScanByWorkspaceAndAgentID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetJFrogXrayScanByWorkspaceAndAgentID", reflect.TypeOf((*MockStore)(nil).GetJFrogXrayScanByWorkspaceAndAgentID), ctx, arg) } // GetLastUpdateCheck mocks base method. -func (m *MockStore) GetLastUpdateCheck(arg0 context.Context) (string, error) { +func (m *MockStore) GetLastUpdateCheck(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLastUpdateCheck", arg0) + ret := m.ctrl.Call(m, "GetLastUpdateCheck", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetLastUpdateCheck indicates an expected call of GetLastUpdateCheck. -func (mr *MockStoreMockRecorder) GetLastUpdateCheck(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetLastUpdateCheck(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLastUpdateCheck", reflect.TypeOf((*MockStore)(nil).GetLastUpdateCheck), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLastUpdateCheck", reflect.TypeOf((*MockStore)(nil).GetLastUpdateCheck), ctx) } // GetLatestCryptoKeyByFeature mocks base method. -func (m *MockStore) GetLatestCryptoKeyByFeature(arg0 context.Context, arg1 database.CryptoKeyFeature) (database.CryptoKey, error) { +func (m *MockStore) GetLatestCryptoKeyByFeature(ctx context.Context, feature database.CryptoKeyFeature) (database.CryptoKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLatestCryptoKeyByFeature", arg0, arg1) + ret := m.ctrl.Call(m, "GetLatestCryptoKeyByFeature", ctx, feature) ret0, _ := ret[0].(database.CryptoKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetLatestCryptoKeyByFeature indicates an expected call of GetLatestCryptoKeyByFeature. -func (mr *MockStoreMockRecorder) GetLatestCryptoKeyByFeature(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetLatestCryptoKeyByFeature(ctx, feature any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestCryptoKeyByFeature", reflect.TypeOf((*MockStore)(nil).GetLatestCryptoKeyByFeature), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestCryptoKeyByFeature", reflect.TypeOf((*MockStore)(nil).GetLatestCryptoKeyByFeature), ctx, feature) } // GetLatestWorkspaceBuildByWorkspaceID mocks base method. -func (m *MockStore) GetLatestWorkspaceBuildByWorkspaceID(arg0 context.Context, arg1 uuid.UUID) (database.WorkspaceBuild, error) { +func (m *MockStore) GetLatestWorkspaceBuildByWorkspaceID(ctx context.Context, workspaceID uuid.UUID) (database.WorkspaceBuild, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLatestWorkspaceBuildByWorkspaceID", arg0, arg1) + ret := m.ctrl.Call(m, "GetLatestWorkspaceBuildByWorkspaceID", ctx, workspaceID) ret0, _ := ret[0].(database.WorkspaceBuild) ret1, _ := ret[1].(error) return ret0, ret1 } // GetLatestWorkspaceBuildByWorkspaceID indicates an expected call of GetLatestWorkspaceBuildByWorkspaceID. -func (mr *MockStoreMockRecorder) GetLatestWorkspaceBuildByWorkspaceID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetLatestWorkspaceBuildByWorkspaceID(ctx, workspaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestWorkspaceBuildByWorkspaceID", reflect.TypeOf((*MockStore)(nil).GetLatestWorkspaceBuildByWorkspaceID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestWorkspaceBuildByWorkspaceID", reflect.TypeOf((*MockStore)(nil).GetLatestWorkspaceBuildByWorkspaceID), ctx, workspaceID) } // GetLatestWorkspaceBuilds mocks base method. -func (m *MockStore) GetLatestWorkspaceBuilds(arg0 context.Context) ([]database.WorkspaceBuild, error) { +func (m *MockStore) GetLatestWorkspaceBuilds(ctx context.Context) ([]database.WorkspaceBuild, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLatestWorkspaceBuilds", arg0) + ret := m.ctrl.Call(m, "GetLatestWorkspaceBuilds", ctx) ret0, _ := ret[0].([]database.WorkspaceBuild) ret1, _ := ret[1].(error) return ret0, ret1 } // GetLatestWorkspaceBuilds indicates an expected call of GetLatestWorkspaceBuilds. -func (mr *MockStoreMockRecorder) GetLatestWorkspaceBuilds(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetLatestWorkspaceBuilds(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestWorkspaceBuilds", reflect.TypeOf((*MockStore)(nil).GetLatestWorkspaceBuilds), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestWorkspaceBuilds", reflect.TypeOf((*MockStore)(nil).GetLatestWorkspaceBuilds), ctx) } // GetLatestWorkspaceBuildsByWorkspaceIDs mocks base method. -func (m *MockStore) GetLatestWorkspaceBuildsByWorkspaceIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.WorkspaceBuild, error) { +func (m *MockStore) GetLatestWorkspaceBuildsByWorkspaceIDs(ctx context.Context, ids []uuid.UUID) ([]database.WorkspaceBuild, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLatestWorkspaceBuildsByWorkspaceIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetLatestWorkspaceBuildsByWorkspaceIDs", ctx, ids) ret0, _ := ret[0].([]database.WorkspaceBuild) ret1, _ := ret[1].(error) return ret0, ret1 } // GetLatestWorkspaceBuildsByWorkspaceIDs indicates an expected call of GetLatestWorkspaceBuildsByWorkspaceIDs. -func (mr *MockStoreMockRecorder) GetLatestWorkspaceBuildsByWorkspaceIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetLatestWorkspaceBuildsByWorkspaceIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestWorkspaceBuildsByWorkspaceIDs", reflect.TypeOf((*MockStore)(nil).GetLatestWorkspaceBuildsByWorkspaceIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestWorkspaceBuildsByWorkspaceIDs", reflect.TypeOf((*MockStore)(nil).GetLatestWorkspaceBuildsByWorkspaceIDs), ctx, ids) } // GetLicenseByID mocks base method. -func (m *MockStore) GetLicenseByID(arg0 context.Context, arg1 int32) (database.License, error) { +func (m *MockStore) GetLicenseByID(ctx context.Context, id int32) (database.License, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLicenseByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetLicenseByID", ctx, id) ret0, _ := ret[0].(database.License) ret1, _ := ret[1].(error) return ret0, ret1 } // GetLicenseByID indicates an expected call of GetLicenseByID. -func (mr *MockStoreMockRecorder) GetLicenseByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetLicenseByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLicenseByID", reflect.TypeOf((*MockStore)(nil).GetLicenseByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLicenseByID", reflect.TypeOf((*MockStore)(nil).GetLicenseByID), ctx, id) } // GetLicenses mocks base method. -func (m *MockStore) GetLicenses(arg0 context.Context) ([]database.License, error) { +func (m *MockStore) GetLicenses(ctx context.Context) ([]database.License, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLicenses", arg0) + ret := m.ctrl.Call(m, "GetLicenses", ctx) ret0, _ := ret[0].([]database.License) ret1, _ := ret[1].(error) return ret0, ret1 } // GetLicenses indicates an expected call of GetLicenses. -func (mr *MockStoreMockRecorder) GetLicenses(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetLicenses(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLicenses", reflect.TypeOf((*MockStore)(nil).GetLicenses), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLicenses", reflect.TypeOf((*MockStore)(nil).GetLicenses), ctx) } // GetLogoURL mocks base method. -func (m *MockStore) GetLogoURL(arg0 context.Context) (string, error) { +func (m *MockStore) GetLogoURL(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetLogoURL", arg0) + ret := m.ctrl.Call(m, "GetLogoURL", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetLogoURL indicates an expected call of GetLogoURL. -func (mr *MockStoreMockRecorder) GetLogoURL(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetLogoURL(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLogoURL", reflect.TypeOf((*MockStore)(nil).GetLogoURL), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLogoURL", reflect.TypeOf((*MockStore)(nil).GetLogoURL), ctx) } // GetNotificationMessagesByStatus mocks base method. -func (m *MockStore) GetNotificationMessagesByStatus(arg0 context.Context, arg1 database.GetNotificationMessagesByStatusParams) ([]database.NotificationMessage, error) { +func (m *MockStore) GetNotificationMessagesByStatus(ctx context.Context, arg database.GetNotificationMessagesByStatusParams) ([]database.NotificationMessage, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetNotificationMessagesByStatus", arg0, arg1) + ret := m.ctrl.Call(m, "GetNotificationMessagesByStatus", ctx, arg) ret0, _ := ret[0].([]database.NotificationMessage) ret1, _ := ret[1].(error) return ret0, ret1 } // GetNotificationMessagesByStatus indicates an expected call of GetNotificationMessagesByStatus. -func (mr *MockStoreMockRecorder) GetNotificationMessagesByStatus(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetNotificationMessagesByStatus(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationMessagesByStatus", reflect.TypeOf((*MockStore)(nil).GetNotificationMessagesByStatus), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationMessagesByStatus", reflect.TypeOf((*MockStore)(nil).GetNotificationMessagesByStatus), ctx, arg) } // GetNotificationReportGeneratorLogByTemplate mocks base method. -func (m *MockStore) GetNotificationReportGeneratorLogByTemplate(arg0 context.Context, arg1 uuid.UUID) (database.NotificationReportGeneratorLog, error) { +func (m *MockStore) GetNotificationReportGeneratorLogByTemplate(ctx context.Context, templateID uuid.UUID) (database.NotificationReportGeneratorLog, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetNotificationReportGeneratorLogByTemplate", arg0, arg1) + ret := m.ctrl.Call(m, "GetNotificationReportGeneratorLogByTemplate", ctx, templateID) ret0, _ := ret[0].(database.NotificationReportGeneratorLog) ret1, _ := ret[1].(error) return ret0, ret1 } // GetNotificationReportGeneratorLogByTemplate indicates an expected call of GetNotificationReportGeneratorLogByTemplate. -func (mr *MockStoreMockRecorder) GetNotificationReportGeneratorLogByTemplate(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetNotificationReportGeneratorLogByTemplate(ctx, templateID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationReportGeneratorLogByTemplate", reflect.TypeOf((*MockStore)(nil).GetNotificationReportGeneratorLogByTemplate), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationReportGeneratorLogByTemplate", reflect.TypeOf((*MockStore)(nil).GetNotificationReportGeneratorLogByTemplate), ctx, templateID) } // GetNotificationTemplateByID mocks base method. -func (m *MockStore) GetNotificationTemplateByID(arg0 context.Context, arg1 uuid.UUID) (database.NotificationTemplate, error) { +func (m *MockStore) GetNotificationTemplateByID(ctx context.Context, id uuid.UUID) (database.NotificationTemplate, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetNotificationTemplateByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetNotificationTemplateByID", ctx, id) ret0, _ := ret[0].(database.NotificationTemplate) ret1, _ := ret[1].(error) return ret0, ret1 } // GetNotificationTemplateByID indicates an expected call of GetNotificationTemplateByID. -func (mr *MockStoreMockRecorder) GetNotificationTemplateByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetNotificationTemplateByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationTemplateByID", reflect.TypeOf((*MockStore)(nil).GetNotificationTemplateByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationTemplateByID", reflect.TypeOf((*MockStore)(nil).GetNotificationTemplateByID), ctx, id) } // GetNotificationTemplatesByKind mocks base method. -func (m *MockStore) GetNotificationTemplatesByKind(arg0 context.Context, arg1 database.NotificationTemplateKind) ([]database.NotificationTemplate, error) { +func (m *MockStore) GetNotificationTemplatesByKind(ctx context.Context, kind database.NotificationTemplateKind) ([]database.NotificationTemplate, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetNotificationTemplatesByKind", arg0, arg1) + ret := m.ctrl.Call(m, "GetNotificationTemplatesByKind", ctx, kind) ret0, _ := ret[0].([]database.NotificationTemplate) ret1, _ := ret[1].(error) return ret0, ret1 } // GetNotificationTemplatesByKind indicates an expected call of GetNotificationTemplatesByKind. -func (mr *MockStoreMockRecorder) GetNotificationTemplatesByKind(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetNotificationTemplatesByKind(ctx, kind any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationTemplatesByKind", reflect.TypeOf((*MockStore)(nil).GetNotificationTemplatesByKind), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationTemplatesByKind", reflect.TypeOf((*MockStore)(nil).GetNotificationTemplatesByKind), ctx, kind) } // GetNotificationsSettings mocks base method. -func (m *MockStore) GetNotificationsSettings(arg0 context.Context) (string, error) { +func (m *MockStore) GetNotificationsSettings(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetNotificationsSettings", arg0) + ret := m.ctrl.Call(m, "GetNotificationsSettings", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetNotificationsSettings indicates an expected call of GetNotificationsSettings. -func (mr *MockStoreMockRecorder) GetNotificationsSettings(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetNotificationsSettings(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationsSettings", reflect.TypeOf((*MockStore)(nil).GetNotificationsSettings), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNotificationsSettings", reflect.TypeOf((*MockStore)(nil).GetNotificationsSettings), ctx) } // GetOAuth2ProviderAppByID mocks base method. -func (m *MockStore) GetOAuth2ProviderAppByID(arg0 context.Context, arg1 uuid.UUID) (database.OAuth2ProviderApp, error) { +func (m *MockStore) GetOAuth2ProviderAppByID(ctx context.Context, id uuid.UUID) (database.OAuth2ProviderApp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderAppByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetOAuth2ProviderAppByID", ctx, id) ret0, _ := ret[0].(database.OAuth2ProviderApp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuth2ProviderAppByID indicates an expected call of GetOAuth2ProviderAppByID. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppByID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppByID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppByID), ctx, id) } // GetOAuth2ProviderAppCodeByID mocks base method. -func (m *MockStore) GetOAuth2ProviderAppCodeByID(arg0 context.Context, arg1 uuid.UUID) (database.OAuth2ProviderAppCode, error) { +func (m *MockStore) GetOAuth2ProviderAppCodeByID(ctx context.Context, id uuid.UUID) (database.OAuth2ProviderAppCode, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderAppCodeByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetOAuth2ProviderAppCodeByID", ctx, id) ret0, _ := ret[0].(database.OAuth2ProviderAppCode) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuth2ProviderAppCodeByID indicates an expected call of GetOAuth2ProviderAppCodeByID. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppCodeByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppCodeByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppCodeByID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppCodeByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppCodeByID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppCodeByID), ctx, id) } // GetOAuth2ProviderAppCodeByPrefix mocks base method. -func (m *MockStore) GetOAuth2ProviderAppCodeByPrefix(arg0 context.Context, arg1 []byte) (database.OAuth2ProviderAppCode, error) { +func (m *MockStore) GetOAuth2ProviderAppCodeByPrefix(ctx context.Context, secretPrefix []byte) (database.OAuth2ProviderAppCode, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderAppCodeByPrefix", arg0, arg1) + ret := m.ctrl.Call(m, "GetOAuth2ProviderAppCodeByPrefix", ctx, secretPrefix) ret0, _ := ret[0].(database.OAuth2ProviderAppCode) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuth2ProviderAppCodeByPrefix indicates an expected call of GetOAuth2ProviderAppCodeByPrefix. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppCodeByPrefix(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppCodeByPrefix(ctx, secretPrefix any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppCodeByPrefix", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppCodeByPrefix), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppCodeByPrefix", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppCodeByPrefix), ctx, secretPrefix) } // GetOAuth2ProviderAppSecretByID mocks base method. -func (m *MockStore) GetOAuth2ProviderAppSecretByID(arg0 context.Context, arg1 uuid.UUID) (database.OAuth2ProviderAppSecret, error) { +func (m *MockStore) GetOAuth2ProviderAppSecretByID(ctx context.Context, id uuid.UUID) (database.OAuth2ProviderAppSecret, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderAppSecretByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetOAuth2ProviderAppSecretByID", ctx, id) ret0, _ := ret[0].(database.OAuth2ProviderAppSecret) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuth2ProviderAppSecretByID indicates an expected call of GetOAuth2ProviderAppSecretByID. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppSecretByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppSecretByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppSecretByID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppSecretByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppSecretByID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppSecretByID), ctx, id) } // GetOAuth2ProviderAppSecretByPrefix mocks base method. -func (m *MockStore) GetOAuth2ProviderAppSecretByPrefix(arg0 context.Context, arg1 []byte) (database.OAuth2ProviderAppSecret, error) { +func (m *MockStore) GetOAuth2ProviderAppSecretByPrefix(ctx context.Context, secretPrefix []byte) (database.OAuth2ProviderAppSecret, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderAppSecretByPrefix", arg0, arg1) + ret := m.ctrl.Call(m, "GetOAuth2ProviderAppSecretByPrefix", ctx, secretPrefix) ret0, _ := ret[0].(database.OAuth2ProviderAppSecret) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuth2ProviderAppSecretByPrefix indicates an expected call of GetOAuth2ProviderAppSecretByPrefix. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppSecretByPrefix(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppSecretByPrefix(ctx, secretPrefix any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppSecretByPrefix", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppSecretByPrefix), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppSecretByPrefix", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppSecretByPrefix), ctx, secretPrefix) } // GetOAuth2ProviderAppSecretsByAppID mocks base method. -func (m *MockStore) GetOAuth2ProviderAppSecretsByAppID(arg0 context.Context, arg1 uuid.UUID) ([]database.OAuth2ProviderAppSecret, error) { +func (m *MockStore) GetOAuth2ProviderAppSecretsByAppID(ctx context.Context, appID uuid.UUID) ([]database.OAuth2ProviderAppSecret, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderAppSecretsByAppID", arg0, arg1) + ret := m.ctrl.Call(m, "GetOAuth2ProviderAppSecretsByAppID", ctx, appID) ret0, _ := ret[0].([]database.OAuth2ProviderAppSecret) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuth2ProviderAppSecretsByAppID indicates an expected call of GetOAuth2ProviderAppSecretsByAppID. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppSecretsByAppID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppSecretsByAppID(ctx, appID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppSecretsByAppID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppSecretsByAppID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppSecretsByAppID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppSecretsByAppID), ctx, appID) } // GetOAuth2ProviderAppTokenByPrefix mocks base method. -func (m *MockStore) GetOAuth2ProviderAppTokenByPrefix(arg0 context.Context, arg1 []byte) (database.OAuth2ProviderAppToken, error) { +func (m *MockStore) GetOAuth2ProviderAppTokenByPrefix(ctx context.Context, hashPrefix []byte) (database.OAuth2ProviderAppToken, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderAppTokenByPrefix", arg0, arg1) + ret := m.ctrl.Call(m, "GetOAuth2ProviderAppTokenByPrefix", ctx, hashPrefix) ret0, _ := ret[0].(database.OAuth2ProviderAppToken) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuth2ProviderAppTokenByPrefix indicates an expected call of GetOAuth2ProviderAppTokenByPrefix. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppTokenByPrefix(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppTokenByPrefix(ctx, hashPrefix any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppTokenByPrefix", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppTokenByPrefix), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppTokenByPrefix", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppTokenByPrefix), ctx, hashPrefix) } // GetOAuth2ProviderApps mocks base method. -func (m *MockStore) GetOAuth2ProviderApps(arg0 context.Context) ([]database.OAuth2ProviderApp, error) { +func (m *MockStore) GetOAuth2ProviderApps(ctx context.Context) ([]database.OAuth2ProviderApp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderApps", arg0) + ret := m.ctrl.Call(m, "GetOAuth2ProviderApps", ctx) ret0, _ := ret[0].([]database.OAuth2ProviderApp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuth2ProviderApps indicates an expected call of GetOAuth2ProviderApps. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderApps(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuth2ProviderApps(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderApps", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderApps), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderApps", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderApps), ctx) } // GetOAuth2ProviderAppsByUserID mocks base method. -func (m *MockStore) GetOAuth2ProviderAppsByUserID(arg0 context.Context, arg1 uuid.UUID) ([]database.GetOAuth2ProviderAppsByUserIDRow, error) { +func (m *MockStore) GetOAuth2ProviderAppsByUserID(ctx context.Context, userID uuid.UUID) ([]database.GetOAuth2ProviderAppsByUserIDRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuth2ProviderAppsByUserID", arg0, arg1) + ret := m.ctrl.Call(m, "GetOAuth2ProviderAppsByUserID", ctx, userID) ret0, _ := ret[0].([]database.GetOAuth2ProviderAppsByUserIDRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuth2ProviderAppsByUserID indicates an expected call of GetOAuth2ProviderAppsByUserID. -func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppsByUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuth2ProviderAppsByUserID(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppsByUserID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppsByUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuth2ProviderAppsByUserID", reflect.TypeOf((*MockStore)(nil).GetOAuth2ProviderAppsByUserID), ctx, userID) } // GetOAuthSigningKey mocks base method. -func (m *MockStore) GetOAuthSigningKey(arg0 context.Context) (string, error) { +func (m *MockStore) GetOAuthSigningKey(ctx context.Context) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOAuthSigningKey", arg0) + ret := m.ctrl.Call(m, "GetOAuthSigningKey", ctx) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOAuthSigningKey indicates an expected call of GetOAuthSigningKey. -func (mr *MockStoreMockRecorder) GetOAuthSigningKey(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOAuthSigningKey(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuthSigningKey", reflect.TypeOf((*MockStore)(nil).GetOAuthSigningKey), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOAuthSigningKey", reflect.TypeOf((*MockStore)(nil).GetOAuthSigningKey), ctx) } // GetOrganizationByID mocks base method. -func (m *MockStore) GetOrganizationByID(arg0 context.Context, arg1 uuid.UUID) (database.Organization, error) { +func (m *MockStore) GetOrganizationByID(ctx context.Context, id uuid.UUID) (database.Organization, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOrganizationByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetOrganizationByID", ctx, id) ret0, _ := ret[0].(database.Organization) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOrganizationByID indicates an expected call of GetOrganizationByID. -func (mr *MockStoreMockRecorder) GetOrganizationByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOrganizationByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizationByID", reflect.TypeOf((*MockStore)(nil).GetOrganizationByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizationByID", reflect.TypeOf((*MockStore)(nil).GetOrganizationByID), ctx, id) } // GetOrganizationByName mocks base method. -func (m *MockStore) GetOrganizationByName(arg0 context.Context, arg1 string) (database.Organization, error) { +func (m *MockStore) GetOrganizationByName(ctx context.Context, name string) (database.Organization, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOrganizationByName", arg0, arg1) + ret := m.ctrl.Call(m, "GetOrganizationByName", ctx, name) ret0, _ := ret[0].(database.Organization) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOrganizationByName indicates an expected call of GetOrganizationByName. -func (mr *MockStoreMockRecorder) GetOrganizationByName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOrganizationByName(ctx, name any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizationByName", reflect.TypeOf((*MockStore)(nil).GetOrganizationByName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizationByName", reflect.TypeOf((*MockStore)(nil).GetOrganizationByName), ctx, name) } // GetOrganizationIDsByMemberIDs mocks base method. -func (m *MockStore) GetOrganizationIDsByMemberIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.GetOrganizationIDsByMemberIDsRow, error) { +func (m *MockStore) GetOrganizationIDsByMemberIDs(ctx context.Context, ids []uuid.UUID) ([]database.GetOrganizationIDsByMemberIDsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOrganizationIDsByMemberIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetOrganizationIDsByMemberIDs", ctx, ids) ret0, _ := ret[0].([]database.GetOrganizationIDsByMemberIDsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOrganizationIDsByMemberIDs indicates an expected call of GetOrganizationIDsByMemberIDs. -func (mr *MockStoreMockRecorder) GetOrganizationIDsByMemberIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOrganizationIDsByMemberIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizationIDsByMemberIDs", reflect.TypeOf((*MockStore)(nil).GetOrganizationIDsByMemberIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizationIDsByMemberIDs", reflect.TypeOf((*MockStore)(nil).GetOrganizationIDsByMemberIDs), ctx, ids) } // GetOrganizations mocks base method. -func (m *MockStore) GetOrganizations(arg0 context.Context, arg1 database.GetOrganizationsParams) ([]database.Organization, error) { +func (m *MockStore) GetOrganizations(ctx context.Context, arg database.GetOrganizationsParams) ([]database.Organization, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOrganizations", arg0, arg1) + ret := m.ctrl.Call(m, "GetOrganizations", ctx, arg) ret0, _ := ret[0].([]database.Organization) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOrganizations indicates an expected call of GetOrganizations. -func (mr *MockStoreMockRecorder) GetOrganizations(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOrganizations(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizations", reflect.TypeOf((*MockStore)(nil).GetOrganizations), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizations", reflect.TypeOf((*MockStore)(nil).GetOrganizations), ctx, arg) } // GetOrganizationsByUserID mocks base method. -func (m *MockStore) GetOrganizationsByUserID(arg0 context.Context, arg1 uuid.UUID) ([]database.Organization, error) { +func (m *MockStore) GetOrganizationsByUserID(ctx context.Context, userID uuid.UUID) ([]database.Organization, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetOrganizationsByUserID", arg0, arg1) + ret := m.ctrl.Call(m, "GetOrganizationsByUserID", ctx, userID) ret0, _ := ret[0].([]database.Organization) ret1, _ := ret[1].(error) return ret0, ret1 } // GetOrganizationsByUserID indicates an expected call of GetOrganizationsByUserID. -func (mr *MockStoreMockRecorder) GetOrganizationsByUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetOrganizationsByUserID(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizationsByUserID", reflect.TypeOf((*MockStore)(nil).GetOrganizationsByUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrganizationsByUserID", reflect.TypeOf((*MockStore)(nil).GetOrganizationsByUserID), ctx, userID) } // GetParameterSchemasByJobID mocks base method. -func (m *MockStore) GetParameterSchemasByJobID(arg0 context.Context, arg1 uuid.UUID) ([]database.ParameterSchema, error) { +func (m *MockStore) GetParameterSchemasByJobID(ctx context.Context, jobID uuid.UUID) ([]database.ParameterSchema, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetParameterSchemasByJobID", arg0, arg1) + ret := m.ctrl.Call(m, "GetParameterSchemasByJobID", ctx, jobID) ret0, _ := ret[0].([]database.ParameterSchema) ret1, _ := ret[1].(error) return ret0, ret1 } // GetParameterSchemasByJobID indicates an expected call of GetParameterSchemasByJobID. -func (mr *MockStoreMockRecorder) GetParameterSchemasByJobID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetParameterSchemasByJobID(ctx, jobID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetParameterSchemasByJobID", reflect.TypeOf((*MockStore)(nil).GetParameterSchemasByJobID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetParameterSchemasByJobID", reflect.TypeOf((*MockStore)(nil).GetParameterSchemasByJobID), ctx, jobID) } // GetPreviousTemplateVersion mocks base method. -func (m *MockStore) GetPreviousTemplateVersion(arg0 context.Context, arg1 database.GetPreviousTemplateVersionParams) (database.TemplateVersion, error) { +func (m *MockStore) GetPreviousTemplateVersion(ctx context.Context, arg database.GetPreviousTemplateVersionParams) (database.TemplateVersion, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetPreviousTemplateVersion", arg0, arg1) + ret := m.ctrl.Call(m, "GetPreviousTemplateVersion", ctx, arg) ret0, _ := ret[0].(database.TemplateVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // GetPreviousTemplateVersion indicates an expected call of GetPreviousTemplateVersion. -func (mr *MockStoreMockRecorder) GetPreviousTemplateVersion(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetPreviousTemplateVersion(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPreviousTemplateVersion", reflect.TypeOf((*MockStore)(nil).GetPreviousTemplateVersion), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPreviousTemplateVersion", reflect.TypeOf((*MockStore)(nil).GetPreviousTemplateVersion), ctx, arg) } // GetProvisionerDaemons mocks base method. -func (m *MockStore) GetProvisionerDaemons(arg0 context.Context) ([]database.ProvisionerDaemon, error) { +func (m *MockStore) GetProvisionerDaemons(ctx context.Context) ([]database.ProvisionerDaemon, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerDaemons", arg0) + ret := m.ctrl.Call(m, "GetProvisionerDaemons", ctx) ret0, _ := ret[0].([]database.ProvisionerDaemon) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerDaemons indicates an expected call of GetProvisionerDaemons. -func (mr *MockStoreMockRecorder) GetProvisionerDaemons(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerDaemons(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerDaemons", reflect.TypeOf((*MockStore)(nil).GetProvisionerDaemons), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerDaemons", reflect.TypeOf((*MockStore)(nil).GetProvisionerDaemons), ctx) } // GetProvisionerDaemonsByOrganization mocks base method. -func (m *MockStore) GetProvisionerDaemonsByOrganization(arg0 context.Context, arg1 database.GetProvisionerDaemonsByOrganizationParams) ([]database.ProvisionerDaemon, error) { +func (m *MockStore) GetProvisionerDaemonsByOrganization(ctx context.Context, arg database.GetProvisionerDaemonsByOrganizationParams) ([]database.ProvisionerDaemon, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerDaemonsByOrganization", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerDaemonsByOrganization", ctx, arg) ret0, _ := ret[0].([]database.ProvisionerDaemon) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerDaemonsByOrganization indicates an expected call of GetProvisionerDaemonsByOrganization. -func (mr *MockStoreMockRecorder) GetProvisionerDaemonsByOrganization(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerDaemonsByOrganization(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerDaemonsByOrganization", reflect.TypeOf((*MockStore)(nil).GetProvisionerDaemonsByOrganization), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerDaemonsByOrganization", reflect.TypeOf((*MockStore)(nil).GetProvisionerDaemonsByOrganization), ctx, arg) } // GetProvisionerJobByID mocks base method. -func (m *MockStore) GetProvisionerJobByID(arg0 context.Context, arg1 uuid.UUID) (database.ProvisionerJob, error) { +func (m *MockStore) GetProvisionerJobByID(ctx context.Context, id uuid.UUID) (database.ProvisionerJob, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerJobByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerJobByID", ctx, id) ret0, _ := ret[0].(database.ProvisionerJob) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerJobByID indicates an expected call of GetProvisionerJobByID. -func (mr *MockStoreMockRecorder) GetProvisionerJobByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerJobByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobByID", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobByID", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobByID), ctx, id) } // GetProvisionerJobTimingsByJobID mocks base method. -func (m *MockStore) GetProvisionerJobTimingsByJobID(arg0 context.Context, arg1 uuid.UUID) ([]database.ProvisionerJobTiming, error) { +func (m *MockStore) GetProvisionerJobTimingsByJobID(ctx context.Context, jobID uuid.UUID) ([]database.ProvisionerJobTiming, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerJobTimingsByJobID", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerJobTimingsByJobID", ctx, jobID) ret0, _ := ret[0].([]database.ProvisionerJobTiming) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerJobTimingsByJobID indicates an expected call of GetProvisionerJobTimingsByJobID. -func (mr *MockStoreMockRecorder) GetProvisionerJobTimingsByJobID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerJobTimingsByJobID(ctx, jobID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobTimingsByJobID", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobTimingsByJobID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobTimingsByJobID", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobTimingsByJobID), ctx, jobID) } // GetProvisionerJobsByIDs mocks base method. -func (m *MockStore) GetProvisionerJobsByIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.ProvisionerJob, error) { +func (m *MockStore) GetProvisionerJobsByIDs(ctx context.Context, ids []uuid.UUID) ([]database.ProvisionerJob, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerJobsByIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerJobsByIDs", ctx, ids) ret0, _ := ret[0].([]database.ProvisionerJob) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerJobsByIDs indicates an expected call of GetProvisionerJobsByIDs. -func (mr *MockStoreMockRecorder) GetProvisionerJobsByIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerJobsByIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobsByIDs", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobsByIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobsByIDs", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobsByIDs), ctx, ids) } // GetProvisionerJobsByIDsWithQueuePosition mocks base method. -func (m *MockStore) GetProvisionerJobsByIDsWithQueuePosition(arg0 context.Context, arg1 []uuid.UUID) ([]database.GetProvisionerJobsByIDsWithQueuePositionRow, error) { +func (m *MockStore) GetProvisionerJobsByIDsWithQueuePosition(ctx context.Context, ids []uuid.UUID) ([]database.GetProvisionerJobsByIDsWithQueuePositionRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerJobsByIDsWithQueuePosition", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerJobsByIDsWithQueuePosition", ctx, ids) ret0, _ := ret[0].([]database.GetProvisionerJobsByIDsWithQueuePositionRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerJobsByIDsWithQueuePosition indicates an expected call of GetProvisionerJobsByIDsWithQueuePosition. -func (mr *MockStoreMockRecorder) GetProvisionerJobsByIDsWithQueuePosition(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerJobsByIDsWithQueuePosition(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobsByIDsWithQueuePosition", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobsByIDsWithQueuePosition), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobsByIDsWithQueuePosition", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobsByIDsWithQueuePosition), ctx, ids) } // GetProvisionerJobsCreatedAfter mocks base method. -func (m *MockStore) GetProvisionerJobsCreatedAfter(arg0 context.Context, arg1 time.Time) ([]database.ProvisionerJob, error) { +func (m *MockStore) GetProvisionerJobsCreatedAfter(ctx context.Context, createdAt time.Time) ([]database.ProvisionerJob, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerJobsCreatedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerJobsCreatedAfter", ctx, createdAt) ret0, _ := ret[0].([]database.ProvisionerJob) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerJobsCreatedAfter indicates an expected call of GetProvisionerJobsCreatedAfter. -func (mr *MockStoreMockRecorder) GetProvisionerJobsCreatedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerJobsCreatedAfter(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobsCreatedAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerJobsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetProvisionerJobsCreatedAfter), ctx, createdAt) } // GetProvisionerKeyByHashedSecret mocks base method. -func (m *MockStore) GetProvisionerKeyByHashedSecret(arg0 context.Context, arg1 []byte) (database.ProvisionerKey, error) { +func (m *MockStore) GetProvisionerKeyByHashedSecret(ctx context.Context, hashedSecret []byte) (database.ProvisionerKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerKeyByHashedSecret", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerKeyByHashedSecret", ctx, hashedSecret) ret0, _ := ret[0].(database.ProvisionerKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerKeyByHashedSecret indicates an expected call of GetProvisionerKeyByHashedSecret. -func (mr *MockStoreMockRecorder) GetProvisionerKeyByHashedSecret(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerKeyByHashedSecret(ctx, hashedSecret any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerKeyByHashedSecret", reflect.TypeOf((*MockStore)(nil).GetProvisionerKeyByHashedSecret), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerKeyByHashedSecret", reflect.TypeOf((*MockStore)(nil).GetProvisionerKeyByHashedSecret), ctx, hashedSecret) } // GetProvisionerKeyByID mocks base method. -func (m *MockStore) GetProvisionerKeyByID(arg0 context.Context, arg1 uuid.UUID) (database.ProvisionerKey, error) { +func (m *MockStore) GetProvisionerKeyByID(ctx context.Context, id uuid.UUID) (database.ProvisionerKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerKeyByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerKeyByID", ctx, id) ret0, _ := ret[0].(database.ProvisionerKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerKeyByID indicates an expected call of GetProvisionerKeyByID. -func (mr *MockStoreMockRecorder) GetProvisionerKeyByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerKeyByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerKeyByID", reflect.TypeOf((*MockStore)(nil).GetProvisionerKeyByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerKeyByID", reflect.TypeOf((*MockStore)(nil).GetProvisionerKeyByID), ctx, id) } // GetProvisionerKeyByName mocks base method. -func (m *MockStore) GetProvisionerKeyByName(arg0 context.Context, arg1 database.GetProvisionerKeyByNameParams) (database.ProvisionerKey, error) { +func (m *MockStore) GetProvisionerKeyByName(ctx context.Context, arg database.GetProvisionerKeyByNameParams) (database.ProvisionerKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerKeyByName", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerKeyByName", ctx, arg) ret0, _ := ret[0].(database.ProvisionerKey) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerKeyByName indicates an expected call of GetProvisionerKeyByName. -func (mr *MockStoreMockRecorder) GetProvisionerKeyByName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerKeyByName(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerKeyByName", reflect.TypeOf((*MockStore)(nil).GetProvisionerKeyByName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerKeyByName", reflect.TypeOf((*MockStore)(nil).GetProvisionerKeyByName), ctx, arg) } // GetProvisionerLogsAfterID mocks base method. -func (m *MockStore) GetProvisionerLogsAfterID(arg0 context.Context, arg1 database.GetProvisionerLogsAfterIDParams) ([]database.ProvisionerJobLog, error) { +func (m *MockStore) GetProvisionerLogsAfterID(ctx context.Context, arg database.GetProvisionerLogsAfterIDParams) ([]database.ProvisionerJobLog, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetProvisionerLogsAfterID", arg0, arg1) + ret := m.ctrl.Call(m, "GetProvisionerLogsAfterID", ctx, arg) ret0, _ := ret[0].([]database.ProvisionerJobLog) ret1, _ := ret[1].(error) return ret0, ret1 } // GetProvisionerLogsAfterID indicates an expected call of GetProvisionerLogsAfterID. -func (mr *MockStoreMockRecorder) GetProvisionerLogsAfterID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetProvisionerLogsAfterID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerLogsAfterID", reflect.TypeOf((*MockStore)(nil).GetProvisionerLogsAfterID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProvisionerLogsAfterID", reflect.TypeOf((*MockStore)(nil).GetProvisionerLogsAfterID), ctx, arg) } // GetQuotaAllowanceForUser mocks base method. -func (m *MockStore) GetQuotaAllowanceForUser(arg0 context.Context, arg1 database.GetQuotaAllowanceForUserParams) (int64, error) { +func (m *MockStore) GetQuotaAllowanceForUser(ctx context.Context, arg database.GetQuotaAllowanceForUserParams) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetQuotaAllowanceForUser", arg0, arg1) + ret := m.ctrl.Call(m, "GetQuotaAllowanceForUser", ctx, arg) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetQuotaAllowanceForUser indicates an expected call of GetQuotaAllowanceForUser. -func (mr *MockStoreMockRecorder) GetQuotaAllowanceForUser(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetQuotaAllowanceForUser(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetQuotaAllowanceForUser", reflect.TypeOf((*MockStore)(nil).GetQuotaAllowanceForUser), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetQuotaAllowanceForUser", reflect.TypeOf((*MockStore)(nil).GetQuotaAllowanceForUser), ctx, arg) } // GetQuotaConsumedForUser mocks base method. -func (m *MockStore) GetQuotaConsumedForUser(arg0 context.Context, arg1 database.GetQuotaConsumedForUserParams) (int64, error) { +func (m *MockStore) GetQuotaConsumedForUser(ctx context.Context, arg database.GetQuotaConsumedForUserParams) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetQuotaConsumedForUser", arg0, arg1) + ret := m.ctrl.Call(m, "GetQuotaConsumedForUser", ctx, arg) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetQuotaConsumedForUser indicates an expected call of GetQuotaConsumedForUser. -func (mr *MockStoreMockRecorder) GetQuotaConsumedForUser(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetQuotaConsumedForUser(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetQuotaConsumedForUser", reflect.TypeOf((*MockStore)(nil).GetQuotaConsumedForUser), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetQuotaConsumedForUser", reflect.TypeOf((*MockStore)(nil).GetQuotaConsumedForUser), ctx, arg) } // GetReplicaByID mocks base method. -func (m *MockStore) GetReplicaByID(arg0 context.Context, arg1 uuid.UUID) (database.Replica, error) { +func (m *MockStore) GetReplicaByID(ctx context.Context, id uuid.UUID) (database.Replica, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetReplicaByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetReplicaByID", ctx, id) ret0, _ := ret[0].(database.Replica) ret1, _ := ret[1].(error) return ret0, ret1 } // GetReplicaByID indicates an expected call of GetReplicaByID. -func (mr *MockStoreMockRecorder) GetReplicaByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetReplicaByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetReplicaByID", reflect.TypeOf((*MockStore)(nil).GetReplicaByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetReplicaByID", reflect.TypeOf((*MockStore)(nil).GetReplicaByID), ctx, id) } // GetReplicasUpdatedAfter mocks base method. -func (m *MockStore) GetReplicasUpdatedAfter(arg0 context.Context, arg1 time.Time) ([]database.Replica, error) { +func (m *MockStore) GetReplicasUpdatedAfter(ctx context.Context, updatedAt time.Time) ([]database.Replica, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetReplicasUpdatedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetReplicasUpdatedAfter", ctx, updatedAt) ret0, _ := ret[0].([]database.Replica) ret1, _ := ret[1].(error) return ret0, ret1 } // GetReplicasUpdatedAfter indicates an expected call of GetReplicasUpdatedAfter. -func (mr *MockStoreMockRecorder) GetReplicasUpdatedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetReplicasUpdatedAfter(ctx, updatedAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetReplicasUpdatedAfter", reflect.TypeOf((*MockStore)(nil).GetReplicasUpdatedAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetReplicasUpdatedAfter", reflect.TypeOf((*MockStore)(nil).GetReplicasUpdatedAfter), ctx, updatedAt) } // GetRuntimeConfig mocks base method. -func (m *MockStore) GetRuntimeConfig(arg0 context.Context, arg1 string) (string, error) { +func (m *MockStore) GetRuntimeConfig(ctx context.Context, key string) (string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetRuntimeConfig", arg0, arg1) + ret := m.ctrl.Call(m, "GetRuntimeConfig", ctx, key) ret0, _ := ret[0].(string) ret1, _ := ret[1].(error) return ret0, ret1 } // GetRuntimeConfig indicates an expected call of GetRuntimeConfig. -func (mr *MockStoreMockRecorder) GetRuntimeConfig(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetRuntimeConfig(ctx, key any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRuntimeConfig", reflect.TypeOf((*MockStore)(nil).GetRuntimeConfig), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRuntimeConfig", reflect.TypeOf((*MockStore)(nil).GetRuntimeConfig), ctx, key) } // GetTailnetAgents mocks base method. -func (m *MockStore) GetTailnetAgents(arg0 context.Context, arg1 uuid.UUID) ([]database.TailnetAgent, error) { +func (m *MockStore) GetTailnetAgents(ctx context.Context, id uuid.UUID) ([]database.TailnetAgent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTailnetAgents", arg0, arg1) + ret := m.ctrl.Call(m, "GetTailnetAgents", ctx, id) ret0, _ := ret[0].([]database.TailnetAgent) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTailnetAgents indicates an expected call of GetTailnetAgents. -func (mr *MockStoreMockRecorder) GetTailnetAgents(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTailnetAgents(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetAgents", reflect.TypeOf((*MockStore)(nil).GetTailnetAgents), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetAgents", reflect.TypeOf((*MockStore)(nil).GetTailnetAgents), ctx, id) } // GetTailnetClientsForAgent mocks base method. -func (m *MockStore) GetTailnetClientsForAgent(arg0 context.Context, arg1 uuid.UUID) ([]database.TailnetClient, error) { +func (m *MockStore) GetTailnetClientsForAgent(ctx context.Context, agentID uuid.UUID) ([]database.TailnetClient, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTailnetClientsForAgent", arg0, arg1) + ret := m.ctrl.Call(m, "GetTailnetClientsForAgent", ctx, agentID) ret0, _ := ret[0].([]database.TailnetClient) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTailnetClientsForAgent indicates an expected call of GetTailnetClientsForAgent. -func (mr *MockStoreMockRecorder) GetTailnetClientsForAgent(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTailnetClientsForAgent(ctx, agentID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetClientsForAgent", reflect.TypeOf((*MockStore)(nil).GetTailnetClientsForAgent), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetClientsForAgent", reflect.TypeOf((*MockStore)(nil).GetTailnetClientsForAgent), ctx, agentID) } // GetTailnetPeers mocks base method. -func (m *MockStore) GetTailnetPeers(arg0 context.Context, arg1 uuid.UUID) ([]database.TailnetPeer, error) { +func (m *MockStore) GetTailnetPeers(ctx context.Context, id uuid.UUID) ([]database.TailnetPeer, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTailnetPeers", arg0, arg1) + ret := m.ctrl.Call(m, "GetTailnetPeers", ctx, id) ret0, _ := ret[0].([]database.TailnetPeer) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTailnetPeers indicates an expected call of GetTailnetPeers. -func (mr *MockStoreMockRecorder) GetTailnetPeers(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTailnetPeers(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetPeers", reflect.TypeOf((*MockStore)(nil).GetTailnetPeers), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetPeers", reflect.TypeOf((*MockStore)(nil).GetTailnetPeers), ctx, id) } // GetTailnetTunnelPeerBindings mocks base method. -func (m *MockStore) GetTailnetTunnelPeerBindings(arg0 context.Context, arg1 uuid.UUID) ([]database.GetTailnetTunnelPeerBindingsRow, error) { +func (m *MockStore) GetTailnetTunnelPeerBindings(ctx context.Context, srcID uuid.UUID) ([]database.GetTailnetTunnelPeerBindingsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTailnetTunnelPeerBindings", arg0, arg1) + ret := m.ctrl.Call(m, "GetTailnetTunnelPeerBindings", ctx, srcID) ret0, _ := ret[0].([]database.GetTailnetTunnelPeerBindingsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTailnetTunnelPeerBindings indicates an expected call of GetTailnetTunnelPeerBindings. -func (mr *MockStoreMockRecorder) GetTailnetTunnelPeerBindings(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTailnetTunnelPeerBindings(ctx, srcID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetTunnelPeerBindings", reflect.TypeOf((*MockStore)(nil).GetTailnetTunnelPeerBindings), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetTunnelPeerBindings", reflect.TypeOf((*MockStore)(nil).GetTailnetTunnelPeerBindings), ctx, srcID) } // GetTailnetTunnelPeerIDs mocks base method. -func (m *MockStore) GetTailnetTunnelPeerIDs(arg0 context.Context, arg1 uuid.UUID) ([]database.GetTailnetTunnelPeerIDsRow, error) { +func (m *MockStore) GetTailnetTunnelPeerIDs(ctx context.Context, srcID uuid.UUID) ([]database.GetTailnetTunnelPeerIDsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTailnetTunnelPeerIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetTailnetTunnelPeerIDs", ctx, srcID) ret0, _ := ret[0].([]database.GetTailnetTunnelPeerIDsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTailnetTunnelPeerIDs indicates an expected call of GetTailnetTunnelPeerIDs. -func (mr *MockStoreMockRecorder) GetTailnetTunnelPeerIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTailnetTunnelPeerIDs(ctx, srcID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetTunnelPeerIDs", reflect.TypeOf((*MockStore)(nil).GetTailnetTunnelPeerIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTailnetTunnelPeerIDs", reflect.TypeOf((*MockStore)(nil).GetTailnetTunnelPeerIDs), ctx, srcID) } // GetTemplateAppInsights mocks base method. -func (m *MockStore) GetTemplateAppInsights(arg0 context.Context, arg1 database.GetTemplateAppInsightsParams) ([]database.GetTemplateAppInsightsRow, error) { +func (m *MockStore) GetTemplateAppInsights(ctx context.Context, arg database.GetTemplateAppInsightsParams) ([]database.GetTemplateAppInsightsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateAppInsights", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateAppInsights", ctx, arg) ret0, _ := ret[0].([]database.GetTemplateAppInsightsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateAppInsights indicates an expected call of GetTemplateAppInsights. -func (mr *MockStoreMockRecorder) GetTemplateAppInsights(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateAppInsights(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateAppInsights", reflect.TypeOf((*MockStore)(nil).GetTemplateAppInsights), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateAppInsights", reflect.TypeOf((*MockStore)(nil).GetTemplateAppInsights), ctx, arg) } // GetTemplateAppInsightsByTemplate mocks base method. -func (m *MockStore) GetTemplateAppInsightsByTemplate(arg0 context.Context, arg1 database.GetTemplateAppInsightsByTemplateParams) ([]database.GetTemplateAppInsightsByTemplateRow, error) { +func (m *MockStore) GetTemplateAppInsightsByTemplate(ctx context.Context, arg database.GetTemplateAppInsightsByTemplateParams) ([]database.GetTemplateAppInsightsByTemplateRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateAppInsightsByTemplate", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateAppInsightsByTemplate", ctx, arg) ret0, _ := ret[0].([]database.GetTemplateAppInsightsByTemplateRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateAppInsightsByTemplate indicates an expected call of GetTemplateAppInsightsByTemplate. -func (mr *MockStoreMockRecorder) GetTemplateAppInsightsByTemplate(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateAppInsightsByTemplate(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateAppInsightsByTemplate", reflect.TypeOf((*MockStore)(nil).GetTemplateAppInsightsByTemplate), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateAppInsightsByTemplate", reflect.TypeOf((*MockStore)(nil).GetTemplateAppInsightsByTemplate), ctx, arg) } // GetTemplateAverageBuildTime mocks base method. -func (m *MockStore) GetTemplateAverageBuildTime(arg0 context.Context, arg1 database.GetTemplateAverageBuildTimeParams) (database.GetTemplateAverageBuildTimeRow, error) { +func (m *MockStore) GetTemplateAverageBuildTime(ctx context.Context, arg database.GetTemplateAverageBuildTimeParams) (database.GetTemplateAverageBuildTimeRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateAverageBuildTime", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateAverageBuildTime", ctx, arg) ret0, _ := ret[0].(database.GetTemplateAverageBuildTimeRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateAverageBuildTime indicates an expected call of GetTemplateAverageBuildTime. -func (mr *MockStoreMockRecorder) GetTemplateAverageBuildTime(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateAverageBuildTime(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateAverageBuildTime", reflect.TypeOf((*MockStore)(nil).GetTemplateAverageBuildTime), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateAverageBuildTime", reflect.TypeOf((*MockStore)(nil).GetTemplateAverageBuildTime), ctx, arg) } // GetTemplateByID mocks base method. -func (m *MockStore) GetTemplateByID(arg0 context.Context, arg1 uuid.UUID) (database.Template, error) { +func (m *MockStore) GetTemplateByID(ctx context.Context, id uuid.UUID) (database.Template, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateByID", ctx, id) ret0, _ := ret[0].(database.Template) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateByID indicates an expected call of GetTemplateByID. -func (mr *MockStoreMockRecorder) GetTemplateByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateByID", reflect.TypeOf((*MockStore)(nil).GetTemplateByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateByID", reflect.TypeOf((*MockStore)(nil).GetTemplateByID), ctx, id) } // GetTemplateByOrganizationAndName mocks base method. -func (m *MockStore) GetTemplateByOrganizationAndName(arg0 context.Context, arg1 database.GetTemplateByOrganizationAndNameParams) (database.Template, error) { +func (m *MockStore) GetTemplateByOrganizationAndName(ctx context.Context, arg database.GetTemplateByOrganizationAndNameParams) (database.Template, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateByOrganizationAndName", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateByOrganizationAndName", ctx, arg) ret0, _ := ret[0].(database.Template) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateByOrganizationAndName indicates an expected call of GetTemplateByOrganizationAndName. -func (mr *MockStoreMockRecorder) GetTemplateByOrganizationAndName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateByOrganizationAndName(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateByOrganizationAndName", reflect.TypeOf((*MockStore)(nil).GetTemplateByOrganizationAndName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateByOrganizationAndName", reflect.TypeOf((*MockStore)(nil).GetTemplateByOrganizationAndName), ctx, arg) } // GetTemplateDAUs mocks base method. -func (m *MockStore) GetTemplateDAUs(arg0 context.Context, arg1 database.GetTemplateDAUsParams) ([]database.GetTemplateDAUsRow, error) { +func (m *MockStore) GetTemplateDAUs(ctx context.Context, arg database.GetTemplateDAUsParams) ([]database.GetTemplateDAUsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateDAUs", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateDAUs", ctx, arg) ret0, _ := ret[0].([]database.GetTemplateDAUsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateDAUs indicates an expected call of GetTemplateDAUs. -func (mr *MockStoreMockRecorder) GetTemplateDAUs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateDAUs(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateDAUs", reflect.TypeOf((*MockStore)(nil).GetTemplateDAUs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateDAUs", reflect.TypeOf((*MockStore)(nil).GetTemplateDAUs), ctx, arg) } // GetTemplateGroupRoles mocks base method. -func (m *MockStore) GetTemplateGroupRoles(arg0 context.Context, arg1 uuid.UUID) ([]database.TemplateGroup, error) { +func (m *MockStore) GetTemplateGroupRoles(ctx context.Context, id uuid.UUID) ([]database.TemplateGroup, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateGroupRoles", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateGroupRoles", ctx, id) ret0, _ := ret[0].([]database.TemplateGroup) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateGroupRoles indicates an expected call of GetTemplateGroupRoles. -func (mr *MockStoreMockRecorder) GetTemplateGroupRoles(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateGroupRoles(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateGroupRoles", reflect.TypeOf((*MockStore)(nil).GetTemplateGroupRoles), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateGroupRoles", reflect.TypeOf((*MockStore)(nil).GetTemplateGroupRoles), ctx, id) } // GetTemplateInsights mocks base method. -func (m *MockStore) GetTemplateInsights(arg0 context.Context, arg1 database.GetTemplateInsightsParams) (database.GetTemplateInsightsRow, error) { +func (m *MockStore) GetTemplateInsights(ctx context.Context, arg database.GetTemplateInsightsParams) (database.GetTemplateInsightsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateInsights", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateInsights", ctx, arg) ret0, _ := ret[0].(database.GetTemplateInsightsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateInsights indicates an expected call of GetTemplateInsights. -func (mr *MockStoreMockRecorder) GetTemplateInsights(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateInsights(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateInsights", reflect.TypeOf((*MockStore)(nil).GetTemplateInsights), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateInsights", reflect.TypeOf((*MockStore)(nil).GetTemplateInsights), ctx, arg) } // GetTemplateInsightsByInterval mocks base method. -func (m *MockStore) GetTemplateInsightsByInterval(arg0 context.Context, arg1 database.GetTemplateInsightsByIntervalParams) ([]database.GetTemplateInsightsByIntervalRow, error) { +func (m *MockStore) GetTemplateInsightsByInterval(ctx context.Context, arg database.GetTemplateInsightsByIntervalParams) ([]database.GetTemplateInsightsByIntervalRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateInsightsByInterval", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateInsightsByInterval", ctx, arg) ret0, _ := ret[0].([]database.GetTemplateInsightsByIntervalRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateInsightsByInterval indicates an expected call of GetTemplateInsightsByInterval. -func (mr *MockStoreMockRecorder) GetTemplateInsightsByInterval(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateInsightsByInterval(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateInsightsByInterval", reflect.TypeOf((*MockStore)(nil).GetTemplateInsightsByInterval), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateInsightsByInterval", reflect.TypeOf((*MockStore)(nil).GetTemplateInsightsByInterval), ctx, arg) } // GetTemplateInsightsByTemplate mocks base method. -func (m *MockStore) GetTemplateInsightsByTemplate(arg0 context.Context, arg1 database.GetTemplateInsightsByTemplateParams) ([]database.GetTemplateInsightsByTemplateRow, error) { +func (m *MockStore) GetTemplateInsightsByTemplate(ctx context.Context, arg database.GetTemplateInsightsByTemplateParams) ([]database.GetTemplateInsightsByTemplateRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateInsightsByTemplate", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateInsightsByTemplate", ctx, arg) ret0, _ := ret[0].([]database.GetTemplateInsightsByTemplateRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateInsightsByTemplate indicates an expected call of GetTemplateInsightsByTemplate. -func (mr *MockStoreMockRecorder) GetTemplateInsightsByTemplate(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateInsightsByTemplate(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateInsightsByTemplate", reflect.TypeOf((*MockStore)(nil).GetTemplateInsightsByTemplate), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateInsightsByTemplate", reflect.TypeOf((*MockStore)(nil).GetTemplateInsightsByTemplate), ctx, arg) } // GetTemplateParameterInsights mocks base method. -func (m *MockStore) GetTemplateParameterInsights(arg0 context.Context, arg1 database.GetTemplateParameterInsightsParams) ([]database.GetTemplateParameterInsightsRow, error) { +func (m *MockStore) GetTemplateParameterInsights(ctx context.Context, arg database.GetTemplateParameterInsightsParams) ([]database.GetTemplateParameterInsightsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateParameterInsights", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateParameterInsights", ctx, arg) ret0, _ := ret[0].([]database.GetTemplateParameterInsightsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateParameterInsights indicates an expected call of GetTemplateParameterInsights. -func (mr *MockStoreMockRecorder) GetTemplateParameterInsights(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateParameterInsights(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateParameterInsights", reflect.TypeOf((*MockStore)(nil).GetTemplateParameterInsights), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateParameterInsights", reflect.TypeOf((*MockStore)(nil).GetTemplateParameterInsights), ctx, arg) } // GetTemplateUsageStats mocks base method. -func (m *MockStore) GetTemplateUsageStats(arg0 context.Context, arg1 database.GetTemplateUsageStatsParams) ([]database.TemplateUsageStat, error) { +func (m *MockStore) GetTemplateUsageStats(ctx context.Context, arg database.GetTemplateUsageStatsParams) ([]database.TemplateUsageStat, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateUsageStats", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateUsageStats", ctx, arg) ret0, _ := ret[0].([]database.TemplateUsageStat) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateUsageStats indicates an expected call of GetTemplateUsageStats. -func (mr *MockStoreMockRecorder) GetTemplateUsageStats(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateUsageStats(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateUsageStats", reflect.TypeOf((*MockStore)(nil).GetTemplateUsageStats), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateUsageStats", reflect.TypeOf((*MockStore)(nil).GetTemplateUsageStats), ctx, arg) } // GetTemplateUserRoles mocks base method. -func (m *MockStore) GetTemplateUserRoles(arg0 context.Context, arg1 uuid.UUID) ([]database.TemplateUser, error) { +func (m *MockStore) GetTemplateUserRoles(ctx context.Context, id uuid.UUID) ([]database.TemplateUser, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateUserRoles", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateUserRoles", ctx, id) ret0, _ := ret[0].([]database.TemplateUser) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateUserRoles indicates an expected call of GetTemplateUserRoles. -func (mr *MockStoreMockRecorder) GetTemplateUserRoles(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateUserRoles(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateUserRoles", reflect.TypeOf((*MockStore)(nil).GetTemplateUserRoles), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateUserRoles", reflect.TypeOf((*MockStore)(nil).GetTemplateUserRoles), ctx, id) } // GetTemplateVersionByID mocks base method. -func (m *MockStore) GetTemplateVersionByID(arg0 context.Context, arg1 uuid.UUID) (database.TemplateVersion, error) { +func (m *MockStore) GetTemplateVersionByID(ctx context.Context, id uuid.UUID) (database.TemplateVersion, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateVersionByID", ctx, id) ret0, _ := ret[0].(database.TemplateVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateVersionByID indicates an expected call of GetTemplateVersionByID. -func (mr *MockStoreMockRecorder) GetTemplateVersionByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateVersionByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionByID", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionByID", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionByID), ctx, id) } // GetTemplateVersionByJobID mocks base method. -func (m *MockStore) GetTemplateVersionByJobID(arg0 context.Context, arg1 uuid.UUID) (database.TemplateVersion, error) { +func (m *MockStore) GetTemplateVersionByJobID(ctx context.Context, jobID uuid.UUID) (database.TemplateVersion, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionByJobID", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateVersionByJobID", ctx, jobID) ret0, _ := ret[0].(database.TemplateVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateVersionByJobID indicates an expected call of GetTemplateVersionByJobID. -func (mr *MockStoreMockRecorder) GetTemplateVersionByJobID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateVersionByJobID(ctx, jobID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionByJobID", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionByJobID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionByJobID", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionByJobID), ctx, jobID) } // GetTemplateVersionByTemplateIDAndName mocks base method. -func (m *MockStore) GetTemplateVersionByTemplateIDAndName(arg0 context.Context, arg1 database.GetTemplateVersionByTemplateIDAndNameParams) (database.TemplateVersion, error) { +func (m *MockStore) GetTemplateVersionByTemplateIDAndName(ctx context.Context, arg database.GetTemplateVersionByTemplateIDAndNameParams) (database.TemplateVersion, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionByTemplateIDAndName", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateVersionByTemplateIDAndName", ctx, arg) ret0, _ := ret[0].(database.TemplateVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateVersionByTemplateIDAndName indicates an expected call of GetTemplateVersionByTemplateIDAndName. -func (mr *MockStoreMockRecorder) GetTemplateVersionByTemplateIDAndName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateVersionByTemplateIDAndName(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionByTemplateIDAndName", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionByTemplateIDAndName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionByTemplateIDAndName", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionByTemplateIDAndName), ctx, arg) } // GetTemplateVersionParameters mocks base method. -func (m *MockStore) GetTemplateVersionParameters(arg0 context.Context, arg1 uuid.UUID) ([]database.TemplateVersionParameter, error) { +func (m *MockStore) GetTemplateVersionParameters(ctx context.Context, templateVersionID uuid.UUID) ([]database.TemplateVersionParameter, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionParameters", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateVersionParameters", ctx, templateVersionID) ret0, _ := ret[0].([]database.TemplateVersionParameter) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateVersionParameters indicates an expected call of GetTemplateVersionParameters. -func (mr *MockStoreMockRecorder) GetTemplateVersionParameters(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateVersionParameters(ctx, templateVersionID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionParameters", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionParameters), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionParameters", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionParameters), ctx, templateVersionID) } // GetTemplateVersionVariables mocks base method. -func (m *MockStore) GetTemplateVersionVariables(arg0 context.Context, arg1 uuid.UUID) ([]database.TemplateVersionVariable, error) { +func (m *MockStore) GetTemplateVersionVariables(ctx context.Context, templateVersionID uuid.UUID) ([]database.TemplateVersionVariable, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionVariables", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateVersionVariables", ctx, templateVersionID) ret0, _ := ret[0].([]database.TemplateVersionVariable) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateVersionVariables indicates an expected call of GetTemplateVersionVariables. -func (mr *MockStoreMockRecorder) GetTemplateVersionVariables(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateVersionVariables(ctx, templateVersionID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionVariables", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionVariables), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionVariables", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionVariables), ctx, templateVersionID) } // GetTemplateVersionWorkspaceTags mocks base method. -func (m *MockStore) GetTemplateVersionWorkspaceTags(arg0 context.Context, arg1 uuid.UUID) ([]database.TemplateVersionWorkspaceTag, error) { +func (m *MockStore) GetTemplateVersionWorkspaceTags(ctx context.Context, templateVersionID uuid.UUID) ([]database.TemplateVersionWorkspaceTag, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionWorkspaceTags", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateVersionWorkspaceTags", ctx, templateVersionID) ret0, _ := ret[0].([]database.TemplateVersionWorkspaceTag) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateVersionWorkspaceTags indicates an expected call of GetTemplateVersionWorkspaceTags. -func (mr *MockStoreMockRecorder) GetTemplateVersionWorkspaceTags(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateVersionWorkspaceTags(ctx, templateVersionID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionWorkspaceTags", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionWorkspaceTags), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionWorkspaceTags", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionWorkspaceTags), ctx, templateVersionID) } // GetTemplateVersionsByIDs mocks base method. -func (m *MockStore) GetTemplateVersionsByIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.TemplateVersion, error) { +func (m *MockStore) GetTemplateVersionsByIDs(ctx context.Context, ids []uuid.UUID) ([]database.TemplateVersion, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionsByIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateVersionsByIDs", ctx, ids) ret0, _ := ret[0].([]database.TemplateVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateVersionsByIDs indicates an expected call of GetTemplateVersionsByIDs. -func (mr *MockStoreMockRecorder) GetTemplateVersionsByIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateVersionsByIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionsByIDs", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionsByIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionsByIDs", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionsByIDs), ctx, ids) } // GetTemplateVersionsByTemplateID mocks base method. -func (m *MockStore) GetTemplateVersionsByTemplateID(arg0 context.Context, arg1 database.GetTemplateVersionsByTemplateIDParams) ([]database.TemplateVersion, error) { +func (m *MockStore) GetTemplateVersionsByTemplateID(ctx context.Context, arg database.GetTemplateVersionsByTemplateIDParams) ([]database.TemplateVersion, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionsByTemplateID", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateVersionsByTemplateID", ctx, arg) ret0, _ := ret[0].([]database.TemplateVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateVersionsByTemplateID indicates an expected call of GetTemplateVersionsByTemplateID. -func (mr *MockStoreMockRecorder) GetTemplateVersionsByTemplateID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateVersionsByTemplateID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionsByTemplateID", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionsByTemplateID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionsByTemplateID", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionsByTemplateID), ctx, arg) } // GetTemplateVersionsCreatedAfter mocks base method. -func (m *MockStore) GetTemplateVersionsCreatedAfter(arg0 context.Context, arg1 time.Time) ([]database.TemplateVersion, error) { +func (m *MockStore) GetTemplateVersionsCreatedAfter(ctx context.Context, createdAt time.Time) ([]database.TemplateVersion, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplateVersionsCreatedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplateVersionsCreatedAfter", ctx, createdAt) ret0, _ := ret[0].([]database.TemplateVersion) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplateVersionsCreatedAfter indicates an expected call of GetTemplateVersionsCreatedAfter. -func (mr *MockStoreMockRecorder) GetTemplateVersionsCreatedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplateVersionsCreatedAfter(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionsCreatedAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplateVersionsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetTemplateVersionsCreatedAfter), ctx, createdAt) } // GetTemplates mocks base method. -func (m *MockStore) GetTemplates(arg0 context.Context) ([]database.Template, error) { +func (m *MockStore) GetTemplates(ctx context.Context) ([]database.Template, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplates", arg0) + ret := m.ctrl.Call(m, "GetTemplates", ctx) ret0, _ := ret[0].([]database.Template) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplates indicates an expected call of GetTemplates. -func (mr *MockStoreMockRecorder) GetTemplates(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplates(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplates", reflect.TypeOf((*MockStore)(nil).GetTemplates), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplates", reflect.TypeOf((*MockStore)(nil).GetTemplates), ctx) } // GetTemplatesWithFilter mocks base method. -func (m *MockStore) GetTemplatesWithFilter(arg0 context.Context, arg1 database.GetTemplatesWithFilterParams) ([]database.Template, error) { +func (m *MockStore) GetTemplatesWithFilter(ctx context.Context, arg database.GetTemplatesWithFilterParams) ([]database.Template, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetTemplatesWithFilter", arg0, arg1) + ret := m.ctrl.Call(m, "GetTemplatesWithFilter", ctx, arg) ret0, _ := ret[0].([]database.Template) ret1, _ := ret[1].(error) return ret0, ret1 } // GetTemplatesWithFilter indicates an expected call of GetTemplatesWithFilter. -func (mr *MockStoreMockRecorder) GetTemplatesWithFilter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetTemplatesWithFilter(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplatesWithFilter", reflect.TypeOf((*MockStore)(nil).GetTemplatesWithFilter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTemplatesWithFilter", reflect.TypeOf((*MockStore)(nil).GetTemplatesWithFilter), ctx, arg) } // GetUnexpiredLicenses mocks base method. -func (m *MockStore) GetUnexpiredLicenses(arg0 context.Context) ([]database.License, error) { +func (m *MockStore) GetUnexpiredLicenses(ctx context.Context) ([]database.License, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUnexpiredLicenses", arg0) + ret := m.ctrl.Call(m, "GetUnexpiredLicenses", ctx) ret0, _ := ret[0].([]database.License) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUnexpiredLicenses indicates an expected call of GetUnexpiredLicenses. -func (mr *MockStoreMockRecorder) GetUnexpiredLicenses(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUnexpiredLicenses(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUnexpiredLicenses", reflect.TypeOf((*MockStore)(nil).GetUnexpiredLicenses), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUnexpiredLicenses", reflect.TypeOf((*MockStore)(nil).GetUnexpiredLicenses), ctx) } // GetUserActivityInsights mocks base method. -func (m *MockStore) GetUserActivityInsights(arg0 context.Context, arg1 database.GetUserActivityInsightsParams) ([]database.GetUserActivityInsightsRow, error) { +func (m *MockStore) GetUserActivityInsights(ctx context.Context, arg database.GetUserActivityInsightsParams) ([]database.GetUserActivityInsightsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserActivityInsights", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserActivityInsights", ctx, arg) ret0, _ := ret[0].([]database.GetUserActivityInsightsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserActivityInsights indicates an expected call of GetUserActivityInsights. -func (mr *MockStoreMockRecorder) GetUserActivityInsights(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserActivityInsights(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserActivityInsights", reflect.TypeOf((*MockStore)(nil).GetUserActivityInsights), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserActivityInsights", reflect.TypeOf((*MockStore)(nil).GetUserActivityInsights), ctx, arg) } // GetUserByEmailOrUsername mocks base method. -func (m *MockStore) GetUserByEmailOrUsername(arg0 context.Context, arg1 database.GetUserByEmailOrUsernameParams) (database.User, error) { +func (m *MockStore) GetUserByEmailOrUsername(ctx context.Context, arg database.GetUserByEmailOrUsernameParams) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserByEmailOrUsername", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserByEmailOrUsername", ctx, arg) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserByEmailOrUsername indicates an expected call of GetUserByEmailOrUsername. -func (mr *MockStoreMockRecorder) GetUserByEmailOrUsername(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserByEmailOrUsername(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserByEmailOrUsername", reflect.TypeOf((*MockStore)(nil).GetUserByEmailOrUsername), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserByEmailOrUsername", reflect.TypeOf((*MockStore)(nil).GetUserByEmailOrUsername), ctx, arg) } // GetUserByID mocks base method. -func (m *MockStore) GetUserByID(arg0 context.Context, arg1 uuid.UUID) (database.User, error) { +func (m *MockStore) GetUserByID(ctx context.Context, id uuid.UUID) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserByID", ctx, id) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserByID indicates an expected call of GetUserByID. -func (mr *MockStoreMockRecorder) GetUserByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserByID", reflect.TypeOf((*MockStore)(nil).GetUserByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserByID", reflect.TypeOf((*MockStore)(nil).GetUserByID), ctx, id) } // GetUserCount mocks base method. -func (m *MockStore) GetUserCount(arg0 context.Context) (int64, error) { +func (m *MockStore) GetUserCount(ctx context.Context) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserCount", arg0) + ret := m.ctrl.Call(m, "GetUserCount", ctx) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserCount indicates an expected call of GetUserCount. -func (mr *MockStoreMockRecorder) GetUserCount(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserCount(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserCount", reflect.TypeOf((*MockStore)(nil).GetUserCount), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserCount", reflect.TypeOf((*MockStore)(nil).GetUserCount), ctx) } // GetUserLatencyInsights mocks base method. -func (m *MockStore) GetUserLatencyInsights(arg0 context.Context, arg1 database.GetUserLatencyInsightsParams) ([]database.GetUserLatencyInsightsRow, error) { +func (m *MockStore) GetUserLatencyInsights(ctx context.Context, arg database.GetUserLatencyInsightsParams) ([]database.GetUserLatencyInsightsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserLatencyInsights", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserLatencyInsights", ctx, arg) ret0, _ := ret[0].([]database.GetUserLatencyInsightsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserLatencyInsights indicates an expected call of GetUserLatencyInsights. -func (mr *MockStoreMockRecorder) GetUserLatencyInsights(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserLatencyInsights(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserLatencyInsights", reflect.TypeOf((*MockStore)(nil).GetUserLatencyInsights), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserLatencyInsights", reflect.TypeOf((*MockStore)(nil).GetUserLatencyInsights), ctx, arg) } // GetUserLinkByLinkedID mocks base method. -func (m *MockStore) GetUserLinkByLinkedID(arg0 context.Context, arg1 string) (database.UserLink, error) { +func (m *MockStore) GetUserLinkByLinkedID(ctx context.Context, linkedID string) (database.UserLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserLinkByLinkedID", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserLinkByLinkedID", ctx, linkedID) ret0, _ := ret[0].(database.UserLink) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserLinkByLinkedID indicates an expected call of GetUserLinkByLinkedID. -func (mr *MockStoreMockRecorder) GetUserLinkByLinkedID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserLinkByLinkedID(ctx, linkedID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserLinkByLinkedID", reflect.TypeOf((*MockStore)(nil).GetUserLinkByLinkedID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserLinkByLinkedID", reflect.TypeOf((*MockStore)(nil).GetUserLinkByLinkedID), ctx, linkedID) } // GetUserLinkByUserIDLoginType mocks base method. -func (m *MockStore) GetUserLinkByUserIDLoginType(arg0 context.Context, arg1 database.GetUserLinkByUserIDLoginTypeParams) (database.UserLink, error) { +func (m *MockStore) GetUserLinkByUserIDLoginType(ctx context.Context, arg database.GetUserLinkByUserIDLoginTypeParams) (database.UserLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserLinkByUserIDLoginType", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserLinkByUserIDLoginType", ctx, arg) ret0, _ := ret[0].(database.UserLink) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserLinkByUserIDLoginType indicates an expected call of GetUserLinkByUserIDLoginType. -func (mr *MockStoreMockRecorder) GetUserLinkByUserIDLoginType(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserLinkByUserIDLoginType(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserLinkByUserIDLoginType", reflect.TypeOf((*MockStore)(nil).GetUserLinkByUserIDLoginType), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserLinkByUserIDLoginType", reflect.TypeOf((*MockStore)(nil).GetUserLinkByUserIDLoginType), ctx, arg) } // GetUserLinksByUserID mocks base method. -func (m *MockStore) GetUserLinksByUserID(arg0 context.Context, arg1 uuid.UUID) ([]database.UserLink, error) { +func (m *MockStore) GetUserLinksByUserID(ctx context.Context, userID uuid.UUID) ([]database.UserLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserLinksByUserID", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserLinksByUserID", ctx, userID) ret0, _ := ret[0].([]database.UserLink) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserLinksByUserID indicates an expected call of GetUserLinksByUserID. -func (mr *MockStoreMockRecorder) GetUserLinksByUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserLinksByUserID(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserLinksByUserID", reflect.TypeOf((*MockStore)(nil).GetUserLinksByUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserLinksByUserID", reflect.TypeOf((*MockStore)(nil).GetUserLinksByUserID), ctx, userID) } // GetUserNotificationPreferences mocks base method. -func (m *MockStore) GetUserNotificationPreferences(arg0 context.Context, arg1 uuid.UUID) ([]database.NotificationPreference, error) { +func (m *MockStore) GetUserNotificationPreferences(ctx context.Context, userID uuid.UUID) ([]database.NotificationPreference, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserNotificationPreferences", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserNotificationPreferences", ctx, userID) ret0, _ := ret[0].([]database.NotificationPreference) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserNotificationPreferences indicates an expected call of GetUserNotificationPreferences. -func (mr *MockStoreMockRecorder) GetUserNotificationPreferences(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserNotificationPreferences(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserNotificationPreferences", reflect.TypeOf((*MockStore)(nil).GetUserNotificationPreferences), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserNotificationPreferences", reflect.TypeOf((*MockStore)(nil).GetUserNotificationPreferences), ctx, userID) } // GetUserWorkspaceBuildParameters mocks base method. -func (m *MockStore) GetUserWorkspaceBuildParameters(arg0 context.Context, arg1 database.GetUserWorkspaceBuildParametersParams) ([]database.GetUserWorkspaceBuildParametersRow, error) { +func (m *MockStore) GetUserWorkspaceBuildParameters(ctx context.Context, arg database.GetUserWorkspaceBuildParametersParams) ([]database.GetUserWorkspaceBuildParametersRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUserWorkspaceBuildParameters", arg0, arg1) + ret := m.ctrl.Call(m, "GetUserWorkspaceBuildParameters", ctx, arg) ret0, _ := ret[0].([]database.GetUserWorkspaceBuildParametersRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUserWorkspaceBuildParameters indicates an expected call of GetUserWorkspaceBuildParameters. -func (mr *MockStoreMockRecorder) GetUserWorkspaceBuildParameters(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUserWorkspaceBuildParameters(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserWorkspaceBuildParameters", reflect.TypeOf((*MockStore)(nil).GetUserWorkspaceBuildParameters), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUserWorkspaceBuildParameters", reflect.TypeOf((*MockStore)(nil).GetUserWorkspaceBuildParameters), ctx, arg) } // GetUsers mocks base method. -func (m *MockStore) GetUsers(arg0 context.Context, arg1 database.GetUsersParams) ([]database.GetUsersRow, error) { +func (m *MockStore) GetUsers(ctx context.Context, arg database.GetUsersParams) ([]database.GetUsersRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUsers", arg0, arg1) + ret := m.ctrl.Call(m, "GetUsers", ctx, arg) ret0, _ := ret[0].([]database.GetUsersRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUsers indicates an expected call of GetUsers. -func (mr *MockStoreMockRecorder) GetUsers(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUsers(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUsers", reflect.TypeOf((*MockStore)(nil).GetUsers), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUsers", reflect.TypeOf((*MockStore)(nil).GetUsers), ctx, arg) } // GetUsersByIDs mocks base method. -func (m *MockStore) GetUsersByIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.User, error) { +func (m *MockStore) GetUsersByIDs(ctx context.Context, ids []uuid.UUID) ([]database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetUsersByIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetUsersByIDs", ctx, ids) ret0, _ := ret[0].([]database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // GetUsersByIDs indicates an expected call of GetUsersByIDs. -func (mr *MockStoreMockRecorder) GetUsersByIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetUsersByIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUsersByIDs", reflect.TypeOf((*MockStore)(nil).GetUsersByIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetUsersByIDs", reflect.TypeOf((*MockStore)(nil).GetUsersByIDs), ctx, ids) } // GetWorkspaceAgentAndLatestBuildByAuthToken mocks base method. -func (m *MockStore) GetWorkspaceAgentAndLatestBuildByAuthToken(arg0 context.Context, arg1 uuid.UUID) (database.GetWorkspaceAgentAndLatestBuildByAuthTokenRow, error) { +func (m *MockStore) GetWorkspaceAgentAndLatestBuildByAuthToken(ctx context.Context, authToken uuid.UUID) (database.GetWorkspaceAgentAndLatestBuildByAuthTokenRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentAndLatestBuildByAuthToken", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentAndLatestBuildByAuthToken", ctx, authToken) ret0, _ := ret[0].(database.GetWorkspaceAgentAndLatestBuildByAuthTokenRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentAndLatestBuildByAuthToken indicates an expected call of GetWorkspaceAgentAndLatestBuildByAuthToken. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentAndLatestBuildByAuthToken(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentAndLatestBuildByAuthToken(ctx, authToken any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentAndLatestBuildByAuthToken", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentAndLatestBuildByAuthToken), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentAndLatestBuildByAuthToken", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentAndLatestBuildByAuthToken), ctx, authToken) } // GetWorkspaceAgentByID mocks base method. -func (m *MockStore) GetWorkspaceAgentByID(arg0 context.Context, arg1 uuid.UUID) (database.WorkspaceAgent, error) { +func (m *MockStore) GetWorkspaceAgentByID(ctx context.Context, id uuid.UUID) (database.WorkspaceAgent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentByID", ctx, id) ret0, _ := ret[0].(database.WorkspaceAgent) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentByID indicates an expected call of GetWorkspaceAgentByID. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentByID), ctx, id) } // GetWorkspaceAgentByInstanceID mocks base method. -func (m *MockStore) GetWorkspaceAgentByInstanceID(arg0 context.Context, arg1 string) (database.WorkspaceAgent, error) { +func (m *MockStore) GetWorkspaceAgentByInstanceID(ctx context.Context, authInstanceID string) (database.WorkspaceAgent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentByInstanceID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentByInstanceID", ctx, authInstanceID) ret0, _ := ret[0].(database.WorkspaceAgent) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentByInstanceID indicates an expected call of GetWorkspaceAgentByInstanceID. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentByInstanceID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentByInstanceID(ctx, authInstanceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentByInstanceID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentByInstanceID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentByInstanceID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentByInstanceID), ctx, authInstanceID) } // GetWorkspaceAgentLifecycleStateByID mocks base method. -func (m *MockStore) GetWorkspaceAgentLifecycleStateByID(arg0 context.Context, arg1 uuid.UUID) (database.GetWorkspaceAgentLifecycleStateByIDRow, error) { +func (m *MockStore) GetWorkspaceAgentLifecycleStateByID(ctx context.Context, id uuid.UUID) (database.GetWorkspaceAgentLifecycleStateByIDRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentLifecycleStateByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentLifecycleStateByID", ctx, id) ret0, _ := ret[0].(database.GetWorkspaceAgentLifecycleStateByIDRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentLifecycleStateByID indicates an expected call of GetWorkspaceAgentLifecycleStateByID. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentLifecycleStateByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentLifecycleStateByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentLifecycleStateByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentLifecycleStateByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentLifecycleStateByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentLifecycleStateByID), ctx, id) } // GetWorkspaceAgentLogSourcesByAgentIDs mocks base method. -func (m *MockStore) GetWorkspaceAgentLogSourcesByAgentIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.WorkspaceAgentLogSource, error) { +func (m *MockStore) GetWorkspaceAgentLogSourcesByAgentIDs(ctx context.Context, ids []uuid.UUID) ([]database.WorkspaceAgentLogSource, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentLogSourcesByAgentIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentLogSourcesByAgentIDs", ctx, ids) ret0, _ := ret[0].([]database.WorkspaceAgentLogSource) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentLogSourcesByAgentIDs indicates an expected call of GetWorkspaceAgentLogSourcesByAgentIDs. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentLogSourcesByAgentIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentLogSourcesByAgentIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentLogSourcesByAgentIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentLogSourcesByAgentIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentLogSourcesByAgentIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentLogSourcesByAgentIDs), ctx, ids) } // GetWorkspaceAgentLogsAfter mocks base method. -func (m *MockStore) GetWorkspaceAgentLogsAfter(arg0 context.Context, arg1 database.GetWorkspaceAgentLogsAfterParams) ([]database.WorkspaceAgentLog, error) { +func (m *MockStore) GetWorkspaceAgentLogsAfter(ctx context.Context, arg database.GetWorkspaceAgentLogsAfterParams) ([]database.WorkspaceAgentLog, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentLogsAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentLogsAfter", ctx, arg) ret0, _ := ret[0].([]database.WorkspaceAgentLog) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentLogsAfter indicates an expected call of GetWorkspaceAgentLogsAfter. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentLogsAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentLogsAfter(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentLogsAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentLogsAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentLogsAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentLogsAfter), ctx, arg) } // GetWorkspaceAgentMetadata mocks base method. -func (m *MockStore) GetWorkspaceAgentMetadata(arg0 context.Context, arg1 database.GetWorkspaceAgentMetadataParams) ([]database.WorkspaceAgentMetadatum, error) { +func (m *MockStore) GetWorkspaceAgentMetadata(ctx context.Context, arg database.GetWorkspaceAgentMetadataParams) ([]database.WorkspaceAgentMetadatum, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentMetadata", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentMetadata", ctx, arg) ret0, _ := ret[0].([]database.WorkspaceAgentMetadatum) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentMetadata indicates an expected call of GetWorkspaceAgentMetadata. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentMetadata(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentMetadata(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentMetadata", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentMetadata), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentMetadata", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentMetadata), ctx, arg) } // GetWorkspaceAgentPortShare mocks base method. -func (m *MockStore) GetWorkspaceAgentPortShare(arg0 context.Context, arg1 database.GetWorkspaceAgentPortShareParams) (database.WorkspaceAgentPortShare, error) { +func (m *MockStore) GetWorkspaceAgentPortShare(ctx context.Context, arg database.GetWorkspaceAgentPortShareParams) (database.WorkspaceAgentPortShare, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentPortShare", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentPortShare", ctx, arg) ret0, _ := ret[0].(database.WorkspaceAgentPortShare) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentPortShare indicates an expected call of GetWorkspaceAgentPortShare. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentPortShare(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentPortShare(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentPortShare", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentPortShare), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentPortShare", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentPortShare), ctx, arg) } // GetWorkspaceAgentScriptTimingsByBuildID mocks base method. -func (m *MockStore) GetWorkspaceAgentScriptTimingsByBuildID(arg0 context.Context, arg1 uuid.UUID) ([]database.GetWorkspaceAgentScriptTimingsByBuildIDRow, error) { +func (m *MockStore) GetWorkspaceAgentScriptTimingsByBuildID(ctx context.Context, id uuid.UUID) ([]database.GetWorkspaceAgentScriptTimingsByBuildIDRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentScriptTimingsByBuildID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentScriptTimingsByBuildID", ctx, id) ret0, _ := ret[0].([]database.GetWorkspaceAgentScriptTimingsByBuildIDRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentScriptTimingsByBuildID indicates an expected call of GetWorkspaceAgentScriptTimingsByBuildID. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentScriptTimingsByBuildID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentScriptTimingsByBuildID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentScriptTimingsByBuildID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentScriptTimingsByBuildID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentScriptTimingsByBuildID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentScriptTimingsByBuildID), ctx, id) } // GetWorkspaceAgentScriptsByAgentIDs mocks base method. -func (m *MockStore) GetWorkspaceAgentScriptsByAgentIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.WorkspaceAgentScript, error) { +func (m *MockStore) GetWorkspaceAgentScriptsByAgentIDs(ctx context.Context, ids []uuid.UUID) ([]database.WorkspaceAgentScript, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentScriptsByAgentIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentScriptsByAgentIDs", ctx, ids) ret0, _ := ret[0].([]database.WorkspaceAgentScript) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentScriptsByAgentIDs indicates an expected call of GetWorkspaceAgentScriptsByAgentIDs. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentScriptsByAgentIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentScriptsByAgentIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentScriptsByAgentIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentScriptsByAgentIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentScriptsByAgentIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentScriptsByAgentIDs), ctx, ids) } // GetWorkspaceAgentStats mocks base method. -func (m *MockStore) GetWorkspaceAgentStats(arg0 context.Context, arg1 time.Time) ([]database.GetWorkspaceAgentStatsRow, error) { +func (m *MockStore) GetWorkspaceAgentStats(ctx context.Context, createdAt time.Time) ([]database.GetWorkspaceAgentStatsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentStats", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentStats", ctx, createdAt) ret0, _ := ret[0].([]database.GetWorkspaceAgentStatsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentStats indicates an expected call of GetWorkspaceAgentStats. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentStats(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentStats(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentStats", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentStats), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentStats", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentStats), ctx, createdAt) } // GetWorkspaceAgentStatsAndLabels mocks base method. -func (m *MockStore) GetWorkspaceAgentStatsAndLabels(arg0 context.Context, arg1 time.Time) ([]database.GetWorkspaceAgentStatsAndLabelsRow, error) { +func (m *MockStore) GetWorkspaceAgentStatsAndLabels(ctx context.Context, createdAt time.Time) ([]database.GetWorkspaceAgentStatsAndLabelsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentStatsAndLabels", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentStatsAndLabels", ctx, createdAt) ret0, _ := ret[0].([]database.GetWorkspaceAgentStatsAndLabelsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentStatsAndLabels indicates an expected call of GetWorkspaceAgentStatsAndLabels. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentStatsAndLabels(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentStatsAndLabels(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentStatsAndLabels", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentStatsAndLabels), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentStatsAndLabels", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentStatsAndLabels), ctx, createdAt) } // GetWorkspaceAgentUsageStats mocks base method. -func (m *MockStore) GetWorkspaceAgentUsageStats(arg0 context.Context, arg1 time.Time) ([]database.GetWorkspaceAgentUsageStatsRow, error) { +func (m *MockStore) GetWorkspaceAgentUsageStats(ctx context.Context, createdAt time.Time) ([]database.GetWorkspaceAgentUsageStatsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentUsageStats", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentUsageStats", ctx, createdAt) ret0, _ := ret[0].([]database.GetWorkspaceAgentUsageStatsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentUsageStats indicates an expected call of GetWorkspaceAgentUsageStats. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentUsageStats(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentUsageStats(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentUsageStats", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentUsageStats), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentUsageStats", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentUsageStats), ctx, createdAt) } // GetWorkspaceAgentUsageStatsAndLabels mocks base method. -func (m *MockStore) GetWorkspaceAgentUsageStatsAndLabels(arg0 context.Context, arg1 time.Time) ([]database.GetWorkspaceAgentUsageStatsAndLabelsRow, error) { +func (m *MockStore) GetWorkspaceAgentUsageStatsAndLabels(ctx context.Context, createdAt time.Time) ([]database.GetWorkspaceAgentUsageStatsAndLabelsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentUsageStatsAndLabels", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentUsageStatsAndLabels", ctx, createdAt) ret0, _ := ret[0].([]database.GetWorkspaceAgentUsageStatsAndLabelsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentUsageStatsAndLabels indicates an expected call of GetWorkspaceAgentUsageStatsAndLabels. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentUsageStatsAndLabels(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentUsageStatsAndLabels(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentUsageStatsAndLabels", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentUsageStatsAndLabels), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentUsageStatsAndLabels", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentUsageStatsAndLabels), ctx, createdAt) } // GetWorkspaceAgentsByResourceIDs mocks base method. -func (m *MockStore) GetWorkspaceAgentsByResourceIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.WorkspaceAgent, error) { +func (m *MockStore) GetWorkspaceAgentsByResourceIDs(ctx context.Context, ids []uuid.UUID) ([]database.WorkspaceAgent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentsByResourceIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentsByResourceIDs", ctx, ids) ret0, _ := ret[0].([]database.WorkspaceAgent) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentsByResourceIDs indicates an expected call of GetWorkspaceAgentsByResourceIDs. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentsByResourceIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentsByResourceIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentsByResourceIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentsByResourceIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentsByResourceIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentsByResourceIDs), ctx, ids) } // GetWorkspaceAgentsCreatedAfter mocks base method. -func (m *MockStore) GetWorkspaceAgentsCreatedAfter(arg0 context.Context, arg1 time.Time) ([]database.WorkspaceAgent, error) { +func (m *MockStore) GetWorkspaceAgentsCreatedAfter(ctx context.Context, createdAt time.Time) ([]database.WorkspaceAgent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentsCreatedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentsCreatedAfter", ctx, createdAt) ret0, _ := ret[0].([]database.WorkspaceAgent) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentsCreatedAfter indicates an expected call of GetWorkspaceAgentsCreatedAfter. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentsCreatedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentsCreatedAfter(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentsCreatedAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentsCreatedAfter), ctx, createdAt) } // GetWorkspaceAgentsInLatestBuildByWorkspaceID mocks base method. -func (m *MockStore) GetWorkspaceAgentsInLatestBuildByWorkspaceID(arg0 context.Context, arg1 uuid.UUID) ([]database.WorkspaceAgent, error) { +func (m *MockStore) GetWorkspaceAgentsInLatestBuildByWorkspaceID(ctx context.Context, workspaceID uuid.UUID) ([]database.WorkspaceAgent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAgentsInLatestBuildByWorkspaceID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAgentsInLatestBuildByWorkspaceID", ctx, workspaceID) ret0, _ := ret[0].([]database.WorkspaceAgent) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAgentsInLatestBuildByWorkspaceID indicates an expected call of GetWorkspaceAgentsInLatestBuildByWorkspaceID. -func (mr *MockStoreMockRecorder) GetWorkspaceAgentsInLatestBuildByWorkspaceID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAgentsInLatestBuildByWorkspaceID(ctx, workspaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentsInLatestBuildByWorkspaceID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentsInLatestBuildByWorkspaceID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAgentsInLatestBuildByWorkspaceID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAgentsInLatestBuildByWorkspaceID), ctx, workspaceID) } // GetWorkspaceAppByAgentIDAndSlug mocks base method. -func (m *MockStore) GetWorkspaceAppByAgentIDAndSlug(arg0 context.Context, arg1 database.GetWorkspaceAppByAgentIDAndSlugParams) (database.WorkspaceApp, error) { +func (m *MockStore) GetWorkspaceAppByAgentIDAndSlug(ctx context.Context, arg database.GetWorkspaceAppByAgentIDAndSlugParams) (database.WorkspaceApp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAppByAgentIDAndSlug", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAppByAgentIDAndSlug", ctx, arg) ret0, _ := ret[0].(database.WorkspaceApp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAppByAgentIDAndSlug indicates an expected call of GetWorkspaceAppByAgentIDAndSlug. -func (mr *MockStoreMockRecorder) GetWorkspaceAppByAgentIDAndSlug(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAppByAgentIDAndSlug(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAppByAgentIDAndSlug", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAppByAgentIDAndSlug), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAppByAgentIDAndSlug", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAppByAgentIDAndSlug), ctx, arg) } // GetWorkspaceAppsByAgentID mocks base method. -func (m *MockStore) GetWorkspaceAppsByAgentID(arg0 context.Context, arg1 uuid.UUID) ([]database.WorkspaceApp, error) { +func (m *MockStore) GetWorkspaceAppsByAgentID(ctx context.Context, agentID uuid.UUID) ([]database.WorkspaceApp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAppsByAgentID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAppsByAgentID", ctx, agentID) ret0, _ := ret[0].([]database.WorkspaceApp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAppsByAgentID indicates an expected call of GetWorkspaceAppsByAgentID. -func (mr *MockStoreMockRecorder) GetWorkspaceAppsByAgentID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAppsByAgentID(ctx, agentID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAppsByAgentID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAppsByAgentID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAppsByAgentID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAppsByAgentID), ctx, agentID) } // GetWorkspaceAppsByAgentIDs mocks base method. -func (m *MockStore) GetWorkspaceAppsByAgentIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.WorkspaceApp, error) { +func (m *MockStore) GetWorkspaceAppsByAgentIDs(ctx context.Context, ids []uuid.UUID) ([]database.WorkspaceApp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAppsByAgentIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAppsByAgentIDs", ctx, ids) ret0, _ := ret[0].([]database.WorkspaceApp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAppsByAgentIDs indicates an expected call of GetWorkspaceAppsByAgentIDs. -func (mr *MockStoreMockRecorder) GetWorkspaceAppsByAgentIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAppsByAgentIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAppsByAgentIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAppsByAgentIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAppsByAgentIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAppsByAgentIDs), ctx, ids) } // GetWorkspaceAppsCreatedAfter mocks base method. -func (m *MockStore) GetWorkspaceAppsCreatedAfter(arg0 context.Context, arg1 time.Time) ([]database.WorkspaceApp, error) { +func (m *MockStore) GetWorkspaceAppsCreatedAfter(ctx context.Context, createdAt time.Time) ([]database.WorkspaceApp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceAppsCreatedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceAppsCreatedAfter", ctx, createdAt) ret0, _ := ret[0].([]database.WorkspaceApp) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceAppsCreatedAfter indicates an expected call of GetWorkspaceAppsCreatedAfter. -func (mr *MockStoreMockRecorder) GetWorkspaceAppsCreatedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceAppsCreatedAfter(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAppsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAppsCreatedAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceAppsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceAppsCreatedAfter), ctx, createdAt) } // GetWorkspaceBuildByID mocks base method. -func (m *MockStore) GetWorkspaceBuildByID(arg0 context.Context, arg1 uuid.UUID) (database.WorkspaceBuild, error) { +func (m *MockStore) GetWorkspaceBuildByID(ctx context.Context, id uuid.UUID) (database.WorkspaceBuild, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceBuildByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceBuildByID", ctx, id) ret0, _ := ret[0].(database.WorkspaceBuild) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceBuildByID indicates an expected call of GetWorkspaceBuildByID. -func (mr *MockStoreMockRecorder) GetWorkspaceBuildByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceBuildByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildByID), ctx, id) } // GetWorkspaceBuildByJobID mocks base method. -func (m *MockStore) GetWorkspaceBuildByJobID(arg0 context.Context, arg1 uuid.UUID) (database.WorkspaceBuild, error) { +func (m *MockStore) GetWorkspaceBuildByJobID(ctx context.Context, jobID uuid.UUID) (database.WorkspaceBuild, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceBuildByJobID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceBuildByJobID", ctx, jobID) ret0, _ := ret[0].(database.WorkspaceBuild) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceBuildByJobID indicates an expected call of GetWorkspaceBuildByJobID. -func (mr *MockStoreMockRecorder) GetWorkspaceBuildByJobID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceBuildByJobID(ctx, jobID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildByJobID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildByJobID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildByJobID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildByJobID), ctx, jobID) } // GetWorkspaceBuildByWorkspaceIDAndBuildNumber mocks base method. -func (m *MockStore) GetWorkspaceBuildByWorkspaceIDAndBuildNumber(arg0 context.Context, arg1 database.GetWorkspaceBuildByWorkspaceIDAndBuildNumberParams) (database.WorkspaceBuild, error) { +func (m *MockStore) GetWorkspaceBuildByWorkspaceIDAndBuildNumber(ctx context.Context, arg database.GetWorkspaceBuildByWorkspaceIDAndBuildNumberParams) (database.WorkspaceBuild, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceBuildByWorkspaceIDAndBuildNumber", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceBuildByWorkspaceIDAndBuildNumber", ctx, arg) ret0, _ := ret[0].(database.WorkspaceBuild) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceBuildByWorkspaceIDAndBuildNumber indicates an expected call of GetWorkspaceBuildByWorkspaceIDAndBuildNumber. -func (mr *MockStoreMockRecorder) GetWorkspaceBuildByWorkspaceIDAndBuildNumber(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceBuildByWorkspaceIDAndBuildNumber(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildByWorkspaceIDAndBuildNumber", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildByWorkspaceIDAndBuildNumber), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildByWorkspaceIDAndBuildNumber", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildByWorkspaceIDAndBuildNumber), ctx, arg) } // GetWorkspaceBuildParameters mocks base method. -func (m *MockStore) GetWorkspaceBuildParameters(arg0 context.Context, arg1 uuid.UUID) ([]database.WorkspaceBuildParameter, error) { +func (m *MockStore) GetWorkspaceBuildParameters(ctx context.Context, workspaceBuildID uuid.UUID) ([]database.WorkspaceBuildParameter, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceBuildParameters", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceBuildParameters", ctx, workspaceBuildID) ret0, _ := ret[0].([]database.WorkspaceBuildParameter) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceBuildParameters indicates an expected call of GetWorkspaceBuildParameters. -func (mr *MockStoreMockRecorder) GetWorkspaceBuildParameters(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceBuildParameters(ctx, workspaceBuildID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildParameters", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildParameters), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildParameters", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildParameters), ctx, workspaceBuildID) } // GetWorkspaceBuildStatsByTemplates mocks base method. -func (m *MockStore) GetWorkspaceBuildStatsByTemplates(arg0 context.Context, arg1 time.Time) ([]database.GetWorkspaceBuildStatsByTemplatesRow, error) { +func (m *MockStore) GetWorkspaceBuildStatsByTemplates(ctx context.Context, since time.Time) ([]database.GetWorkspaceBuildStatsByTemplatesRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceBuildStatsByTemplates", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceBuildStatsByTemplates", ctx, since) ret0, _ := ret[0].([]database.GetWorkspaceBuildStatsByTemplatesRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceBuildStatsByTemplates indicates an expected call of GetWorkspaceBuildStatsByTemplates. -func (mr *MockStoreMockRecorder) GetWorkspaceBuildStatsByTemplates(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceBuildStatsByTemplates(ctx, since any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildStatsByTemplates", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildStatsByTemplates), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildStatsByTemplates", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildStatsByTemplates), ctx, since) } // GetWorkspaceBuildsByWorkspaceID mocks base method. -func (m *MockStore) GetWorkspaceBuildsByWorkspaceID(arg0 context.Context, arg1 database.GetWorkspaceBuildsByWorkspaceIDParams) ([]database.WorkspaceBuild, error) { +func (m *MockStore) GetWorkspaceBuildsByWorkspaceID(ctx context.Context, arg database.GetWorkspaceBuildsByWorkspaceIDParams) ([]database.WorkspaceBuild, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceBuildsByWorkspaceID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceBuildsByWorkspaceID", ctx, arg) ret0, _ := ret[0].([]database.WorkspaceBuild) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceBuildsByWorkspaceID indicates an expected call of GetWorkspaceBuildsByWorkspaceID. -func (mr *MockStoreMockRecorder) GetWorkspaceBuildsByWorkspaceID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceBuildsByWorkspaceID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildsByWorkspaceID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildsByWorkspaceID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildsByWorkspaceID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildsByWorkspaceID), ctx, arg) } // GetWorkspaceBuildsCreatedAfter mocks base method. -func (m *MockStore) GetWorkspaceBuildsCreatedAfter(arg0 context.Context, arg1 time.Time) ([]database.WorkspaceBuild, error) { +func (m *MockStore) GetWorkspaceBuildsCreatedAfter(ctx context.Context, createdAt time.Time) ([]database.WorkspaceBuild, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceBuildsCreatedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceBuildsCreatedAfter", ctx, createdAt) ret0, _ := ret[0].([]database.WorkspaceBuild) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceBuildsCreatedAfter indicates an expected call of GetWorkspaceBuildsCreatedAfter. -func (mr *MockStoreMockRecorder) GetWorkspaceBuildsCreatedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceBuildsCreatedAfter(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildsCreatedAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceBuildsCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceBuildsCreatedAfter), ctx, createdAt) } // GetWorkspaceByAgentID mocks base method. -func (m *MockStore) GetWorkspaceByAgentID(arg0 context.Context, arg1 uuid.UUID) (database.Workspace, error) { +func (m *MockStore) GetWorkspaceByAgentID(ctx context.Context, agentID uuid.UUID) (database.Workspace, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceByAgentID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceByAgentID", ctx, agentID) ret0, _ := ret[0].(database.Workspace) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceByAgentID indicates an expected call of GetWorkspaceByAgentID. -func (mr *MockStoreMockRecorder) GetWorkspaceByAgentID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceByAgentID(ctx, agentID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceByAgentID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceByAgentID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceByAgentID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceByAgentID), ctx, agentID) } // GetWorkspaceByID mocks base method. -func (m *MockStore) GetWorkspaceByID(arg0 context.Context, arg1 uuid.UUID) (database.Workspace, error) { +func (m *MockStore) GetWorkspaceByID(ctx context.Context, id uuid.UUID) (database.Workspace, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceByID", ctx, id) ret0, _ := ret[0].(database.Workspace) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceByID indicates an expected call of GetWorkspaceByID. -func (mr *MockStoreMockRecorder) GetWorkspaceByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceByID), ctx, id) } // GetWorkspaceByOwnerIDAndName mocks base method. -func (m *MockStore) GetWorkspaceByOwnerIDAndName(arg0 context.Context, arg1 database.GetWorkspaceByOwnerIDAndNameParams) (database.Workspace, error) { +func (m *MockStore) GetWorkspaceByOwnerIDAndName(ctx context.Context, arg database.GetWorkspaceByOwnerIDAndNameParams) (database.Workspace, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceByOwnerIDAndName", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceByOwnerIDAndName", ctx, arg) ret0, _ := ret[0].(database.Workspace) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceByOwnerIDAndName indicates an expected call of GetWorkspaceByOwnerIDAndName. -func (mr *MockStoreMockRecorder) GetWorkspaceByOwnerIDAndName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceByOwnerIDAndName(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceByOwnerIDAndName", reflect.TypeOf((*MockStore)(nil).GetWorkspaceByOwnerIDAndName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceByOwnerIDAndName", reflect.TypeOf((*MockStore)(nil).GetWorkspaceByOwnerIDAndName), ctx, arg) } // GetWorkspaceByWorkspaceAppID mocks base method. -func (m *MockStore) GetWorkspaceByWorkspaceAppID(arg0 context.Context, arg1 uuid.UUID) (database.Workspace, error) { +func (m *MockStore) GetWorkspaceByWorkspaceAppID(ctx context.Context, workspaceAppID uuid.UUID) (database.Workspace, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceByWorkspaceAppID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceByWorkspaceAppID", ctx, workspaceAppID) ret0, _ := ret[0].(database.Workspace) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceByWorkspaceAppID indicates an expected call of GetWorkspaceByWorkspaceAppID. -func (mr *MockStoreMockRecorder) GetWorkspaceByWorkspaceAppID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceByWorkspaceAppID(ctx, workspaceAppID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceByWorkspaceAppID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceByWorkspaceAppID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceByWorkspaceAppID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceByWorkspaceAppID), ctx, workspaceAppID) } // GetWorkspaceModulesByJobID mocks base method. -func (m *MockStore) GetWorkspaceModulesByJobID(arg0 context.Context, arg1 uuid.UUID) ([]database.WorkspaceModule, error) { +func (m *MockStore) GetWorkspaceModulesByJobID(ctx context.Context, jobID uuid.UUID) ([]database.WorkspaceModule, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceModulesByJobID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceModulesByJobID", ctx, jobID) ret0, _ := ret[0].([]database.WorkspaceModule) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceModulesByJobID indicates an expected call of GetWorkspaceModulesByJobID. -func (mr *MockStoreMockRecorder) GetWorkspaceModulesByJobID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceModulesByJobID(ctx, jobID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceModulesByJobID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceModulesByJobID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceModulesByJobID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceModulesByJobID), ctx, jobID) } // GetWorkspaceModulesCreatedAfter mocks base method. -func (m *MockStore) GetWorkspaceModulesCreatedAfter(arg0 context.Context, arg1 time.Time) ([]database.WorkspaceModule, error) { +func (m *MockStore) GetWorkspaceModulesCreatedAfter(ctx context.Context, createdAt time.Time) ([]database.WorkspaceModule, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceModulesCreatedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceModulesCreatedAfter", ctx, createdAt) ret0, _ := ret[0].([]database.WorkspaceModule) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceModulesCreatedAfter indicates an expected call of GetWorkspaceModulesCreatedAfter. -func (mr *MockStoreMockRecorder) GetWorkspaceModulesCreatedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceModulesCreatedAfter(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceModulesCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceModulesCreatedAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceModulesCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceModulesCreatedAfter), ctx, createdAt) } // GetWorkspaceProxies mocks base method. -func (m *MockStore) GetWorkspaceProxies(arg0 context.Context) ([]database.WorkspaceProxy, error) { +func (m *MockStore) GetWorkspaceProxies(ctx context.Context) ([]database.WorkspaceProxy, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceProxies", arg0) + ret := m.ctrl.Call(m, "GetWorkspaceProxies", ctx) ret0, _ := ret[0].([]database.WorkspaceProxy) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceProxies indicates an expected call of GetWorkspaceProxies. -func (mr *MockStoreMockRecorder) GetWorkspaceProxies(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceProxies(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceProxies", reflect.TypeOf((*MockStore)(nil).GetWorkspaceProxies), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceProxies", reflect.TypeOf((*MockStore)(nil).GetWorkspaceProxies), ctx) } // GetWorkspaceProxyByHostname mocks base method. -func (m *MockStore) GetWorkspaceProxyByHostname(arg0 context.Context, arg1 database.GetWorkspaceProxyByHostnameParams) (database.WorkspaceProxy, error) { +func (m *MockStore) GetWorkspaceProxyByHostname(ctx context.Context, arg database.GetWorkspaceProxyByHostnameParams) (database.WorkspaceProxy, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceProxyByHostname", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceProxyByHostname", ctx, arg) ret0, _ := ret[0].(database.WorkspaceProxy) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceProxyByHostname indicates an expected call of GetWorkspaceProxyByHostname. -func (mr *MockStoreMockRecorder) GetWorkspaceProxyByHostname(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceProxyByHostname(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceProxyByHostname", reflect.TypeOf((*MockStore)(nil).GetWorkspaceProxyByHostname), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceProxyByHostname", reflect.TypeOf((*MockStore)(nil).GetWorkspaceProxyByHostname), ctx, arg) } // GetWorkspaceProxyByID mocks base method. -func (m *MockStore) GetWorkspaceProxyByID(arg0 context.Context, arg1 uuid.UUID) (database.WorkspaceProxy, error) { +func (m *MockStore) GetWorkspaceProxyByID(ctx context.Context, id uuid.UUID) (database.WorkspaceProxy, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceProxyByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceProxyByID", ctx, id) ret0, _ := ret[0].(database.WorkspaceProxy) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceProxyByID indicates an expected call of GetWorkspaceProxyByID. -func (mr *MockStoreMockRecorder) GetWorkspaceProxyByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceProxyByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceProxyByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceProxyByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceProxyByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceProxyByID), ctx, id) } // GetWorkspaceProxyByName mocks base method. -func (m *MockStore) GetWorkspaceProxyByName(arg0 context.Context, arg1 string) (database.WorkspaceProxy, error) { +func (m *MockStore) GetWorkspaceProxyByName(ctx context.Context, name string) (database.WorkspaceProxy, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceProxyByName", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceProxyByName", ctx, name) ret0, _ := ret[0].(database.WorkspaceProxy) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceProxyByName indicates an expected call of GetWorkspaceProxyByName. -func (mr *MockStoreMockRecorder) GetWorkspaceProxyByName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceProxyByName(ctx, name any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceProxyByName", reflect.TypeOf((*MockStore)(nil).GetWorkspaceProxyByName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceProxyByName", reflect.TypeOf((*MockStore)(nil).GetWorkspaceProxyByName), ctx, name) } // GetWorkspaceResourceByID mocks base method. -func (m *MockStore) GetWorkspaceResourceByID(arg0 context.Context, arg1 uuid.UUID) (database.WorkspaceResource, error) { +func (m *MockStore) GetWorkspaceResourceByID(ctx context.Context, id uuid.UUID) (database.WorkspaceResource, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceResourceByID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceResourceByID", ctx, id) ret0, _ := ret[0].(database.WorkspaceResource) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceResourceByID indicates an expected call of GetWorkspaceResourceByID. -func (mr *MockStoreMockRecorder) GetWorkspaceResourceByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceResourceByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourceByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourceByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourceByID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourceByID), ctx, id) } // GetWorkspaceResourceMetadataByResourceIDs mocks base method. -func (m *MockStore) GetWorkspaceResourceMetadataByResourceIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.WorkspaceResourceMetadatum, error) { +func (m *MockStore) GetWorkspaceResourceMetadataByResourceIDs(ctx context.Context, ids []uuid.UUID) ([]database.WorkspaceResourceMetadatum, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceResourceMetadataByResourceIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceResourceMetadataByResourceIDs", ctx, ids) ret0, _ := ret[0].([]database.WorkspaceResourceMetadatum) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceResourceMetadataByResourceIDs indicates an expected call of GetWorkspaceResourceMetadataByResourceIDs. -func (mr *MockStoreMockRecorder) GetWorkspaceResourceMetadataByResourceIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceResourceMetadataByResourceIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourceMetadataByResourceIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourceMetadataByResourceIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourceMetadataByResourceIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourceMetadataByResourceIDs), ctx, ids) } // GetWorkspaceResourceMetadataCreatedAfter mocks base method. -func (m *MockStore) GetWorkspaceResourceMetadataCreatedAfter(arg0 context.Context, arg1 time.Time) ([]database.WorkspaceResourceMetadatum, error) { +func (m *MockStore) GetWorkspaceResourceMetadataCreatedAfter(ctx context.Context, createdAt time.Time) ([]database.WorkspaceResourceMetadatum, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceResourceMetadataCreatedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceResourceMetadataCreatedAfter", ctx, createdAt) ret0, _ := ret[0].([]database.WorkspaceResourceMetadatum) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceResourceMetadataCreatedAfter indicates an expected call of GetWorkspaceResourceMetadataCreatedAfter. -func (mr *MockStoreMockRecorder) GetWorkspaceResourceMetadataCreatedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceResourceMetadataCreatedAfter(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourceMetadataCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourceMetadataCreatedAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourceMetadataCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourceMetadataCreatedAfter), ctx, createdAt) } // GetWorkspaceResourcesByJobID mocks base method. -func (m *MockStore) GetWorkspaceResourcesByJobID(arg0 context.Context, arg1 uuid.UUID) ([]database.WorkspaceResource, error) { +func (m *MockStore) GetWorkspaceResourcesByJobID(ctx context.Context, jobID uuid.UUID) ([]database.WorkspaceResource, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceResourcesByJobID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceResourcesByJobID", ctx, jobID) ret0, _ := ret[0].([]database.WorkspaceResource) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceResourcesByJobID indicates an expected call of GetWorkspaceResourcesByJobID. -func (mr *MockStoreMockRecorder) GetWorkspaceResourcesByJobID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceResourcesByJobID(ctx, jobID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourcesByJobID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourcesByJobID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourcesByJobID", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourcesByJobID), ctx, jobID) } // GetWorkspaceResourcesByJobIDs mocks base method. -func (m *MockStore) GetWorkspaceResourcesByJobIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.WorkspaceResource, error) { +func (m *MockStore) GetWorkspaceResourcesByJobIDs(ctx context.Context, ids []uuid.UUID) ([]database.WorkspaceResource, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceResourcesByJobIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceResourcesByJobIDs", ctx, ids) ret0, _ := ret[0].([]database.WorkspaceResource) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceResourcesByJobIDs indicates an expected call of GetWorkspaceResourcesByJobIDs. -func (mr *MockStoreMockRecorder) GetWorkspaceResourcesByJobIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceResourcesByJobIDs(ctx, ids any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourcesByJobIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourcesByJobIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourcesByJobIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourcesByJobIDs), ctx, ids) } // GetWorkspaceResourcesCreatedAfter mocks base method. -func (m *MockStore) GetWorkspaceResourcesCreatedAfter(arg0 context.Context, arg1 time.Time) ([]database.WorkspaceResource, error) { +func (m *MockStore) GetWorkspaceResourcesCreatedAfter(ctx context.Context, createdAt time.Time) ([]database.WorkspaceResource, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceResourcesCreatedAfter", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceResourcesCreatedAfter", ctx, createdAt) ret0, _ := ret[0].([]database.WorkspaceResource) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceResourcesCreatedAfter indicates an expected call of GetWorkspaceResourcesCreatedAfter. -func (mr *MockStoreMockRecorder) GetWorkspaceResourcesCreatedAfter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceResourcesCreatedAfter(ctx, createdAt any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourcesCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourcesCreatedAfter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceResourcesCreatedAfter", reflect.TypeOf((*MockStore)(nil).GetWorkspaceResourcesCreatedAfter), ctx, createdAt) } // GetWorkspaceUniqueOwnerCountByTemplateIDs mocks base method. -func (m *MockStore) GetWorkspaceUniqueOwnerCountByTemplateIDs(arg0 context.Context, arg1 []uuid.UUID) ([]database.GetWorkspaceUniqueOwnerCountByTemplateIDsRow, error) { +func (m *MockStore) GetWorkspaceUniqueOwnerCountByTemplateIDs(ctx context.Context, templateIds []uuid.UUID) ([]database.GetWorkspaceUniqueOwnerCountByTemplateIDsRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaceUniqueOwnerCountByTemplateIDs", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaceUniqueOwnerCountByTemplateIDs", ctx, templateIds) ret0, _ := ret[0].([]database.GetWorkspaceUniqueOwnerCountByTemplateIDsRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaceUniqueOwnerCountByTemplateIDs indicates an expected call of GetWorkspaceUniqueOwnerCountByTemplateIDs. -func (mr *MockStoreMockRecorder) GetWorkspaceUniqueOwnerCountByTemplateIDs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaceUniqueOwnerCountByTemplateIDs(ctx, templateIds any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceUniqueOwnerCountByTemplateIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceUniqueOwnerCountByTemplateIDs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaceUniqueOwnerCountByTemplateIDs", reflect.TypeOf((*MockStore)(nil).GetWorkspaceUniqueOwnerCountByTemplateIDs), ctx, templateIds) } // GetWorkspaces mocks base method. -func (m *MockStore) GetWorkspaces(arg0 context.Context, arg1 database.GetWorkspacesParams) ([]database.GetWorkspacesRow, error) { +func (m *MockStore) GetWorkspaces(ctx context.Context, arg database.GetWorkspacesParams) ([]database.GetWorkspacesRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspaces", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspaces", ctx, arg) ret0, _ := ret[0].([]database.GetWorkspacesRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspaces indicates an expected call of GetWorkspaces. -func (mr *MockStoreMockRecorder) GetWorkspaces(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspaces(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaces", reflect.TypeOf((*MockStore)(nil).GetWorkspaces), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspaces", reflect.TypeOf((*MockStore)(nil).GetWorkspaces), ctx, arg) } // GetWorkspacesAndAgentsByOwnerID mocks base method. -func (m *MockStore) GetWorkspacesAndAgentsByOwnerID(arg0 context.Context, arg1 uuid.UUID) ([]database.GetWorkspacesAndAgentsByOwnerIDRow, error) { +func (m *MockStore) GetWorkspacesAndAgentsByOwnerID(ctx context.Context, ownerID uuid.UUID) ([]database.GetWorkspacesAndAgentsByOwnerIDRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspacesAndAgentsByOwnerID", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspacesAndAgentsByOwnerID", ctx, ownerID) ret0, _ := ret[0].([]database.GetWorkspacesAndAgentsByOwnerIDRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspacesAndAgentsByOwnerID indicates an expected call of GetWorkspacesAndAgentsByOwnerID. -func (mr *MockStoreMockRecorder) GetWorkspacesAndAgentsByOwnerID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspacesAndAgentsByOwnerID(ctx, ownerID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspacesAndAgentsByOwnerID", reflect.TypeOf((*MockStore)(nil).GetWorkspacesAndAgentsByOwnerID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspacesAndAgentsByOwnerID", reflect.TypeOf((*MockStore)(nil).GetWorkspacesAndAgentsByOwnerID), ctx, ownerID) } // GetWorkspacesEligibleForTransition mocks base method. -func (m *MockStore) GetWorkspacesEligibleForTransition(arg0 context.Context, arg1 time.Time) ([]database.GetWorkspacesEligibleForTransitionRow, error) { +func (m *MockStore) GetWorkspacesEligibleForTransition(ctx context.Context, now time.Time) ([]database.GetWorkspacesEligibleForTransitionRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "GetWorkspacesEligibleForTransition", arg0, arg1) + ret := m.ctrl.Call(m, "GetWorkspacesEligibleForTransition", ctx, now) ret0, _ := ret[0].([]database.GetWorkspacesEligibleForTransitionRow) ret1, _ := ret[1].(error) return ret0, ret1 } // GetWorkspacesEligibleForTransition indicates an expected call of GetWorkspacesEligibleForTransition. -func (mr *MockStoreMockRecorder) GetWorkspacesEligibleForTransition(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) GetWorkspacesEligibleForTransition(ctx, now any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspacesEligibleForTransition", reflect.TypeOf((*MockStore)(nil).GetWorkspacesEligibleForTransition), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspacesEligibleForTransition", reflect.TypeOf((*MockStore)(nil).GetWorkspacesEligibleForTransition), ctx, now) } // InTx mocks base method. @@ -3577,2200 +3563,2200 @@ func (mr *MockStoreMockRecorder) InTx(arg0, arg1 any) *gomock.Call { } // InsertAPIKey mocks base method. -func (m *MockStore) InsertAPIKey(arg0 context.Context, arg1 database.InsertAPIKeyParams) (database.APIKey, error) { +func (m *MockStore) InsertAPIKey(ctx context.Context, arg database.InsertAPIKeyParams) (database.APIKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertAPIKey", arg0, arg1) + ret := m.ctrl.Call(m, "InsertAPIKey", ctx, arg) ret0, _ := ret[0].(database.APIKey) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertAPIKey indicates an expected call of InsertAPIKey. -func (mr *MockStoreMockRecorder) InsertAPIKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertAPIKey(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertAPIKey", reflect.TypeOf((*MockStore)(nil).InsertAPIKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertAPIKey", reflect.TypeOf((*MockStore)(nil).InsertAPIKey), ctx, arg) } // InsertAllUsersGroup mocks base method. -func (m *MockStore) InsertAllUsersGroup(arg0 context.Context, arg1 uuid.UUID) (database.Group, error) { +func (m *MockStore) InsertAllUsersGroup(ctx context.Context, organizationID uuid.UUID) (database.Group, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertAllUsersGroup", arg0, arg1) + ret := m.ctrl.Call(m, "InsertAllUsersGroup", ctx, organizationID) ret0, _ := ret[0].(database.Group) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertAllUsersGroup indicates an expected call of InsertAllUsersGroup. -func (mr *MockStoreMockRecorder) InsertAllUsersGroup(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertAllUsersGroup(ctx, organizationID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertAllUsersGroup", reflect.TypeOf((*MockStore)(nil).InsertAllUsersGroup), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertAllUsersGroup", reflect.TypeOf((*MockStore)(nil).InsertAllUsersGroup), ctx, organizationID) } // InsertAuditLog mocks base method. -func (m *MockStore) InsertAuditLog(arg0 context.Context, arg1 database.InsertAuditLogParams) (database.AuditLog, error) { +func (m *MockStore) InsertAuditLog(ctx context.Context, arg database.InsertAuditLogParams) (database.AuditLog, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertAuditLog", arg0, arg1) + ret := m.ctrl.Call(m, "InsertAuditLog", ctx, arg) ret0, _ := ret[0].(database.AuditLog) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertAuditLog indicates an expected call of InsertAuditLog. -func (mr *MockStoreMockRecorder) InsertAuditLog(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertAuditLog(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertAuditLog", reflect.TypeOf((*MockStore)(nil).InsertAuditLog), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertAuditLog", reflect.TypeOf((*MockStore)(nil).InsertAuditLog), ctx, arg) } // InsertCryptoKey mocks base method. -func (m *MockStore) InsertCryptoKey(arg0 context.Context, arg1 database.InsertCryptoKeyParams) (database.CryptoKey, error) { +func (m *MockStore) InsertCryptoKey(ctx context.Context, arg database.InsertCryptoKeyParams) (database.CryptoKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertCryptoKey", arg0, arg1) + ret := m.ctrl.Call(m, "InsertCryptoKey", ctx, arg) ret0, _ := ret[0].(database.CryptoKey) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertCryptoKey indicates an expected call of InsertCryptoKey. -func (mr *MockStoreMockRecorder) InsertCryptoKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertCryptoKey(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertCryptoKey", reflect.TypeOf((*MockStore)(nil).InsertCryptoKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertCryptoKey", reflect.TypeOf((*MockStore)(nil).InsertCryptoKey), ctx, arg) } // InsertCustomRole mocks base method. -func (m *MockStore) InsertCustomRole(arg0 context.Context, arg1 database.InsertCustomRoleParams) (database.CustomRole, error) { +func (m *MockStore) InsertCustomRole(ctx context.Context, arg database.InsertCustomRoleParams) (database.CustomRole, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertCustomRole", arg0, arg1) + ret := m.ctrl.Call(m, "InsertCustomRole", ctx, arg) ret0, _ := ret[0].(database.CustomRole) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertCustomRole indicates an expected call of InsertCustomRole. -func (mr *MockStoreMockRecorder) InsertCustomRole(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertCustomRole(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertCustomRole", reflect.TypeOf((*MockStore)(nil).InsertCustomRole), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertCustomRole", reflect.TypeOf((*MockStore)(nil).InsertCustomRole), ctx, arg) } // InsertDBCryptKey mocks base method. -func (m *MockStore) InsertDBCryptKey(arg0 context.Context, arg1 database.InsertDBCryptKeyParams) error { +func (m *MockStore) InsertDBCryptKey(ctx context.Context, arg database.InsertDBCryptKeyParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertDBCryptKey", arg0, arg1) + ret := m.ctrl.Call(m, "InsertDBCryptKey", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertDBCryptKey indicates an expected call of InsertDBCryptKey. -func (mr *MockStoreMockRecorder) InsertDBCryptKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertDBCryptKey(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertDBCryptKey", reflect.TypeOf((*MockStore)(nil).InsertDBCryptKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertDBCryptKey", reflect.TypeOf((*MockStore)(nil).InsertDBCryptKey), ctx, arg) } // InsertDERPMeshKey mocks base method. -func (m *MockStore) InsertDERPMeshKey(arg0 context.Context, arg1 string) error { +func (m *MockStore) InsertDERPMeshKey(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertDERPMeshKey", arg0, arg1) + ret := m.ctrl.Call(m, "InsertDERPMeshKey", ctx, value) ret0, _ := ret[0].(error) return ret0 } // InsertDERPMeshKey indicates an expected call of InsertDERPMeshKey. -func (mr *MockStoreMockRecorder) InsertDERPMeshKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertDERPMeshKey(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertDERPMeshKey", reflect.TypeOf((*MockStore)(nil).InsertDERPMeshKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertDERPMeshKey", reflect.TypeOf((*MockStore)(nil).InsertDERPMeshKey), ctx, value) } // InsertDeploymentID mocks base method. -func (m *MockStore) InsertDeploymentID(arg0 context.Context, arg1 string) error { +func (m *MockStore) InsertDeploymentID(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertDeploymentID", arg0, arg1) + ret := m.ctrl.Call(m, "InsertDeploymentID", ctx, value) ret0, _ := ret[0].(error) return ret0 } // InsertDeploymentID indicates an expected call of InsertDeploymentID. -func (mr *MockStoreMockRecorder) InsertDeploymentID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertDeploymentID(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertDeploymentID", reflect.TypeOf((*MockStore)(nil).InsertDeploymentID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertDeploymentID", reflect.TypeOf((*MockStore)(nil).InsertDeploymentID), ctx, value) } // InsertExternalAuthLink mocks base method. -func (m *MockStore) InsertExternalAuthLink(arg0 context.Context, arg1 database.InsertExternalAuthLinkParams) (database.ExternalAuthLink, error) { +func (m *MockStore) InsertExternalAuthLink(ctx context.Context, arg database.InsertExternalAuthLinkParams) (database.ExternalAuthLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertExternalAuthLink", arg0, arg1) + ret := m.ctrl.Call(m, "InsertExternalAuthLink", ctx, arg) ret0, _ := ret[0].(database.ExternalAuthLink) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertExternalAuthLink indicates an expected call of InsertExternalAuthLink. -func (mr *MockStoreMockRecorder) InsertExternalAuthLink(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertExternalAuthLink(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertExternalAuthLink", reflect.TypeOf((*MockStore)(nil).InsertExternalAuthLink), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertExternalAuthLink", reflect.TypeOf((*MockStore)(nil).InsertExternalAuthLink), ctx, arg) } // InsertFile mocks base method. -func (m *MockStore) InsertFile(arg0 context.Context, arg1 database.InsertFileParams) (database.File, error) { +func (m *MockStore) InsertFile(ctx context.Context, arg database.InsertFileParams) (database.File, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertFile", arg0, arg1) + ret := m.ctrl.Call(m, "InsertFile", ctx, arg) ret0, _ := ret[0].(database.File) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertFile indicates an expected call of InsertFile. -func (mr *MockStoreMockRecorder) InsertFile(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertFile(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertFile", reflect.TypeOf((*MockStore)(nil).InsertFile), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertFile", reflect.TypeOf((*MockStore)(nil).InsertFile), ctx, arg) } // InsertGitSSHKey mocks base method. -func (m *MockStore) InsertGitSSHKey(arg0 context.Context, arg1 database.InsertGitSSHKeyParams) (database.GitSSHKey, error) { +func (m *MockStore) InsertGitSSHKey(ctx context.Context, arg database.InsertGitSSHKeyParams) (database.GitSSHKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertGitSSHKey", arg0, arg1) + ret := m.ctrl.Call(m, "InsertGitSSHKey", ctx, arg) ret0, _ := ret[0].(database.GitSSHKey) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertGitSSHKey indicates an expected call of InsertGitSSHKey. -func (mr *MockStoreMockRecorder) InsertGitSSHKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertGitSSHKey(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertGitSSHKey", reflect.TypeOf((*MockStore)(nil).InsertGitSSHKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertGitSSHKey", reflect.TypeOf((*MockStore)(nil).InsertGitSSHKey), ctx, arg) } // InsertGroup mocks base method. -func (m *MockStore) InsertGroup(arg0 context.Context, arg1 database.InsertGroupParams) (database.Group, error) { +func (m *MockStore) InsertGroup(ctx context.Context, arg database.InsertGroupParams) (database.Group, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertGroup", arg0, arg1) + ret := m.ctrl.Call(m, "InsertGroup", ctx, arg) ret0, _ := ret[0].(database.Group) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertGroup indicates an expected call of InsertGroup. -func (mr *MockStoreMockRecorder) InsertGroup(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertGroup(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertGroup", reflect.TypeOf((*MockStore)(nil).InsertGroup), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertGroup", reflect.TypeOf((*MockStore)(nil).InsertGroup), ctx, arg) } // InsertGroupMember mocks base method. -func (m *MockStore) InsertGroupMember(arg0 context.Context, arg1 database.InsertGroupMemberParams) error { +func (m *MockStore) InsertGroupMember(ctx context.Context, arg database.InsertGroupMemberParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertGroupMember", arg0, arg1) + ret := m.ctrl.Call(m, "InsertGroupMember", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertGroupMember indicates an expected call of InsertGroupMember. -func (mr *MockStoreMockRecorder) InsertGroupMember(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertGroupMember(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertGroupMember", reflect.TypeOf((*MockStore)(nil).InsertGroupMember), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertGroupMember", reflect.TypeOf((*MockStore)(nil).InsertGroupMember), ctx, arg) } // InsertLicense mocks base method. -func (m *MockStore) InsertLicense(arg0 context.Context, arg1 database.InsertLicenseParams) (database.License, error) { +func (m *MockStore) InsertLicense(ctx context.Context, arg database.InsertLicenseParams) (database.License, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertLicense", arg0, arg1) + ret := m.ctrl.Call(m, "InsertLicense", ctx, arg) ret0, _ := ret[0].(database.License) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertLicense indicates an expected call of InsertLicense. -func (mr *MockStoreMockRecorder) InsertLicense(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertLicense(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertLicense", reflect.TypeOf((*MockStore)(nil).InsertLicense), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertLicense", reflect.TypeOf((*MockStore)(nil).InsertLicense), ctx, arg) } // InsertMissingGroups mocks base method. -func (m *MockStore) InsertMissingGroups(arg0 context.Context, arg1 database.InsertMissingGroupsParams) ([]database.Group, error) { +func (m *MockStore) InsertMissingGroups(ctx context.Context, arg database.InsertMissingGroupsParams) ([]database.Group, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertMissingGroups", arg0, arg1) + ret := m.ctrl.Call(m, "InsertMissingGroups", ctx, arg) ret0, _ := ret[0].([]database.Group) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertMissingGroups indicates an expected call of InsertMissingGroups. -func (mr *MockStoreMockRecorder) InsertMissingGroups(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertMissingGroups(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertMissingGroups", reflect.TypeOf((*MockStore)(nil).InsertMissingGroups), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertMissingGroups", reflect.TypeOf((*MockStore)(nil).InsertMissingGroups), ctx, arg) } // InsertOAuth2ProviderApp mocks base method. -func (m *MockStore) InsertOAuth2ProviderApp(arg0 context.Context, arg1 database.InsertOAuth2ProviderAppParams) (database.OAuth2ProviderApp, error) { +func (m *MockStore) InsertOAuth2ProviderApp(ctx context.Context, arg database.InsertOAuth2ProviderAppParams) (database.OAuth2ProviderApp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertOAuth2ProviderApp", arg0, arg1) + ret := m.ctrl.Call(m, "InsertOAuth2ProviderApp", ctx, arg) ret0, _ := ret[0].(database.OAuth2ProviderApp) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertOAuth2ProviderApp indicates an expected call of InsertOAuth2ProviderApp. -func (mr *MockStoreMockRecorder) InsertOAuth2ProviderApp(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertOAuth2ProviderApp(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOAuth2ProviderApp", reflect.TypeOf((*MockStore)(nil).InsertOAuth2ProviderApp), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOAuth2ProviderApp", reflect.TypeOf((*MockStore)(nil).InsertOAuth2ProviderApp), ctx, arg) } // InsertOAuth2ProviderAppCode mocks base method. -func (m *MockStore) InsertOAuth2ProviderAppCode(arg0 context.Context, arg1 database.InsertOAuth2ProviderAppCodeParams) (database.OAuth2ProviderAppCode, error) { +func (m *MockStore) InsertOAuth2ProviderAppCode(ctx context.Context, arg database.InsertOAuth2ProviderAppCodeParams) (database.OAuth2ProviderAppCode, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertOAuth2ProviderAppCode", arg0, arg1) + ret := m.ctrl.Call(m, "InsertOAuth2ProviderAppCode", ctx, arg) ret0, _ := ret[0].(database.OAuth2ProviderAppCode) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertOAuth2ProviderAppCode indicates an expected call of InsertOAuth2ProviderAppCode. -func (mr *MockStoreMockRecorder) InsertOAuth2ProviderAppCode(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertOAuth2ProviderAppCode(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOAuth2ProviderAppCode", reflect.TypeOf((*MockStore)(nil).InsertOAuth2ProviderAppCode), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOAuth2ProviderAppCode", reflect.TypeOf((*MockStore)(nil).InsertOAuth2ProviderAppCode), ctx, arg) } // InsertOAuth2ProviderAppSecret mocks base method. -func (m *MockStore) InsertOAuth2ProviderAppSecret(arg0 context.Context, arg1 database.InsertOAuth2ProviderAppSecretParams) (database.OAuth2ProviderAppSecret, error) { +func (m *MockStore) InsertOAuth2ProviderAppSecret(ctx context.Context, arg database.InsertOAuth2ProviderAppSecretParams) (database.OAuth2ProviderAppSecret, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertOAuth2ProviderAppSecret", arg0, arg1) + ret := m.ctrl.Call(m, "InsertOAuth2ProviderAppSecret", ctx, arg) ret0, _ := ret[0].(database.OAuth2ProviderAppSecret) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertOAuth2ProviderAppSecret indicates an expected call of InsertOAuth2ProviderAppSecret. -func (mr *MockStoreMockRecorder) InsertOAuth2ProviderAppSecret(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertOAuth2ProviderAppSecret(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOAuth2ProviderAppSecret", reflect.TypeOf((*MockStore)(nil).InsertOAuth2ProviderAppSecret), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOAuth2ProviderAppSecret", reflect.TypeOf((*MockStore)(nil).InsertOAuth2ProviderAppSecret), ctx, arg) } // InsertOAuth2ProviderAppToken mocks base method. -func (m *MockStore) InsertOAuth2ProviderAppToken(arg0 context.Context, arg1 database.InsertOAuth2ProviderAppTokenParams) (database.OAuth2ProviderAppToken, error) { +func (m *MockStore) InsertOAuth2ProviderAppToken(ctx context.Context, arg database.InsertOAuth2ProviderAppTokenParams) (database.OAuth2ProviderAppToken, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertOAuth2ProviderAppToken", arg0, arg1) + ret := m.ctrl.Call(m, "InsertOAuth2ProviderAppToken", ctx, arg) ret0, _ := ret[0].(database.OAuth2ProviderAppToken) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertOAuth2ProviderAppToken indicates an expected call of InsertOAuth2ProviderAppToken. -func (mr *MockStoreMockRecorder) InsertOAuth2ProviderAppToken(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertOAuth2ProviderAppToken(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOAuth2ProviderAppToken", reflect.TypeOf((*MockStore)(nil).InsertOAuth2ProviderAppToken), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOAuth2ProviderAppToken", reflect.TypeOf((*MockStore)(nil).InsertOAuth2ProviderAppToken), ctx, arg) } // InsertOrganization mocks base method. -func (m *MockStore) InsertOrganization(arg0 context.Context, arg1 database.InsertOrganizationParams) (database.Organization, error) { +func (m *MockStore) InsertOrganization(ctx context.Context, arg database.InsertOrganizationParams) (database.Organization, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertOrganization", arg0, arg1) + ret := m.ctrl.Call(m, "InsertOrganization", ctx, arg) ret0, _ := ret[0].(database.Organization) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertOrganization indicates an expected call of InsertOrganization. -func (mr *MockStoreMockRecorder) InsertOrganization(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertOrganization(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOrganization", reflect.TypeOf((*MockStore)(nil).InsertOrganization), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOrganization", reflect.TypeOf((*MockStore)(nil).InsertOrganization), ctx, arg) } // InsertOrganizationMember mocks base method. -func (m *MockStore) InsertOrganizationMember(arg0 context.Context, arg1 database.InsertOrganizationMemberParams) (database.OrganizationMember, error) { +func (m *MockStore) InsertOrganizationMember(ctx context.Context, arg database.InsertOrganizationMemberParams) (database.OrganizationMember, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertOrganizationMember", arg0, arg1) + ret := m.ctrl.Call(m, "InsertOrganizationMember", ctx, arg) ret0, _ := ret[0].(database.OrganizationMember) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertOrganizationMember indicates an expected call of InsertOrganizationMember. -func (mr *MockStoreMockRecorder) InsertOrganizationMember(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertOrganizationMember(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOrganizationMember", reflect.TypeOf((*MockStore)(nil).InsertOrganizationMember), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertOrganizationMember", reflect.TypeOf((*MockStore)(nil).InsertOrganizationMember), ctx, arg) } // InsertProvisionerJob mocks base method. -func (m *MockStore) InsertProvisionerJob(arg0 context.Context, arg1 database.InsertProvisionerJobParams) (database.ProvisionerJob, error) { +func (m *MockStore) InsertProvisionerJob(ctx context.Context, arg database.InsertProvisionerJobParams) (database.ProvisionerJob, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertProvisionerJob", arg0, arg1) + ret := m.ctrl.Call(m, "InsertProvisionerJob", ctx, arg) ret0, _ := ret[0].(database.ProvisionerJob) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertProvisionerJob indicates an expected call of InsertProvisionerJob. -func (mr *MockStoreMockRecorder) InsertProvisionerJob(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertProvisionerJob(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertProvisionerJob", reflect.TypeOf((*MockStore)(nil).InsertProvisionerJob), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertProvisionerJob", reflect.TypeOf((*MockStore)(nil).InsertProvisionerJob), ctx, arg) } // InsertProvisionerJobLogs mocks base method. -func (m *MockStore) InsertProvisionerJobLogs(arg0 context.Context, arg1 database.InsertProvisionerJobLogsParams) ([]database.ProvisionerJobLog, error) { +func (m *MockStore) InsertProvisionerJobLogs(ctx context.Context, arg database.InsertProvisionerJobLogsParams) ([]database.ProvisionerJobLog, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertProvisionerJobLogs", arg0, arg1) + ret := m.ctrl.Call(m, "InsertProvisionerJobLogs", ctx, arg) ret0, _ := ret[0].([]database.ProvisionerJobLog) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertProvisionerJobLogs indicates an expected call of InsertProvisionerJobLogs. -func (mr *MockStoreMockRecorder) InsertProvisionerJobLogs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertProvisionerJobLogs(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertProvisionerJobLogs", reflect.TypeOf((*MockStore)(nil).InsertProvisionerJobLogs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertProvisionerJobLogs", reflect.TypeOf((*MockStore)(nil).InsertProvisionerJobLogs), ctx, arg) } // InsertProvisionerJobTimings mocks base method. -func (m *MockStore) InsertProvisionerJobTimings(arg0 context.Context, arg1 database.InsertProvisionerJobTimingsParams) ([]database.ProvisionerJobTiming, error) { +func (m *MockStore) InsertProvisionerJobTimings(ctx context.Context, arg database.InsertProvisionerJobTimingsParams) ([]database.ProvisionerJobTiming, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertProvisionerJobTimings", arg0, arg1) + ret := m.ctrl.Call(m, "InsertProvisionerJobTimings", ctx, arg) ret0, _ := ret[0].([]database.ProvisionerJobTiming) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertProvisionerJobTimings indicates an expected call of InsertProvisionerJobTimings. -func (mr *MockStoreMockRecorder) InsertProvisionerJobTimings(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertProvisionerJobTimings(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertProvisionerJobTimings", reflect.TypeOf((*MockStore)(nil).InsertProvisionerJobTimings), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertProvisionerJobTimings", reflect.TypeOf((*MockStore)(nil).InsertProvisionerJobTimings), ctx, arg) } // InsertProvisionerKey mocks base method. -func (m *MockStore) InsertProvisionerKey(arg0 context.Context, arg1 database.InsertProvisionerKeyParams) (database.ProvisionerKey, error) { +func (m *MockStore) InsertProvisionerKey(ctx context.Context, arg database.InsertProvisionerKeyParams) (database.ProvisionerKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertProvisionerKey", arg0, arg1) + ret := m.ctrl.Call(m, "InsertProvisionerKey", ctx, arg) ret0, _ := ret[0].(database.ProvisionerKey) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertProvisionerKey indicates an expected call of InsertProvisionerKey. -func (mr *MockStoreMockRecorder) InsertProvisionerKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertProvisionerKey(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertProvisionerKey", reflect.TypeOf((*MockStore)(nil).InsertProvisionerKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertProvisionerKey", reflect.TypeOf((*MockStore)(nil).InsertProvisionerKey), ctx, arg) } // InsertReplica mocks base method. -func (m *MockStore) InsertReplica(arg0 context.Context, arg1 database.InsertReplicaParams) (database.Replica, error) { +func (m *MockStore) InsertReplica(ctx context.Context, arg database.InsertReplicaParams) (database.Replica, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertReplica", arg0, arg1) + ret := m.ctrl.Call(m, "InsertReplica", ctx, arg) ret0, _ := ret[0].(database.Replica) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertReplica indicates an expected call of InsertReplica. -func (mr *MockStoreMockRecorder) InsertReplica(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertReplica(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertReplica", reflect.TypeOf((*MockStore)(nil).InsertReplica), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertReplica", reflect.TypeOf((*MockStore)(nil).InsertReplica), ctx, arg) } // InsertTemplate mocks base method. -func (m *MockStore) InsertTemplate(arg0 context.Context, arg1 database.InsertTemplateParams) error { +func (m *MockStore) InsertTemplate(ctx context.Context, arg database.InsertTemplateParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertTemplate", arg0, arg1) + ret := m.ctrl.Call(m, "InsertTemplate", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertTemplate indicates an expected call of InsertTemplate. -func (mr *MockStoreMockRecorder) InsertTemplate(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertTemplate(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplate", reflect.TypeOf((*MockStore)(nil).InsertTemplate), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplate", reflect.TypeOf((*MockStore)(nil).InsertTemplate), ctx, arg) } // InsertTemplateVersion mocks base method. -func (m *MockStore) InsertTemplateVersion(arg0 context.Context, arg1 database.InsertTemplateVersionParams) error { +func (m *MockStore) InsertTemplateVersion(ctx context.Context, arg database.InsertTemplateVersionParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertTemplateVersion", arg0, arg1) + ret := m.ctrl.Call(m, "InsertTemplateVersion", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertTemplateVersion indicates an expected call of InsertTemplateVersion. -func (mr *MockStoreMockRecorder) InsertTemplateVersion(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertTemplateVersion(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplateVersion", reflect.TypeOf((*MockStore)(nil).InsertTemplateVersion), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplateVersion", reflect.TypeOf((*MockStore)(nil).InsertTemplateVersion), ctx, arg) } // InsertTemplateVersionParameter mocks base method. -func (m *MockStore) InsertTemplateVersionParameter(arg0 context.Context, arg1 database.InsertTemplateVersionParameterParams) (database.TemplateVersionParameter, error) { +func (m *MockStore) InsertTemplateVersionParameter(ctx context.Context, arg database.InsertTemplateVersionParameterParams) (database.TemplateVersionParameter, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertTemplateVersionParameter", arg0, arg1) + ret := m.ctrl.Call(m, "InsertTemplateVersionParameter", ctx, arg) ret0, _ := ret[0].(database.TemplateVersionParameter) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertTemplateVersionParameter indicates an expected call of InsertTemplateVersionParameter. -func (mr *MockStoreMockRecorder) InsertTemplateVersionParameter(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertTemplateVersionParameter(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplateVersionParameter", reflect.TypeOf((*MockStore)(nil).InsertTemplateVersionParameter), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplateVersionParameter", reflect.TypeOf((*MockStore)(nil).InsertTemplateVersionParameter), ctx, arg) } // InsertTemplateVersionVariable mocks base method. -func (m *MockStore) InsertTemplateVersionVariable(arg0 context.Context, arg1 database.InsertTemplateVersionVariableParams) (database.TemplateVersionVariable, error) { +func (m *MockStore) InsertTemplateVersionVariable(ctx context.Context, arg database.InsertTemplateVersionVariableParams) (database.TemplateVersionVariable, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertTemplateVersionVariable", arg0, arg1) + ret := m.ctrl.Call(m, "InsertTemplateVersionVariable", ctx, arg) ret0, _ := ret[0].(database.TemplateVersionVariable) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertTemplateVersionVariable indicates an expected call of InsertTemplateVersionVariable. -func (mr *MockStoreMockRecorder) InsertTemplateVersionVariable(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertTemplateVersionVariable(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplateVersionVariable", reflect.TypeOf((*MockStore)(nil).InsertTemplateVersionVariable), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplateVersionVariable", reflect.TypeOf((*MockStore)(nil).InsertTemplateVersionVariable), ctx, arg) } // InsertTemplateVersionWorkspaceTag mocks base method. -func (m *MockStore) InsertTemplateVersionWorkspaceTag(arg0 context.Context, arg1 database.InsertTemplateVersionWorkspaceTagParams) (database.TemplateVersionWorkspaceTag, error) { +func (m *MockStore) InsertTemplateVersionWorkspaceTag(ctx context.Context, arg database.InsertTemplateVersionWorkspaceTagParams) (database.TemplateVersionWorkspaceTag, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertTemplateVersionWorkspaceTag", arg0, arg1) + ret := m.ctrl.Call(m, "InsertTemplateVersionWorkspaceTag", ctx, arg) ret0, _ := ret[0].(database.TemplateVersionWorkspaceTag) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertTemplateVersionWorkspaceTag indicates an expected call of InsertTemplateVersionWorkspaceTag. -func (mr *MockStoreMockRecorder) InsertTemplateVersionWorkspaceTag(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertTemplateVersionWorkspaceTag(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplateVersionWorkspaceTag", reflect.TypeOf((*MockStore)(nil).InsertTemplateVersionWorkspaceTag), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertTemplateVersionWorkspaceTag", reflect.TypeOf((*MockStore)(nil).InsertTemplateVersionWorkspaceTag), ctx, arg) } // InsertUser mocks base method. -func (m *MockStore) InsertUser(arg0 context.Context, arg1 database.InsertUserParams) (database.User, error) { +func (m *MockStore) InsertUser(ctx context.Context, arg database.InsertUserParams) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertUser", arg0, arg1) + ret := m.ctrl.Call(m, "InsertUser", ctx, arg) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertUser indicates an expected call of InsertUser. -func (mr *MockStoreMockRecorder) InsertUser(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertUser(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUser", reflect.TypeOf((*MockStore)(nil).InsertUser), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUser", reflect.TypeOf((*MockStore)(nil).InsertUser), ctx, arg) } // InsertUserGroupsByID mocks base method. -func (m *MockStore) InsertUserGroupsByID(arg0 context.Context, arg1 database.InsertUserGroupsByIDParams) ([]uuid.UUID, error) { +func (m *MockStore) InsertUserGroupsByID(ctx context.Context, arg database.InsertUserGroupsByIDParams) ([]uuid.UUID, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertUserGroupsByID", arg0, arg1) + ret := m.ctrl.Call(m, "InsertUserGroupsByID", ctx, arg) ret0, _ := ret[0].([]uuid.UUID) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertUserGroupsByID indicates an expected call of InsertUserGroupsByID. -func (mr *MockStoreMockRecorder) InsertUserGroupsByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertUserGroupsByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUserGroupsByID", reflect.TypeOf((*MockStore)(nil).InsertUserGroupsByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUserGroupsByID", reflect.TypeOf((*MockStore)(nil).InsertUserGroupsByID), ctx, arg) } // InsertUserGroupsByName mocks base method. -func (m *MockStore) InsertUserGroupsByName(arg0 context.Context, arg1 database.InsertUserGroupsByNameParams) error { +func (m *MockStore) InsertUserGroupsByName(ctx context.Context, arg database.InsertUserGroupsByNameParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertUserGroupsByName", arg0, arg1) + ret := m.ctrl.Call(m, "InsertUserGroupsByName", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertUserGroupsByName indicates an expected call of InsertUserGroupsByName. -func (mr *MockStoreMockRecorder) InsertUserGroupsByName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertUserGroupsByName(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUserGroupsByName", reflect.TypeOf((*MockStore)(nil).InsertUserGroupsByName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUserGroupsByName", reflect.TypeOf((*MockStore)(nil).InsertUserGroupsByName), ctx, arg) } // InsertUserLink mocks base method. -func (m *MockStore) InsertUserLink(arg0 context.Context, arg1 database.InsertUserLinkParams) (database.UserLink, error) { +func (m *MockStore) InsertUserLink(ctx context.Context, arg database.InsertUserLinkParams) (database.UserLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertUserLink", arg0, arg1) + ret := m.ctrl.Call(m, "InsertUserLink", ctx, arg) ret0, _ := ret[0].(database.UserLink) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertUserLink indicates an expected call of InsertUserLink. -func (mr *MockStoreMockRecorder) InsertUserLink(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertUserLink(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUserLink", reflect.TypeOf((*MockStore)(nil).InsertUserLink), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertUserLink", reflect.TypeOf((*MockStore)(nil).InsertUserLink), ctx, arg) } // InsertWorkspace mocks base method. -func (m *MockStore) InsertWorkspace(arg0 context.Context, arg1 database.InsertWorkspaceParams) (database.WorkspaceTable, error) { +func (m *MockStore) InsertWorkspace(ctx context.Context, arg database.InsertWorkspaceParams) (database.WorkspaceTable, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspace", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspace", ctx, arg) ret0, _ := ret[0].(database.WorkspaceTable) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspace indicates an expected call of InsertWorkspace. -func (mr *MockStoreMockRecorder) InsertWorkspace(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspace(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspace", reflect.TypeOf((*MockStore)(nil).InsertWorkspace), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspace", reflect.TypeOf((*MockStore)(nil).InsertWorkspace), ctx, arg) } // InsertWorkspaceAgent mocks base method. -func (m *MockStore) InsertWorkspaceAgent(arg0 context.Context, arg1 database.InsertWorkspaceAgentParams) (database.WorkspaceAgent, error) { +func (m *MockStore) InsertWorkspaceAgent(ctx context.Context, arg database.InsertWorkspaceAgentParams) (database.WorkspaceAgent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceAgent", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceAgent", ctx, arg) ret0, _ := ret[0].(database.WorkspaceAgent) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceAgent indicates an expected call of InsertWorkspaceAgent. -func (mr *MockStoreMockRecorder) InsertWorkspaceAgent(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceAgent(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgent", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgent), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgent", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgent), ctx, arg) } // InsertWorkspaceAgentLogSources mocks base method. -func (m *MockStore) InsertWorkspaceAgentLogSources(arg0 context.Context, arg1 database.InsertWorkspaceAgentLogSourcesParams) ([]database.WorkspaceAgentLogSource, error) { +func (m *MockStore) InsertWorkspaceAgentLogSources(ctx context.Context, arg database.InsertWorkspaceAgentLogSourcesParams) ([]database.WorkspaceAgentLogSource, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceAgentLogSources", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceAgentLogSources", ctx, arg) ret0, _ := ret[0].([]database.WorkspaceAgentLogSource) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceAgentLogSources indicates an expected call of InsertWorkspaceAgentLogSources. -func (mr *MockStoreMockRecorder) InsertWorkspaceAgentLogSources(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceAgentLogSources(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentLogSources", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentLogSources), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentLogSources", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentLogSources), ctx, arg) } // InsertWorkspaceAgentLogs mocks base method. -func (m *MockStore) InsertWorkspaceAgentLogs(arg0 context.Context, arg1 database.InsertWorkspaceAgentLogsParams) ([]database.WorkspaceAgentLog, error) { +func (m *MockStore) InsertWorkspaceAgentLogs(ctx context.Context, arg database.InsertWorkspaceAgentLogsParams) ([]database.WorkspaceAgentLog, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceAgentLogs", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceAgentLogs", ctx, arg) ret0, _ := ret[0].([]database.WorkspaceAgentLog) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceAgentLogs indicates an expected call of InsertWorkspaceAgentLogs. -func (mr *MockStoreMockRecorder) InsertWorkspaceAgentLogs(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceAgentLogs(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentLogs", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentLogs), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentLogs", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentLogs), ctx, arg) } // InsertWorkspaceAgentMetadata mocks base method. -func (m *MockStore) InsertWorkspaceAgentMetadata(arg0 context.Context, arg1 database.InsertWorkspaceAgentMetadataParams) error { +func (m *MockStore) InsertWorkspaceAgentMetadata(ctx context.Context, arg database.InsertWorkspaceAgentMetadataParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceAgentMetadata", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceAgentMetadata", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertWorkspaceAgentMetadata indicates an expected call of InsertWorkspaceAgentMetadata. -func (mr *MockStoreMockRecorder) InsertWorkspaceAgentMetadata(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceAgentMetadata(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentMetadata", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentMetadata), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentMetadata", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentMetadata), ctx, arg) } // InsertWorkspaceAgentScriptTimings mocks base method. -func (m *MockStore) InsertWorkspaceAgentScriptTimings(arg0 context.Context, arg1 database.InsertWorkspaceAgentScriptTimingsParams) (database.WorkspaceAgentScriptTiming, error) { +func (m *MockStore) InsertWorkspaceAgentScriptTimings(ctx context.Context, arg database.InsertWorkspaceAgentScriptTimingsParams) (database.WorkspaceAgentScriptTiming, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceAgentScriptTimings", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceAgentScriptTimings", ctx, arg) ret0, _ := ret[0].(database.WorkspaceAgentScriptTiming) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceAgentScriptTimings indicates an expected call of InsertWorkspaceAgentScriptTimings. -func (mr *MockStoreMockRecorder) InsertWorkspaceAgentScriptTimings(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceAgentScriptTimings(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentScriptTimings", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentScriptTimings), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentScriptTimings", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentScriptTimings), ctx, arg) } // InsertWorkspaceAgentScripts mocks base method. -func (m *MockStore) InsertWorkspaceAgentScripts(arg0 context.Context, arg1 database.InsertWorkspaceAgentScriptsParams) ([]database.WorkspaceAgentScript, error) { +func (m *MockStore) InsertWorkspaceAgentScripts(ctx context.Context, arg database.InsertWorkspaceAgentScriptsParams) ([]database.WorkspaceAgentScript, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceAgentScripts", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceAgentScripts", ctx, arg) ret0, _ := ret[0].([]database.WorkspaceAgentScript) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceAgentScripts indicates an expected call of InsertWorkspaceAgentScripts. -func (mr *MockStoreMockRecorder) InsertWorkspaceAgentScripts(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceAgentScripts(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentScripts", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentScripts), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentScripts", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentScripts), ctx, arg) } // InsertWorkspaceAgentStats mocks base method. -func (m *MockStore) InsertWorkspaceAgentStats(arg0 context.Context, arg1 database.InsertWorkspaceAgentStatsParams) error { +func (m *MockStore) InsertWorkspaceAgentStats(ctx context.Context, arg database.InsertWorkspaceAgentStatsParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceAgentStats", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceAgentStats", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertWorkspaceAgentStats indicates an expected call of InsertWorkspaceAgentStats. -func (mr *MockStoreMockRecorder) InsertWorkspaceAgentStats(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceAgentStats(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentStats", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentStats), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAgentStats", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAgentStats), ctx, arg) } // InsertWorkspaceApp mocks base method. -func (m *MockStore) InsertWorkspaceApp(arg0 context.Context, arg1 database.InsertWorkspaceAppParams) (database.WorkspaceApp, error) { +func (m *MockStore) InsertWorkspaceApp(ctx context.Context, arg database.InsertWorkspaceAppParams) (database.WorkspaceApp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceApp", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceApp", ctx, arg) ret0, _ := ret[0].(database.WorkspaceApp) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceApp indicates an expected call of InsertWorkspaceApp. -func (mr *MockStoreMockRecorder) InsertWorkspaceApp(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceApp(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceApp", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceApp), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceApp", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceApp), ctx, arg) } // InsertWorkspaceAppStats mocks base method. -func (m *MockStore) InsertWorkspaceAppStats(arg0 context.Context, arg1 database.InsertWorkspaceAppStatsParams) error { +func (m *MockStore) InsertWorkspaceAppStats(ctx context.Context, arg database.InsertWorkspaceAppStatsParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceAppStats", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceAppStats", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertWorkspaceAppStats indicates an expected call of InsertWorkspaceAppStats. -func (mr *MockStoreMockRecorder) InsertWorkspaceAppStats(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceAppStats(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAppStats", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAppStats), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceAppStats", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceAppStats), ctx, arg) } // InsertWorkspaceBuild mocks base method. -func (m *MockStore) InsertWorkspaceBuild(arg0 context.Context, arg1 database.InsertWorkspaceBuildParams) error { +func (m *MockStore) InsertWorkspaceBuild(ctx context.Context, arg database.InsertWorkspaceBuildParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceBuild", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceBuild", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertWorkspaceBuild indicates an expected call of InsertWorkspaceBuild. -func (mr *MockStoreMockRecorder) InsertWorkspaceBuild(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceBuild(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceBuild", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceBuild), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceBuild", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceBuild), ctx, arg) } // InsertWorkspaceBuildParameters mocks base method. -func (m *MockStore) InsertWorkspaceBuildParameters(arg0 context.Context, arg1 database.InsertWorkspaceBuildParametersParams) error { +func (m *MockStore) InsertWorkspaceBuildParameters(ctx context.Context, arg database.InsertWorkspaceBuildParametersParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceBuildParameters", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceBuildParameters", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // InsertWorkspaceBuildParameters indicates an expected call of InsertWorkspaceBuildParameters. -func (mr *MockStoreMockRecorder) InsertWorkspaceBuildParameters(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceBuildParameters(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceBuildParameters", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceBuildParameters), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceBuildParameters", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceBuildParameters), ctx, arg) } // InsertWorkspaceModule mocks base method. -func (m *MockStore) InsertWorkspaceModule(arg0 context.Context, arg1 database.InsertWorkspaceModuleParams) (database.WorkspaceModule, error) { +func (m *MockStore) InsertWorkspaceModule(ctx context.Context, arg database.InsertWorkspaceModuleParams) (database.WorkspaceModule, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceModule", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceModule", ctx, arg) ret0, _ := ret[0].(database.WorkspaceModule) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceModule indicates an expected call of InsertWorkspaceModule. -func (mr *MockStoreMockRecorder) InsertWorkspaceModule(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceModule(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceModule", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceModule), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceModule", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceModule), ctx, arg) } // InsertWorkspaceProxy mocks base method. -func (m *MockStore) InsertWorkspaceProxy(arg0 context.Context, arg1 database.InsertWorkspaceProxyParams) (database.WorkspaceProxy, error) { +func (m *MockStore) InsertWorkspaceProxy(ctx context.Context, arg database.InsertWorkspaceProxyParams) (database.WorkspaceProxy, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceProxy", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceProxy", ctx, arg) ret0, _ := ret[0].(database.WorkspaceProxy) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceProxy indicates an expected call of InsertWorkspaceProxy. -func (mr *MockStoreMockRecorder) InsertWorkspaceProxy(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceProxy(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceProxy", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceProxy), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceProxy", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceProxy), ctx, arg) } // InsertWorkspaceResource mocks base method. -func (m *MockStore) InsertWorkspaceResource(arg0 context.Context, arg1 database.InsertWorkspaceResourceParams) (database.WorkspaceResource, error) { +func (m *MockStore) InsertWorkspaceResource(ctx context.Context, arg database.InsertWorkspaceResourceParams) (database.WorkspaceResource, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceResource", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceResource", ctx, arg) ret0, _ := ret[0].(database.WorkspaceResource) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceResource indicates an expected call of InsertWorkspaceResource. -func (mr *MockStoreMockRecorder) InsertWorkspaceResource(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceResource(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceResource", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceResource), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceResource", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceResource), ctx, arg) } // InsertWorkspaceResourceMetadata mocks base method. -func (m *MockStore) InsertWorkspaceResourceMetadata(arg0 context.Context, arg1 database.InsertWorkspaceResourceMetadataParams) ([]database.WorkspaceResourceMetadatum, error) { +func (m *MockStore) InsertWorkspaceResourceMetadata(ctx context.Context, arg database.InsertWorkspaceResourceMetadataParams) ([]database.WorkspaceResourceMetadatum, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "InsertWorkspaceResourceMetadata", arg0, arg1) + ret := m.ctrl.Call(m, "InsertWorkspaceResourceMetadata", ctx, arg) ret0, _ := ret[0].([]database.WorkspaceResourceMetadatum) ret1, _ := ret[1].(error) return ret0, ret1 } // InsertWorkspaceResourceMetadata indicates an expected call of InsertWorkspaceResourceMetadata. -func (mr *MockStoreMockRecorder) InsertWorkspaceResourceMetadata(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) InsertWorkspaceResourceMetadata(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceResourceMetadata", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceResourceMetadata), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertWorkspaceResourceMetadata", reflect.TypeOf((*MockStore)(nil).InsertWorkspaceResourceMetadata), ctx, arg) } // ListProvisionerKeysByOrganization mocks base method. -func (m *MockStore) ListProvisionerKeysByOrganization(arg0 context.Context, arg1 uuid.UUID) ([]database.ProvisionerKey, error) { +func (m *MockStore) ListProvisionerKeysByOrganization(ctx context.Context, organizationID uuid.UUID) ([]database.ProvisionerKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListProvisionerKeysByOrganization", arg0, arg1) + ret := m.ctrl.Call(m, "ListProvisionerKeysByOrganization", ctx, organizationID) ret0, _ := ret[0].([]database.ProvisionerKey) ret1, _ := ret[1].(error) return ret0, ret1 } // ListProvisionerKeysByOrganization indicates an expected call of ListProvisionerKeysByOrganization. -func (mr *MockStoreMockRecorder) ListProvisionerKeysByOrganization(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) ListProvisionerKeysByOrganization(ctx, organizationID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProvisionerKeysByOrganization", reflect.TypeOf((*MockStore)(nil).ListProvisionerKeysByOrganization), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProvisionerKeysByOrganization", reflect.TypeOf((*MockStore)(nil).ListProvisionerKeysByOrganization), ctx, organizationID) } // ListProvisionerKeysByOrganizationExcludeReserved mocks base method. -func (m *MockStore) ListProvisionerKeysByOrganizationExcludeReserved(arg0 context.Context, arg1 uuid.UUID) ([]database.ProvisionerKey, error) { +func (m *MockStore) ListProvisionerKeysByOrganizationExcludeReserved(ctx context.Context, organizationID uuid.UUID) ([]database.ProvisionerKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListProvisionerKeysByOrganizationExcludeReserved", arg0, arg1) + ret := m.ctrl.Call(m, "ListProvisionerKeysByOrganizationExcludeReserved", ctx, organizationID) ret0, _ := ret[0].([]database.ProvisionerKey) ret1, _ := ret[1].(error) return ret0, ret1 } // ListProvisionerKeysByOrganizationExcludeReserved indicates an expected call of ListProvisionerKeysByOrganizationExcludeReserved. -func (mr *MockStoreMockRecorder) ListProvisionerKeysByOrganizationExcludeReserved(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) ListProvisionerKeysByOrganizationExcludeReserved(ctx, organizationID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProvisionerKeysByOrganizationExcludeReserved", reflect.TypeOf((*MockStore)(nil).ListProvisionerKeysByOrganizationExcludeReserved), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListProvisionerKeysByOrganizationExcludeReserved", reflect.TypeOf((*MockStore)(nil).ListProvisionerKeysByOrganizationExcludeReserved), ctx, organizationID) } // ListWorkspaceAgentPortShares mocks base method. -func (m *MockStore) ListWorkspaceAgentPortShares(arg0 context.Context, arg1 uuid.UUID) ([]database.WorkspaceAgentPortShare, error) { +func (m *MockStore) ListWorkspaceAgentPortShares(ctx context.Context, workspaceID uuid.UUID) ([]database.WorkspaceAgentPortShare, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ListWorkspaceAgentPortShares", arg0, arg1) + ret := m.ctrl.Call(m, "ListWorkspaceAgentPortShares", ctx, workspaceID) ret0, _ := ret[0].([]database.WorkspaceAgentPortShare) ret1, _ := ret[1].(error) return ret0, ret1 } // ListWorkspaceAgentPortShares indicates an expected call of ListWorkspaceAgentPortShares. -func (mr *MockStoreMockRecorder) ListWorkspaceAgentPortShares(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) ListWorkspaceAgentPortShares(ctx, workspaceID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListWorkspaceAgentPortShares", reflect.TypeOf((*MockStore)(nil).ListWorkspaceAgentPortShares), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListWorkspaceAgentPortShares", reflect.TypeOf((*MockStore)(nil).ListWorkspaceAgentPortShares), ctx, workspaceID) } // OIDCClaimFieldValues mocks base method. -func (m *MockStore) OIDCClaimFieldValues(arg0 context.Context, arg1 database.OIDCClaimFieldValuesParams) ([]string, error) { +func (m *MockStore) OIDCClaimFieldValues(ctx context.Context, arg database.OIDCClaimFieldValuesParams) ([]string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "OIDCClaimFieldValues", arg0, arg1) + ret := m.ctrl.Call(m, "OIDCClaimFieldValues", ctx, arg) ret0, _ := ret[0].([]string) ret1, _ := ret[1].(error) return ret0, ret1 } // OIDCClaimFieldValues indicates an expected call of OIDCClaimFieldValues. -func (mr *MockStoreMockRecorder) OIDCClaimFieldValues(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) OIDCClaimFieldValues(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OIDCClaimFieldValues", reflect.TypeOf((*MockStore)(nil).OIDCClaimFieldValues), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OIDCClaimFieldValues", reflect.TypeOf((*MockStore)(nil).OIDCClaimFieldValues), ctx, arg) } // OIDCClaimFields mocks base method. -func (m *MockStore) OIDCClaimFields(arg0 context.Context, arg1 uuid.UUID) ([]string, error) { +func (m *MockStore) OIDCClaimFields(ctx context.Context, organizationID uuid.UUID) ([]string, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "OIDCClaimFields", arg0, arg1) + ret := m.ctrl.Call(m, "OIDCClaimFields", ctx, organizationID) ret0, _ := ret[0].([]string) ret1, _ := ret[1].(error) return ret0, ret1 } // OIDCClaimFields indicates an expected call of OIDCClaimFields. -func (mr *MockStoreMockRecorder) OIDCClaimFields(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) OIDCClaimFields(ctx, organizationID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OIDCClaimFields", reflect.TypeOf((*MockStore)(nil).OIDCClaimFields), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OIDCClaimFields", reflect.TypeOf((*MockStore)(nil).OIDCClaimFields), ctx, organizationID) } // OrganizationMembers mocks base method. -func (m *MockStore) OrganizationMembers(arg0 context.Context, arg1 database.OrganizationMembersParams) ([]database.OrganizationMembersRow, error) { +func (m *MockStore) OrganizationMembers(ctx context.Context, arg database.OrganizationMembersParams) ([]database.OrganizationMembersRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "OrganizationMembers", arg0, arg1) + ret := m.ctrl.Call(m, "OrganizationMembers", ctx, arg) ret0, _ := ret[0].([]database.OrganizationMembersRow) ret1, _ := ret[1].(error) return ret0, ret1 } // OrganizationMembers indicates an expected call of OrganizationMembers. -func (mr *MockStoreMockRecorder) OrganizationMembers(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) OrganizationMembers(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OrganizationMembers", reflect.TypeOf((*MockStore)(nil).OrganizationMembers), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OrganizationMembers", reflect.TypeOf((*MockStore)(nil).OrganizationMembers), ctx, arg) } // PGLocks mocks base method. -func (m *MockStore) PGLocks(arg0 context.Context) (database.PGLocks, error) { +func (m *MockStore) PGLocks(ctx context.Context) (database.PGLocks, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "PGLocks", arg0) + ret := m.ctrl.Call(m, "PGLocks", ctx) ret0, _ := ret[0].(database.PGLocks) ret1, _ := ret[1].(error) return ret0, ret1 } // PGLocks indicates an expected call of PGLocks. -func (mr *MockStoreMockRecorder) PGLocks(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) PGLocks(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PGLocks", reflect.TypeOf((*MockStore)(nil).PGLocks), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PGLocks", reflect.TypeOf((*MockStore)(nil).PGLocks), ctx) } // Ping mocks base method. -func (m *MockStore) Ping(arg0 context.Context) (time.Duration, error) { +func (m *MockStore) Ping(ctx context.Context) (time.Duration, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "Ping", arg0) + ret := m.ctrl.Call(m, "Ping", ctx) ret0, _ := ret[0].(time.Duration) ret1, _ := ret[1].(error) return ret0, ret1 } // Ping indicates an expected call of Ping. -func (mr *MockStoreMockRecorder) Ping(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) Ping(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Ping", reflect.TypeOf((*MockStore)(nil).Ping), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Ping", reflect.TypeOf((*MockStore)(nil).Ping), ctx) } // ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate mocks base method. -func (m *MockStore) ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate(ctx context.Context, templateID uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate", arg0, arg1) + ret := m.ctrl.Call(m, "ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate", ctx, templateID) ret0, _ := ret[0].(error) return ret0 } // ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate indicates an expected call of ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate. -func (mr *MockStoreMockRecorder) ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate(ctx, templateID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate", reflect.TypeOf((*MockStore)(nil).ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate", reflect.TypeOf((*MockStore)(nil).ReduceWorkspaceAgentShareLevelToAuthenticatedByTemplate), ctx, templateID) } // RegisterWorkspaceProxy mocks base method. -func (m *MockStore) RegisterWorkspaceProxy(arg0 context.Context, arg1 database.RegisterWorkspaceProxyParams) (database.WorkspaceProxy, error) { +func (m *MockStore) RegisterWorkspaceProxy(ctx context.Context, arg database.RegisterWorkspaceProxyParams) (database.WorkspaceProxy, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "RegisterWorkspaceProxy", arg0, arg1) + ret := m.ctrl.Call(m, "RegisterWorkspaceProxy", ctx, arg) ret0, _ := ret[0].(database.WorkspaceProxy) ret1, _ := ret[1].(error) return ret0, ret1 } // RegisterWorkspaceProxy indicates an expected call of RegisterWorkspaceProxy. -func (mr *MockStoreMockRecorder) RegisterWorkspaceProxy(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) RegisterWorkspaceProxy(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterWorkspaceProxy", reflect.TypeOf((*MockStore)(nil).RegisterWorkspaceProxy), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterWorkspaceProxy", reflect.TypeOf((*MockStore)(nil).RegisterWorkspaceProxy), ctx, arg) } // RemoveRefreshToken mocks base method. -func (m *MockStore) RemoveRefreshToken(arg0 context.Context, arg1 database.RemoveRefreshTokenParams) error { +func (m *MockStore) RemoveRefreshToken(ctx context.Context, arg database.RemoveRefreshTokenParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "RemoveRefreshToken", arg0, arg1) + ret := m.ctrl.Call(m, "RemoveRefreshToken", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // RemoveRefreshToken indicates an expected call of RemoveRefreshToken. -func (mr *MockStoreMockRecorder) RemoveRefreshToken(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) RemoveRefreshToken(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveRefreshToken", reflect.TypeOf((*MockStore)(nil).RemoveRefreshToken), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveRefreshToken", reflect.TypeOf((*MockStore)(nil).RemoveRefreshToken), ctx, arg) } // RemoveUserFromAllGroups mocks base method. -func (m *MockStore) RemoveUserFromAllGroups(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) RemoveUserFromAllGroups(ctx context.Context, userID uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "RemoveUserFromAllGroups", arg0, arg1) + ret := m.ctrl.Call(m, "RemoveUserFromAllGroups", ctx, userID) ret0, _ := ret[0].(error) return ret0 } // RemoveUserFromAllGroups indicates an expected call of RemoveUserFromAllGroups. -func (mr *MockStoreMockRecorder) RemoveUserFromAllGroups(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) RemoveUserFromAllGroups(ctx, userID any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveUserFromAllGroups", reflect.TypeOf((*MockStore)(nil).RemoveUserFromAllGroups), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveUserFromAllGroups", reflect.TypeOf((*MockStore)(nil).RemoveUserFromAllGroups), ctx, userID) } // RemoveUserFromGroups mocks base method. -func (m *MockStore) RemoveUserFromGroups(arg0 context.Context, arg1 database.RemoveUserFromGroupsParams) ([]uuid.UUID, error) { +func (m *MockStore) RemoveUserFromGroups(ctx context.Context, arg database.RemoveUserFromGroupsParams) ([]uuid.UUID, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "RemoveUserFromGroups", arg0, arg1) + ret := m.ctrl.Call(m, "RemoveUserFromGroups", ctx, arg) ret0, _ := ret[0].([]uuid.UUID) ret1, _ := ret[1].(error) return ret0, ret1 } // RemoveUserFromGroups indicates an expected call of RemoveUserFromGroups. -func (mr *MockStoreMockRecorder) RemoveUserFromGroups(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) RemoveUserFromGroups(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveUserFromGroups", reflect.TypeOf((*MockStore)(nil).RemoveUserFromGroups), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveUserFromGroups", reflect.TypeOf((*MockStore)(nil).RemoveUserFromGroups), ctx, arg) } // RevokeDBCryptKey mocks base method. -func (m *MockStore) RevokeDBCryptKey(arg0 context.Context, arg1 string) error { +func (m *MockStore) RevokeDBCryptKey(ctx context.Context, activeKeyDigest string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "RevokeDBCryptKey", arg0, arg1) + ret := m.ctrl.Call(m, "RevokeDBCryptKey", ctx, activeKeyDigest) ret0, _ := ret[0].(error) return ret0 } // RevokeDBCryptKey indicates an expected call of RevokeDBCryptKey. -func (mr *MockStoreMockRecorder) RevokeDBCryptKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) RevokeDBCryptKey(ctx, activeKeyDigest any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RevokeDBCryptKey", reflect.TypeOf((*MockStore)(nil).RevokeDBCryptKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RevokeDBCryptKey", reflect.TypeOf((*MockStore)(nil).RevokeDBCryptKey), ctx, activeKeyDigest) } // TryAcquireLock mocks base method. -func (m *MockStore) TryAcquireLock(arg0 context.Context, arg1 int64) (bool, error) { +func (m *MockStore) TryAcquireLock(ctx context.Context, pgTryAdvisoryXactLock int64) (bool, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "TryAcquireLock", arg0, arg1) + ret := m.ctrl.Call(m, "TryAcquireLock", ctx, pgTryAdvisoryXactLock) ret0, _ := ret[0].(bool) ret1, _ := ret[1].(error) return ret0, ret1 } // TryAcquireLock indicates an expected call of TryAcquireLock. -func (mr *MockStoreMockRecorder) TryAcquireLock(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) TryAcquireLock(ctx, pgTryAdvisoryXactLock any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TryAcquireLock", reflect.TypeOf((*MockStore)(nil).TryAcquireLock), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TryAcquireLock", reflect.TypeOf((*MockStore)(nil).TryAcquireLock), ctx, pgTryAdvisoryXactLock) } // UnarchiveTemplateVersion mocks base method. -func (m *MockStore) UnarchiveTemplateVersion(arg0 context.Context, arg1 database.UnarchiveTemplateVersionParams) error { +func (m *MockStore) UnarchiveTemplateVersion(ctx context.Context, arg database.UnarchiveTemplateVersionParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UnarchiveTemplateVersion", arg0, arg1) + ret := m.ctrl.Call(m, "UnarchiveTemplateVersion", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UnarchiveTemplateVersion indicates an expected call of UnarchiveTemplateVersion. -func (mr *MockStoreMockRecorder) UnarchiveTemplateVersion(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UnarchiveTemplateVersion(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnarchiveTemplateVersion", reflect.TypeOf((*MockStore)(nil).UnarchiveTemplateVersion), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnarchiveTemplateVersion", reflect.TypeOf((*MockStore)(nil).UnarchiveTemplateVersion), ctx, arg) } // UnfavoriteWorkspace mocks base method. -func (m *MockStore) UnfavoriteWorkspace(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) UnfavoriteWorkspace(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UnfavoriteWorkspace", arg0, arg1) + ret := m.ctrl.Call(m, "UnfavoriteWorkspace", ctx, id) ret0, _ := ret[0].(error) return ret0 } // UnfavoriteWorkspace indicates an expected call of UnfavoriteWorkspace. -func (mr *MockStoreMockRecorder) UnfavoriteWorkspace(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UnfavoriteWorkspace(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnfavoriteWorkspace", reflect.TypeOf((*MockStore)(nil).UnfavoriteWorkspace), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnfavoriteWorkspace", reflect.TypeOf((*MockStore)(nil).UnfavoriteWorkspace), ctx, id) } // UpdateAPIKeyByID mocks base method. -func (m *MockStore) UpdateAPIKeyByID(arg0 context.Context, arg1 database.UpdateAPIKeyByIDParams) error { +func (m *MockStore) UpdateAPIKeyByID(ctx context.Context, arg database.UpdateAPIKeyByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateAPIKeyByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateAPIKeyByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateAPIKeyByID indicates an expected call of UpdateAPIKeyByID. -func (mr *MockStoreMockRecorder) UpdateAPIKeyByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateAPIKeyByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAPIKeyByID", reflect.TypeOf((*MockStore)(nil).UpdateAPIKeyByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAPIKeyByID", reflect.TypeOf((*MockStore)(nil).UpdateAPIKeyByID), ctx, arg) } // UpdateCryptoKeyDeletesAt mocks base method. -func (m *MockStore) UpdateCryptoKeyDeletesAt(arg0 context.Context, arg1 database.UpdateCryptoKeyDeletesAtParams) (database.CryptoKey, error) { +func (m *MockStore) UpdateCryptoKeyDeletesAt(ctx context.Context, arg database.UpdateCryptoKeyDeletesAtParams) (database.CryptoKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateCryptoKeyDeletesAt", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateCryptoKeyDeletesAt", ctx, arg) ret0, _ := ret[0].(database.CryptoKey) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateCryptoKeyDeletesAt indicates an expected call of UpdateCryptoKeyDeletesAt. -func (mr *MockStoreMockRecorder) UpdateCryptoKeyDeletesAt(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateCryptoKeyDeletesAt(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateCryptoKeyDeletesAt", reflect.TypeOf((*MockStore)(nil).UpdateCryptoKeyDeletesAt), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateCryptoKeyDeletesAt", reflect.TypeOf((*MockStore)(nil).UpdateCryptoKeyDeletesAt), ctx, arg) } // UpdateCustomRole mocks base method. -func (m *MockStore) UpdateCustomRole(arg0 context.Context, arg1 database.UpdateCustomRoleParams) (database.CustomRole, error) { +func (m *MockStore) UpdateCustomRole(ctx context.Context, arg database.UpdateCustomRoleParams) (database.CustomRole, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateCustomRole", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateCustomRole", ctx, arg) ret0, _ := ret[0].(database.CustomRole) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateCustomRole indicates an expected call of UpdateCustomRole. -func (mr *MockStoreMockRecorder) UpdateCustomRole(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateCustomRole(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateCustomRole", reflect.TypeOf((*MockStore)(nil).UpdateCustomRole), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateCustomRole", reflect.TypeOf((*MockStore)(nil).UpdateCustomRole), ctx, arg) } // UpdateExternalAuthLink mocks base method. -func (m *MockStore) UpdateExternalAuthLink(arg0 context.Context, arg1 database.UpdateExternalAuthLinkParams) (database.ExternalAuthLink, error) { +func (m *MockStore) UpdateExternalAuthLink(ctx context.Context, arg database.UpdateExternalAuthLinkParams) (database.ExternalAuthLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateExternalAuthLink", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateExternalAuthLink", ctx, arg) ret0, _ := ret[0].(database.ExternalAuthLink) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateExternalAuthLink indicates an expected call of UpdateExternalAuthLink. -func (mr *MockStoreMockRecorder) UpdateExternalAuthLink(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateExternalAuthLink(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateExternalAuthLink", reflect.TypeOf((*MockStore)(nil).UpdateExternalAuthLink), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateExternalAuthLink", reflect.TypeOf((*MockStore)(nil).UpdateExternalAuthLink), ctx, arg) } // UpdateGitSSHKey mocks base method. -func (m *MockStore) UpdateGitSSHKey(arg0 context.Context, arg1 database.UpdateGitSSHKeyParams) (database.GitSSHKey, error) { +func (m *MockStore) UpdateGitSSHKey(ctx context.Context, arg database.UpdateGitSSHKeyParams) (database.GitSSHKey, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateGitSSHKey", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateGitSSHKey", ctx, arg) ret0, _ := ret[0].(database.GitSSHKey) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateGitSSHKey indicates an expected call of UpdateGitSSHKey. -func (mr *MockStoreMockRecorder) UpdateGitSSHKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateGitSSHKey(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateGitSSHKey", reflect.TypeOf((*MockStore)(nil).UpdateGitSSHKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateGitSSHKey", reflect.TypeOf((*MockStore)(nil).UpdateGitSSHKey), ctx, arg) } // UpdateGroupByID mocks base method. -func (m *MockStore) UpdateGroupByID(arg0 context.Context, arg1 database.UpdateGroupByIDParams) (database.Group, error) { +func (m *MockStore) UpdateGroupByID(ctx context.Context, arg database.UpdateGroupByIDParams) (database.Group, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateGroupByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateGroupByID", ctx, arg) ret0, _ := ret[0].(database.Group) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateGroupByID indicates an expected call of UpdateGroupByID. -func (mr *MockStoreMockRecorder) UpdateGroupByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateGroupByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateGroupByID", reflect.TypeOf((*MockStore)(nil).UpdateGroupByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateGroupByID", reflect.TypeOf((*MockStore)(nil).UpdateGroupByID), ctx, arg) } // UpdateInactiveUsersToDormant mocks base method. -func (m *MockStore) UpdateInactiveUsersToDormant(arg0 context.Context, arg1 database.UpdateInactiveUsersToDormantParams) ([]database.UpdateInactiveUsersToDormantRow, error) { +func (m *MockStore) UpdateInactiveUsersToDormant(ctx context.Context, arg database.UpdateInactiveUsersToDormantParams) ([]database.UpdateInactiveUsersToDormantRow, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateInactiveUsersToDormant", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateInactiveUsersToDormant", ctx, arg) ret0, _ := ret[0].([]database.UpdateInactiveUsersToDormantRow) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateInactiveUsersToDormant indicates an expected call of UpdateInactiveUsersToDormant. -func (mr *MockStoreMockRecorder) UpdateInactiveUsersToDormant(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateInactiveUsersToDormant(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateInactiveUsersToDormant", reflect.TypeOf((*MockStore)(nil).UpdateInactiveUsersToDormant), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateInactiveUsersToDormant", reflect.TypeOf((*MockStore)(nil).UpdateInactiveUsersToDormant), ctx, arg) } // UpdateMemberRoles mocks base method. -func (m *MockStore) UpdateMemberRoles(arg0 context.Context, arg1 database.UpdateMemberRolesParams) (database.OrganizationMember, error) { +func (m *MockStore) UpdateMemberRoles(ctx context.Context, arg database.UpdateMemberRolesParams) (database.OrganizationMember, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateMemberRoles", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateMemberRoles", ctx, arg) ret0, _ := ret[0].(database.OrganizationMember) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateMemberRoles indicates an expected call of UpdateMemberRoles. -func (mr *MockStoreMockRecorder) UpdateMemberRoles(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateMemberRoles(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateMemberRoles", reflect.TypeOf((*MockStore)(nil).UpdateMemberRoles), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateMemberRoles", reflect.TypeOf((*MockStore)(nil).UpdateMemberRoles), ctx, arg) } // UpdateNotificationTemplateMethodByID mocks base method. -func (m *MockStore) UpdateNotificationTemplateMethodByID(arg0 context.Context, arg1 database.UpdateNotificationTemplateMethodByIDParams) (database.NotificationTemplate, error) { +func (m *MockStore) UpdateNotificationTemplateMethodByID(ctx context.Context, arg database.UpdateNotificationTemplateMethodByIDParams) (database.NotificationTemplate, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateNotificationTemplateMethodByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateNotificationTemplateMethodByID", ctx, arg) ret0, _ := ret[0].(database.NotificationTemplate) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateNotificationTemplateMethodByID indicates an expected call of UpdateNotificationTemplateMethodByID. -func (mr *MockStoreMockRecorder) UpdateNotificationTemplateMethodByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateNotificationTemplateMethodByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNotificationTemplateMethodByID", reflect.TypeOf((*MockStore)(nil).UpdateNotificationTemplateMethodByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateNotificationTemplateMethodByID", reflect.TypeOf((*MockStore)(nil).UpdateNotificationTemplateMethodByID), ctx, arg) } // UpdateOAuth2ProviderAppByID mocks base method. -func (m *MockStore) UpdateOAuth2ProviderAppByID(arg0 context.Context, arg1 database.UpdateOAuth2ProviderAppByIDParams) (database.OAuth2ProviderApp, error) { +func (m *MockStore) UpdateOAuth2ProviderAppByID(ctx context.Context, arg database.UpdateOAuth2ProviderAppByIDParams) (database.OAuth2ProviderApp, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateOAuth2ProviderAppByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateOAuth2ProviderAppByID", ctx, arg) ret0, _ := ret[0].(database.OAuth2ProviderApp) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateOAuth2ProviderAppByID indicates an expected call of UpdateOAuth2ProviderAppByID. -func (mr *MockStoreMockRecorder) UpdateOAuth2ProviderAppByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateOAuth2ProviderAppByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOAuth2ProviderAppByID", reflect.TypeOf((*MockStore)(nil).UpdateOAuth2ProviderAppByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOAuth2ProviderAppByID", reflect.TypeOf((*MockStore)(nil).UpdateOAuth2ProviderAppByID), ctx, arg) } // UpdateOAuth2ProviderAppSecretByID mocks base method. -func (m *MockStore) UpdateOAuth2ProviderAppSecretByID(arg0 context.Context, arg1 database.UpdateOAuth2ProviderAppSecretByIDParams) (database.OAuth2ProviderAppSecret, error) { +func (m *MockStore) UpdateOAuth2ProviderAppSecretByID(ctx context.Context, arg database.UpdateOAuth2ProviderAppSecretByIDParams) (database.OAuth2ProviderAppSecret, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateOAuth2ProviderAppSecretByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateOAuth2ProviderAppSecretByID", ctx, arg) ret0, _ := ret[0].(database.OAuth2ProviderAppSecret) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateOAuth2ProviderAppSecretByID indicates an expected call of UpdateOAuth2ProviderAppSecretByID. -func (mr *MockStoreMockRecorder) UpdateOAuth2ProviderAppSecretByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateOAuth2ProviderAppSecretByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOAuth2ProviderAppSecretByID", reflect.TypeOf((*MockStore)(nil).UpdateOAuth2ProviderAppSecretByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOAuth2ProviderAppSecretByID", reflect.TypeOf((*MockStore)(nil).UpdateOAuth2ProviderAppSecretByID), ctx, arg) } // UpdateOrganization mocks base method. -func (m *MockStore) UpdateOrganization(arg0 context.Context, arg1 database.UpdateOrganizationParams) (database.Organization, error) { +func (m *MockStore) UpdateOrganization(ctx context.Context, arg database.UpdateOrganizationParams) (database.Organization, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateOrganization", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateOrganization", ctx, arg) ret0, _ := ret[0].(database.Organization) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateOrganization indicates an expected call of UpdateOrganization. -func (mr *MockStoreMockRecorder) UpdateOrganization(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateOrganization(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOrganization", reflect.TypeOf((*MockStore)(nil).UpdateOrganization), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOrganization", reflect.TypeOf((*MockStore)(nil).UpdateOrganization), ctx, arg) } // UpdateProvisionerDaemonLastSeenAt mocks base method. -func (m *MockStore) UpdateProvisionerDaemonLastSeenAt(arg0 context.Context, arg1 database.UpdateProvisionerDaemonLastSeenAtParams) error { +func (m *MockStore) UpdateProvisionerDaemonLastSeenAt(ctx context.Context, arg database.UpdateProvisionerDaemonLastSeenAtParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateProvisionerDaemonLastSeenAt", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateProvisionerDaemonLastSeenAt", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateProvisionerDaemonLastSeenAt indicates an expected call of UpdateProvisionerDaemonLastSeenAt. -func (mr *MockStoreMockRecorder) UpdateProvisionerDaemonLastSeenAt(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateProvisionerDaemonLastSeenAt(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProvisionerDaemonLastSeenAt", reflect.TypeOf((*MockStore)(nil).UpdateProvisionerDaemonLastSeenAt), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProvisionerDaemonLastSeenAt", reflect.TypeOf((*MockStore)(nil).UpdateProvisionerDaemonLastSeenAt), ctx, arg) } // UpdateProvisionerJobByID mocks base method. -func (m *MockStore) UpdateProvisionerJobByID(arg0 context.Context, arg1 database.UpdateProvisionerJobByIDParams) error { +func (m *MockStore) UpdateProvisionerJobByID(ctx context.Context, arg database.UpdateProvisionerJobByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateProvisionerJobByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateProvisionerJobByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateProvisionerJobByID indicates an expected call of UpdateProvisionerJobByID. -func (mr *MockStoreMockRecorder) UpdateProvisionerJobByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateProvisionerJobByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProvisionerJobByID", reflect.TypeOf((*MockStore)(nil).UpdateProvisionerJobByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProvisionerJobByID", reflect.TypeOf((*MockStore)(nil).UpdateProvisionerJobByID), ctx, arg) } // UpdateProvisionerJobWithCancelByID mocks base method. -func (m *MockStore) UpdateProvisionerJobWithCancelByID(arg0 context.Context, arg1 database.UpdateProvisionerJobWithCancelByIDParams) error { +func (m *MockStore) UpdateProvisionerJobWithCancelByID(ctx context.Context, arg database.UpdateProvisionerJobWithCancelByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateProvisionerJobWithCancelByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateProvisionerJobWithCancelByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateProvisionerJobWithCancelByID indicates an expected call of UpdateProvisionerJobWithCancelByID. -func (mr *MockStoreMockRecorder) UpdateProvisionerJobWithCancelByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateProvisionerJobWithCancelByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProvisionerJobWithCancelByID", reflect.TypeOf((*MockStore)(nil).UpdateProvisionerJobWithCancelByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProvisionerJobWithCancelByID", reflect.TypeOf((*MockStore)(nil).UpdateProvisionerJobWithCancelByID), ctx, arg) } // UpdateProvisionerJobWithCompleteByID mocks base method. -func (m *MockStore) UpdateProvisionerJobWithCompleteByID(arg0 context.Context, arg1 database.UpdateProvisionerJobWithCompleteByIDParams) error { +func (m *MockStore) UpdateProvisionerJobWithCompleteByID(ctx context.Context, arg database.UpdateProvisionerJobWithCompleteByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateProvisionerJobWithCompleteByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateProvisionerJobWithCompleteByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateProvisionerJobWithCompleteByID indicates an expected call of UpdateProvisionerJobWithCompleteByID. -func (mr *MockStoreMockRecorder) UpdateProvisionerJobWithCompleteByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateProvisionerJobWithCompleteByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProvisionerJobWithCompleteByID", reflect.TypeOf((*MockStore)(nil).UpdateProvisionerJobWithCompleteByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateProvisionerJobWithCompleteByID", reflect.TypeOf((*MockStore)(nil).UpdateProvisionerJobWithCompleteByID), ctx, arg) } // UpdateReplica mocks base method. -func (m *MockStore) UpdateReplica(arg0 context.Context, arg1 database.UpdateReplicaParams) (database.Replica, error) { +func (m *MockStore) UpdateReplica(ctx context.Context, arg database.UpdateReplicaParams) (database.Replica, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateReplica", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateReplica", ctx, arg) ret0, _ := ret[0].(database.Replica) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateReplica indicates an expected call of UpdateReplica. -func (mr *MockStoreMockRecorder) UpdateReplica(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateReplica(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateReplica", reflect.TypeOf((*MockStore)(nil).UpdateReplica), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateReplica", reflect.TypeOf((*MockStore)(nil).UpdateReplica), ctx, arg) } // UpdateTailnetPeerStatusByCoordinator mocks base method. -func (m *MockStore) UpdateTailnetPeerStatusByCoordinator(arg0 context.Context, arg1 database.UpdateTailnetPeerStatusByCoordinatorParams) error { +func (m *MockStore) UpdateTailnetPeerStatusByCoordinator(ctx context.Context, arg database.UpdateTailnetPeerStatusByCoordinatorParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTailnetPeerStatusByCoordinator", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTailnetPeerStatusByCoordinator", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTailnetPeerStatusByCoordinator indicates an expected call of UpdateTailnetPeerStatusByCoordinator. -func (mr *MockStoreMockRecorder) UpdateTailnetPeerStatusByCoordinator(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTailnetPeerStatusByCoordinator(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTailnetPeerStatusByCoordinator", reflect.TypeOf((*MockStore)(nil).UpdateTailnetPeerStatusByCoordinator), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTailnetPeerStatusByCoordinator", reflect.TypeOf((*MockStore)(nil).UpdateTailnetPeerStatusByCoordinator), ctx, arg) } // UpdateTemplateACLByID mocks base method. -func (m *MockStore) UpdateTemplateACLByID(arg0 context.Context, arg1 database.UpdateTemplateACLByIDParams) error { +func (m *MockStore) UpdateTemplateACLByID(ctx context.Context, arg database.UpdateTemplateACLByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateACLByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateACLByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateACLByID indicates an expected call of UpdateTemplateACLByID. -func (mr *MockStoreMockRecorder) UpdateTemplateACLByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateACLByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateACLByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateACLByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateACLByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateACLByID), ctx, arg) } // UpdateTemplateAccessControlByID mocks base method. -func (m *MockStore) UpdateTemplateAccessControlByID(arg0 context.Context, arg1 database.UpdateTemplateAccessControlByIDParams) error { +func (m *MockStore) UpdateTemplateAccessControlByID(ctx context.Context, arg database.UpdateTemplateAccessControlByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateAccessControlByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateAccessControlByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateAccessControlByID indicates an expected call of UpdateTemplateAccessControlByID. -func (mr *MockStoreMockRecorder) UpdateTemplateAccessControlByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateAccessControlByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateAccessControlByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateAccessControlByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateAccessControlByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateAccessControlByID), ctx, arg) } // UpdateTemplateActiveVersionByID mocks base method. -func (m *MockStore) UpdateTemplateActiveVersionByID(arg0 context.Context, arg1 database.UpdateTemplateActiveVersionByIDParams) error { +func (m *MockStore) UpdateTemplateActiveVersionByID(ctx context.Context, arg database.UpdateTemplateActiveVersionByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateActiveVersionByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateActiveVersionByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateActiveVersionByID indicates an expected call of UpdateTemplateActiveVersionByID. -func (mr *MockStoreMockRecorder) UpdateTemplateActiveVersionByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateActiveVersionByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateActiveVersionByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateActiveVersionByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateActiveVersionByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateActiveVersionByID), ctx, arg) } // UpdateTemplateDeletedByID mocks base method. -func (m *MockStore) UpdateTemplateDeletedByID(arg0 context.Context, arg1 database.UpdateTemplateDeletedByIDParams) error { +func (m *MockStore) UpdateTemplateDeletedByID(ctx context.Context, arg database.UpdateTemplateDeletedByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateDeletedByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateDeletedByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateDeletedByID indicates an expected call of UpdateTemplateDeletedByID. -func (mr *MockStoreMockRecorder) UpdateTemplateDeletedByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateDeletedByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateDeletedByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateDeletedByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateDeletedByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateDeletedByID), ctx, arg) } // UpdateTemplateMetaByID mocks base method. -func (m *MockStore) UpdateTemplateMetaByID(arg0 context.Context, arg1 database.UpdateTemplateMetaByIDParams) error { +func (m *MockStore) UpdateTemplateMetaByID(ctx context.Context, arg database.UpdateTemplateMetaByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateMetaByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateMetaByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateMetaByID indicates an expected call of UpdateTemplateMetaByID. -func (mr *MockStoreMockRecorder) UpdateTemplateMetaByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateMetaByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateMetaByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateMetaByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateMetaByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateMetaByID), ctx, arg) } // UpdateTemplateScheduleByID mocks base method. -func (m *MockStore) UpdateTemplateScheduleByID(arg0 context.Context, arg1 database.UpdateTemplateScheduleByIDParams) error { +func (m *MockStore) UpdateTemplateScheduleByID(ctx context.Context, arg database.UpdateTemplateScheduleByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateScheduleByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateScheduleByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateScheduleByID indicates an expected call of UpdateTemplateScheduleByID. -func (mr *MockStoreMockRecorder) UpdateTemplateScheduleByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateScheduleByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateScheduleByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateScheduleByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateScheduleByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateScheduleByID), ctx, arg) } // UpdateTemplateVersionByID mocks base method. -func (m *MockStore) UpdateTemplateVersionByID(arg0 context.Context, arg1 database.UpdateTemplateVersionByIDParams) error { +func (m *MockStore) UpdateTemplateVersionByID(ctx context.Context, arg database.UpdateTemplateVersionByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateVersionByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateVersionByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateVersionByID indicates an expected call of UpdateTemplateVersionByID. -func (mr *MockStoreMockRecorder) UpdateTemplateVersionByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateVersionByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateVersionByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateVersionByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateVersionByID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateVersionByID), ctx, arg) } // UpdateTemplateVersionDescriptionByJobID mocks base method. -func (m *MockStore) UpdateTemplateVersionDescriptionByJobID(arg0 context.Context, arg1 database.UpdateTemplateVersionDescriptionByJobIDParams) error { +func (m *MockStore) UpdateTemplateVersionDescriptionByJobID(ctx context.Context, arg database.UpdateTemplateVersionDescriptionByJobIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateVersionDescriptionByJobID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateVersionDescriptionByJobID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateVersionDescriptionByJobID indicates an expected call of UpdateTemplateVersionDescriptionByJobID. -func (mr *MockStoreMockRecorder) UpdateTemplateVersionDescriptionByJobID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateVersionDescriptionByJobID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateVersionDescriptionByJobID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateVersionDescriptionByJobID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateVersionDescriptionByJobID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateVersionDescriptionByJobID), ctx, arg) } // UpdateTemplateVersionExternalAuthProvidersByJobID mocks base method. -func (m *MockStore) UpdateTemplateVersionExternalAuthProvidersByJobID(arg0 context.Context, arg1 database.UpdateTemplateVersionExternalAuthProvidersByJobIDParams) error { +func (m *MockStore) UpdateTemplateVersionExternalAuthProvidersByJobID(ctx context.Context, arg database.UpdateTemplateVersionExternalAuthProvidersByJobIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateVersionExternalAuthProvidersByJobID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateVersionExternalAuthProvidersByJobID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateVersionExternalAuthProvidersByJobID indicates an expected call of UpdateTemplateVersionExternalAuthProvidersByJobID. -func (mr *MockStoreMockRecorder) UpdateTemplateVersionExternalAuthProvidersByJobID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateVersionExternalAuthProvidersByJobID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateVersionExternalAuthProvidersByJobID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateVersionExternalAuthProvidersByJobID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateVersionExternalAuthProvidersByJobID", reflect.TypeOf((*MockStore)(nil).UpdateTemplateVersionExternalAuthProvidersByJobID), ctx, arg) } // UpdateTemplateWorkspacesLastUsedAt mocks base method. -func (m *MockStore) UpdateTemplateWorkspacesLastUsedAt(arg0 context.Context, arg1 database.UpdateTemplateWorkspacesLastUsedAtParams) error { +func (m *MockStore) UpdateTemplateWorkspacesLastUsedAt(ctx context.Context, arg database.UpdateTemplateWorkspacesLastUsedAtParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateTemplateWorkspacesLastUsedAt", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateTemplateWorkspacesLastUsedAt", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateTemplateWorkspacesLastUsedAt indicates an expected call of UpdateTemplateWorkspacesLastUsedAt. -func (mr *MockStoreMockRecorder) UpdateTemplateWorkspacesLastUsedAt(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateTemplateWorkspacesLastUsedAt(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateWorkspacesLastUsedAt", reflect.TypeOf((*MockStore)(nil).UpdateTemplateWorkspacesLastUsedAt), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateTemplateWorkspacesLastUsedAt", reflect.TypeOf((*MockStore)(nil).UpdateTemplateWorkspacesLastUsedAt), ctx, arg) } // UpdateUserAppearanceSettings mocks base method. -func (m *MockStore) UpdateUserAppearanceSettings(arg0 context.Context, arg1 database.UpdateUserAppearanceSettingsParams) (database.User, error) { +func (m *MockStore) UpdateUserAppearanceSettings(ctx context.Context, arg database.UpdateUserAppearanceSettingsParams) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserAppearanceSettings", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserAppearanceSettings", ctx, arg) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserAppearanceSettings indicates an expected call of UpdateUserAppearanceSettings. -func (mr *MockStoreMockRecorder) UpdateUserAppearanceSettings(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserAppearanceSettings(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserAppearanceSettings", reflect.TypeOf((*MockStore)(nil).UpdateUserAppearanceSettings), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserAppearanceSettings", reflect.TypeOf((*MockStore)(nil).UpdateUserAppearanceSettings), ctx, arg) } // UpdateUserDeletedByID mocks base method. -func (m *MockStore) UpdateUserDeletedByID(arg0 context.Context, arg1 uuid.UUID) error { +func (m *MockStore) UpdateUserDeletedByID(ctx context.Context, id uuid.UUID) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserDeletedByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserDeletedByID", ctx, id) ret0, _ := ret[0].(error) return ret0 } // UpdateUserDeletedByID indicates an expected call of UpdateUserDeletedByID. -func (mr *MockStoreMockRecorder) UpdateUserDeletedByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserDeletedByID(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserDeletedByID", reflect.TypeOf((*MockStore)(nil).UpdateUserDeletedByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserDeletedByID", reflect.TypeOf((*MockStore)(nil).UpdateUserDeletedByID), ctx, id) } // UpdateUserGithubComUserID mocks base method. -func (m *MockStore) UpdateUserGithubComUserID(arg0 context.Context, arg1 database.UpdateUserGithubComUserIDParams) error { +func (m *MockStore) UpdateUserGithubComUserID(ctx context.Context, arg database.UpdateUserGithubComUserIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserGithubComUserID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserGithubComUserID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateUserGithubComUserID indicates an expected call of UpdateUserGithubComUserID. -func (mr *MockStoreMockRecorder) UpdateUserGithubComUserID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserGithubComUserID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserGithubComUserID", reflect.TypeOf((*MockStore)(nil).UpdateUserGithubComUserID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserGithubComUserID", reflect.TypeOf((*MockStore)(nil).UpdateUserGithubComUserID), ctx, arg) } // UpdateUserHashedOneTimePasscode mocks base method. -func (m *MockStore) UpdateUserHashedOneTimePasscode(arg0 context.Context, arg1 database.UpdateUserHashedOneTimePasscodeParams) error { +func (m *MockStore) UpdateUserHashedOneTimePasscode(ctx context.Context, arg database.UpdateUserHashedOneTimePasscodeParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserHashedOneTimePasscode", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserHashedOneTimePasscode", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateUserHashedOneTimePasscode indicates an expected call of UpdateUserHashedOneTimePasscode. -func (mr *MockStoreMockRecorder) UpdateUserHashedOneTimePasscode(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserHashedOneTimePasscode(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserHashedOneTimePasscode", reflect.TypeOf((*MockStore)(nil).UpdateUserHashedOneTimePasscode), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserHashedOneTimePasscode", reflect.TypeOf((*MockStore)(nil).UpdateUserHashedOneTimePasscode), ctx, arg) } // UpdateUserHashedPassword mocks base method. -func (m *MockStore) UpdateUserHashedPassword(arg0 context.Context, arg1 database.UpdateUserHashedPasswordParams) error { +func (m *MockStore) UpdateUserHashedPassword(ctx context.Context, arg database.UpdateUserHashedPasswordParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserHashedPassword", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserHashedPassword", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateUserHashedPassword indicates an expected call of UpdateUserHashedPassword. -func (mr *MockStoreMockRecorder) UpdateUserHashedPassword(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserHashedPassword(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserHashedPassword", reflect.TypeOf((*MockStore)(nil).UpdateUserHashedPassword), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserHashedPassword", reflect.TypeOf((*MockStore)(nil).UpdateUserHashedPassword), ctx, arg) } // UpdateUserLastSeenAt mocks base method. -func (m *MockStore) UpdateUserLastSeenAt(arg0 context.Context, arg1 database.UpdateUserLastSeenAtParams) (database.User, error) { +func (m *MockStore) UpdateUserLastSeenAt(ctx context.Context, arg database.UpdateUserLastSeenAtParams) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserLastSeenAt", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserLastSeenAt", ctx, arg) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserLastSeenAt indicates an expected call of UpdateUserLastSeenAt. -func (mr *MockStoreMockRecorder) UpdateUserLastSeenAt(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserLastSeenAt(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLastSeenAt", reflect.TypeOf((*MockStore)(nil).UpdateUserLastSeenAt), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLastSeenAt", reflect.TypeOf((*MockStore)(nil).UpdateUserLastSeenAt), ctx, arg) } // UpdateUserLink mocks base method. -func (m *MockStore) UpdateUserLink(arg0 context.Context, arg1 database.UpdateUserLinkParams) (database.UserLink, error) { +func (m *MockStore) UpdateUserLink(ctx context.Context, arg database.UpdateUserLinkParams) (database.UserLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserLink", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserLink", ctx, arg) ret0, _ := ret[0].(database.UserLink) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserLink indicates an expected call of UpdateUserLink. -func (mr *MockStoreMockRecorder) UpdateUserLink(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserLink(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLink", reflect.TypeOf((*MockStore)(nil).UpdateUserLink), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLink", reflect.TypeOf((*MockStore)(nil).UpdateUserLink), ctx, arg) } // UpdateUserLinkedID mocks base method. -func (m *MockStore) UpdateUserLinkedID(arg0 context.Context, arg1 database.UpdateUserLinkedIDParams) (database.UserLink, error) { +func (m *MockStore) UpdateUserLinkedID(ctx context.Context, arg database.UpdateUserLinkedIDParams) (database.UserLink, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserLinkedID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserLinkedID", ctx, arg) ret0, _ := ret[0].(database.UserLink) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserLinkedID indicates an expected call of UpdateUserLinkedID. -func (mr *MockStoreMockRecorder) UpdateUserLinkedID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserLinkedID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLinkedID", reflect.TypeOf((*MockStore)(nil).UpdateUserLinkedID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLinkedID", reflect.TypeOf((*MockStore)(nil).UpdateUserLinkedID), ctx, arg) } // UpdateUserLoginType mocks base method. -func (m *MockStore) UpdateUserLoginType(arg0 context.Context, arg1 database.UpdateUserLoginTypeParams) (database.User, error) { +func (m *MockStore) UpdateUserLoginType(ctx context.Context, arg database.UpdateUserLoginTypeParams) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserLoginType", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserLoginType", ctx, arg) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserLoginType indicates an expected call of UpdateUserLoginType. -func (mr *MockStoreMockRecorder) UpdateUserLoginType(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserLoginType(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLoginType", reflect.TypeOf((*MockStore)(nil).UpdateUserLoginType), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserLoginType", reflect.TypeOf((*MockStore)(nil).UpdateUserLoginType), ctx, arg) } // UpdateUserNotificationPreferences mocks base method. -func (m *MockStore) UpdateUserNotificationPreferences(arg0 context.Context, arg1 database.UpdateUserNotificationPreferencesParams) (int64, error) { +func (m *MockStore) UpdateUserNotificationPreferences(ctx context.Context, arg database.UpdateUserNotificationPreferencesParams) (int64, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserNotificationPreferences", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserNotificationPreferences", ctx, arg) ret0, _ := ret[0].(int64) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserNotificationPreferences indicates an expected call of UpdateUserNotificationPreferences. -func (mr *MockStoreMockRecorder) UpdateUserNotificationPreferences(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserNotificationPreferences(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserNotificationPreferences", reflect.TypeOf((*MockStore)(nil).UpdateUserNotificationPreferences), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserNotificationPreferences", reflect.TypeOf((*MockStore)(nil).UpdateUserNotificationPreferences), ctx, arg) } // UpdateUserProfile mocks base method. -func (m *MockStore) UpdateUserProfile(arg0 context.Context, arg1 database.UpdateUserProfileParams) (database.User, error) { +func (m *MockStore) UpdateUserProfile(ctx context.Context, arg database.UpdateUserProfileParams) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserProfile", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserProfile", ctx, arg) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserProfile indicates an expected call of UpdateUserProfile. -func (mr *MockStoreMockRecorder) UpdateUserProfile(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserProfile(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserProfile", reflect.TypeOf((*MockStore)(nil).UpdateUserProfile), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserProfile", reflect.TypeOf((*MockStore)(nil).UpdateUserProfile), ctx, arg) } // UpdateUserQuietHoursSchedule mocks base method. -func (m *MockStore) UpdateUserQuietHoursSchedule(arg0 context.Context, arg1 database.UpdateUserQuietHoursScheduleParams) (database.User, error) { +func (m *MockStore) UpdateUserQuietHoursSchedule(ctx context.Context, arg database.UpdateUserQuietHoursScheduleParams) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserQuietHoursSchedule", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserQuietHoursSchedule", ctx, arg) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserQuietHoursSchedule indicates an expected call of UpdateUserQuietHoursSchedule. -func (mr *MockStoreMockRecorder) UpdateUserQuietHoursSchedule(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserQuietHoursSchedule(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserQuietHoursSchedule", reflect.TypeOf((*MockStore)(nil).UpdateUserQuietHoursSchedule), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserQuietHoursSchedule", reflect.TypeOf((*MockStore)(nil).UpdateUserQuietHoursSchedule), ctx, arg) } // UpdateUserRoles mocks base method. -func (m *MockStore) UpdateUserRoles(arg0 context.Context, arg1 database.UpdateUserRolesParams) (database.User, error) { +func (m *MockStore) UpdateUserRoles(ctx context.Context, arg database.UpdateUserRolesParams) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserRoles", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserRoles", ctx, arg) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserRoles indicates an expected call of UpdateUserRoles. -func (mr *MockStoreMockRecorder) UpdateUserRoles(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserRoles(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserRoles", reflect.TypeOf((*MockStore)(nil).UpdateUserRoles), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserRoles", reflect.TypeOf((*MockStore)(nil).UpdateUserRoles), ctx, arg) } // UpdateUserStatus mocks base method. -func (m *MockStore) UpdateUserStatus(arg0 context.Context, arg1 database.UpdateUserStatusParams) (database.User, error) { +func (m *MockStore) UpdateUserStatus(ctx context.Context, arg database.UpdateUserStatusParams) (database.User, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateUserStatus", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateUserStatus", ctx, arg) ret0, _ := ret[0].(database.User) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateUserStatus indicates an expected call of UpdateUserStatus. -func (mr *MockStoreMockRecorder) UpdateUserStatus(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateUserStatus(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserStatus", reflect.TypeOf((*MockStore)(nil).UpdateUserStatus), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateUserStatus", reflect.TypeOf((*MockStore)(nil).UpdateUserStatus), ctx, arg) } // UpdateWorkspace mocks base method. -func (m *MockStore) UpdateWorkspace(arg0 context.Context, arg1 database.UpdateWorkspaceParams) (database.WorkspaceTable, error) { +func (m *MockStore) UpdateWorkspace(ctx context.Context, arg database.UpdateWorkspaceParams) (database.WorkspaceTable, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspace", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspace", ctx, arg) ret0, _ := ret[0].(database.WorkspaceTable) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateWorkspace indicates an expected call of UpdateWorkspace. -func (mr *MockStoreMockRecorder) UpdateWorkspace(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspace(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspace", reflect.TypeOf((*MockStore)(nil).UpdateWorkspace), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspace", reflect.TypeOf((*MockStore)(nil).UpdateWorkspace), ctx, arg) } // UpdateWorkspaceAgentConnectionByID mocks base method. -func (m *MockStore) UpdateWorkspaceAgentConnectionByID(arg0 context.Context, arg1 database.UpdateWorkspaceAgentConnectionByIDParams) error { +func (m *MockStore) UpdateWorkspaceAgentConnectionByID(ctx context.Context, arg database.UpdateWorkspaceAgentConnectionByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceAgentConnectionByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceAgentConnectionByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceAgentConnectionByID indicates an expected call of UpdateWorkspaceAgentConnectionByID. -func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentConnectionByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentConnectionByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentConnectionByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentConnectionByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentConnectionByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentConnectionByID), ctx, arg) } // UpdateWorkspaceAgentLifecycleStateByID mocks base method. -func (m *MockStore) UpdateWorkspaceAgentLifecycleStateByID(arg0 context.Context, arg1 database.UpdateWorkspaceAgentLifecycleStateByIDParams) error { +func (m *MockStore) UpdateWorkspaceAgentLifecycleStateByID(ctx context.Context, arg database.UpdateWorkspaceAgentLifecycleStateByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceAgentLifecycleStateByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceAgentLifecycleStateByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceAgentLifecycleStateByID indicates an expected call of UpdateWorkspaceAgentLifecycleStateByID. -func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentLifecycleStateByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentLifecycleStateByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentLifecycleStateByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentLifecycleStateByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentLifecycleStateByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentLifecycleStateByID), ctx, arg) } // UpdateWorkspaceAgentLogOverflowByID mocks base method. -func (m *MockStore) UpdateWorkspaceAgentLogOverflowByID(arg0 context.Context, arg1 database.UpdateWorkspaceAgentLogOverflowByIDParams) error { +func (m *MockStore) UpdateWorkspaceAgentLogOverflowByID(ctx context.Context, arg database.UpdateWorkspaceAgentLogOverflowByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceAgentLogOverflowByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceAgentLogOverflowByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceAgentLogOverflowByID indicates an expected call of UpdateWorkspaceAgentLogOverflowByID. -func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentLogOverflowByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentLogOverflowByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentLogOverflowByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentLogOverflowByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentLogOverflowByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentLogOverflowByID), ctx, arg) } // UpdateWorkspaceAgentMetadata mocks base method. -func (m *MockStore) UpdateWorkspaceAgentMetadata(arg0 context.Context, arg1 database.UpdateWorkspaceAgentMetadataParams) error { +func (m *MockStore) UpdateWorkspaceAgentMetadata(ctx context.Context, arg database.UpdateWorkspaceAgentMetadataParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceAgentMetadata", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceAgentMetadata", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceAgentMetadata indicates an expected call of UpdateWorkspaceAgentMetadata. -func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentMetadata(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentMetadata(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentMetadata", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentMetadata), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentMetadata", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentMetadata), ctx, arg) } // UpdateWorkspaceAgentStartupByID mocks base method. -func (m *MockStore) UpdateWorkspaceAgentStartupByID(arg0 context.Context, arg1 database.UpdateWorkspaceAgentStartupByIDParams) error { +func (m *MockStore) UpdateWorkspaceAgentStartupByID(ctx context.Context, arg database.UpdateWorkspaceAgentStartupByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceAgentStartupByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceAgentStartupByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceAgentStartupByID indicates an expected call of UpdateWorkspaceAgentStartupByID. -func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentStartupByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceAgentStartupByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentStartupByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentStartupByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAgentStartupByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAgentStartupByID), ctx, arg) } // UpdateWorkspaceAppHealthByID mocks base method. -func (m *MockStore) UpdateWorkspaceAppHealthByID(arg0 context.Context, arg1 database.UpdateWorkspaceAppHealthByIDParams) error { +func (m *MockStore) UpdateWorkspaceAppHealthByID(ctx context.Context, arg database.UpdateWorkspaceAppHealthByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceAppHealthByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceAppHealthByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceAppHealthByID indicates an expected call of UpdateWorkspaceAppHealthByID. -func (mr *MockStoreMockRecorder) UpdateWorkspaceAppHealthByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceAppHealthByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAppHealthByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAppHealthByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAppHealthByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAppHealthByID), ctx, arg) } // UpdateWorkspaceAutomaticUpdates mocks base method. -func (m *MockStore) UpdateWorkspaceAutomaticUpdates(arg0 context.Context, arg1 database.UpdateWorkspaceAutomaticUpdatesParams) error { +func (m *MockStore) UpdateWorkspaceAutomaticUpdates(ctx context.Context, arg database.UpdateWorkspaceAutomaticUpdatesParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceAutomaticUpdates", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceAutomaticUpdates", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceAutomaticUpdates indicates an expected call of UpdateWorkspaceAutomaticUpdates. -func (mr *MockStoreMockRecorder) UpdateWorkspaceAutomaticUpdates(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceAutomaticUpdates(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAutomaticUpdates", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAutomaticUpdates), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAutomaticUpdates", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAutomaticUpdates), ctx, arg) } // UpdateWorkspaceAutostart mocks base method. -func (m *MockStore) UpdateWorkspaceAutostart(arg0 context.Context, arg1 database.UpdateWorkspaceAutostartParams) error { +func (m *MockStore) UpdateWorkspaceAutostart(ctx context.Context, arg database.UpdateWorkspaceAutostartParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceAutostart", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceAutostart", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceAutostart indicates an expected call of UpdateWorkspaceAutostart. -func (mr *MockStoreMockRecorder) UpdateWorkspaceAutostart(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceAutostart(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAutostart", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAutostart), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceAutostart", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceAutostart), ctx, arg) } // UpdateWorkspaceBuildCostByID mocks base method. -func (m *MockStore) UpdateWorkspaceBuildCostByID(arg0 context.Context, arg1 database.UpdateWorkspaceBuildCostByIDParams) error { +func (m *MockStore) UpdateWorkspaceBuildCostByID(ctx context.Context, arg database.UpdateWorkspaceBuildCostByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceBuildCostByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceBuildCostByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceBuildCostByID indicates an expected call of UpdateWorkspaceBuildCostByID. -func (mr *MockStoreMockRecorder) UpdateWorkspaceBuildCostByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceBuildCostByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceBuildCostByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceBuildCostByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceBuildCostByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceBuildCostByID), ctx, arg) } // UpdateWorkspaceBuildDeadlineByID mocks base method. -func (m *MockStore) UpdateWorkspaceBuildDeadlineByID(arg0 context.Context, arg1 database.UpdateWorkspaceBuildDeadlineByIDParams) error { +func (m *MockStore) UpdateWorkspaceBuildDeadlineByID(ctx context.Context, arg database.UpdateWorkspaceBuildDeadlineByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceBuildDeadlineByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceBuildDeadlineByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceBuildDeadlineByID indicates an expected call of UpdateWorkspaceBuildDeadlineByID. -func (mr *MockStoreMockRecorder) UpdateWorkspaceBuildDeadlineByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceBuildDeadlineByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceBuildDeadlineByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceBuildDeadlineByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceBuildDeadlineByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceBuildDeadlineByID), ctx, arg) } // UpdateWorkspaceBuildProvisionerStateByID mocks base method. -func (m *MockStore) UpdateWorkspaceBuildProvisionerStateByID(arg0 context.Context, arg1 database.UpdateWorkspaceBuildProvisionerStateByIDParams) error { +func (m *MockStore) UpdateWorkspaceBuildProvisionerStateByID(ctx context.Context, arg database.UpdateWorkspaceBuildProvisionerStateByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceBuildProvisionerStateByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceBuildProvisionerStateByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceBuildProvisionerStateByID indicates an expected call of UpdateWorkspaceBuildProvisionerStateByID. -func (mr *MockStoreMockRecorder) UpdateWorkspaceBuildProvisionerStateByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceBuildProvisionerStateByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceBuildProvisionerStateByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceBuildProvisionerStateByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceBuildProvisionerStateByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceBuildProvisionerStateByID), ctx, arg) } // UpdateWorkspaceDeletedByID mocks base method. -func (m *MockStore) UpdateWorkspaceDeletedByID(arg0 context.Context, arg1 database.UpdateWorkspaceDeletedByIDParams) error { +func (m *MockStore) UpdateWorkspaceDeletedByID(ctx context.Context, arg database.UpdateWorkspaceDeletedByIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceDeletedByID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceDeletedByID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceDeletedByID indicates an expected call of UpdateWorkspaceDeletedByID. -func (mr *MockStoreMockRecorder) UpdateWorkspaceDeletedByID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceDeletedByID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceDeletedByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceDeletedByID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceDeletedByID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceDeletedByID), ctx, arg) } // UpdateWorkspaceDormantDeletingAt mocks base method. -func (m *MockStore) UpdateWorkspaceDormantDeletingAt(arg0 context.Context, arg1 database.UpdateWorkspaceDormantDeletingAtParams) (database.WorkspaceTable, error) { +func (m *MockStore) UpdateWorkspaceDormantDeletingAt(ctx context.Context, arg database.UpdateWorkspaceDormantDeletingAtParams) (database.WorkspaceTable, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceDormantDeletingAt", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceDormantDeletingAt", ctx, arg) ret0, _ := ret[0].(database.WorkspaceTable) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateWorkspaceDormantDeletingAt indicates an expected call of UpdateWorkspaceDormantDeletingAt. -func (mr *MockStoreMockRecorder) UpdateWorkspaceDormantDeletingAt(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceDormantDeletingAt(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceDormantDeletingAt", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceDormantDeletingAt), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceDormantDeletingAt", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceDormantDeletingAt), ctx, arg) } // UpdateWorkspaceLastUsedAt mocks base method. -func (m *MockStore) UpdateWorkspaceLastUsedAt(arg0 context.Context, arg1 database.UpdateWorkspaceLastUsedAtParams) error { +func (m *MockStore) UpdateWorkspaceLastUsedAt(ctx context.Context, arg database.UpdateWorkspaceLastUsedAtParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceLastUsedAt", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceLastUsedAt", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceLastUsedAt indicates an expected call of UpdateWorkspaceLastUsedAt. -func (mr *MockStoreMockRecorder) UpdateWorkspaceLastUsedAt(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceLastUsedAt(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceLastUsedAt", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceLastUsedAt), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceLastUsedAt", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceLastUsedAt), ctx, arg) } // UpdateWorkspaceProxy mocks base method. -func (m *MockStore) UpdateWorkspaceProxy(arg0 context.Context, arg1 database.UpdateWorkspaceProxyParams) (database.WorkspaceProxy, error) { +func (m *MockStore) UpdateWorkspaceProxy(ctx context.Context, arg database.UpdateWorkspaceProxyParams) (database.WorkspaceProxy, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceProxy", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceProxy", ctx, arg) ret0, _ := ret[0].(database.WorkspaceProxy) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateWorkspaceProxy indicates an expected call of UpdateWorkspaceProxy. -func (mr *MockStoreMockRecorder) UpdateWorkspaceProxy(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceProxy(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceProxy", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceProxy), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceProxy", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceProxy), ctx, arg) } // UpdateWorkspaceProxyDeleted mocks base method. -func (m *MockStore) UpdateWorkspaceProxyDeleted(arg0 context.Context, arg1 database.UpdateWorkspaceProxyDeletedParams) error { +func (m *MockStore) UpdateWorkspaceProxyDeleted(ctx context.Context, arg database.UpdateWorkspaceProxyDeletedParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceProxyDeleted", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceProxyDeleted", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceProxyDeleted indicates an expected call of UpdateWorkspaceProxyDeleted. -func (mr *MockStoreMockRecorder) UpdateWorkspaceProxyDeleted(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceProxyDeleted(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceProxyDeleted", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceProxyDeleted), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceProxyDeleted", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceProxyDeleted), ctx, arg) } // UpdateWorkspaceTTL mocks base method. -func (m *MockStore) UpdateWorkspaceTTL(arg0 context.Context, arg1 database.UpdateWorkspaceTTLParams) error { +func (m *MockStore) UpdateWorkspaceTTL(ctx context.Context, arg database.UpdateWorkspaceTTLParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspaceTTL", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspaceTTL", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpdateWorkspaceTTL indicates an expected call of UpdateWorkspaceTTL. -func (mr *MockStoreMockRecorder) UpdateWorkspaceTTL(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspaceTTL(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceTTL", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceTTL), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspaceTTL", reflect.TypeOf((*MockStore)(nil).UpdateWorkspaceTTL), ctx, arg) } // UpdateWorkspacesDormantDeletingAtByTemplateID mocks base method. -func (m *MockStore) UpdateWorkspacesDormantDeletingAtByTemplateID(arg0 context.Context, arg1 database.UpdateWorkspacesDormantDeletingAtByTemplateIDParams) ([]database.WorkspaceTable, error) { +func (m *MockStore) UpdateWorkspacesDormantDeletingAtByTemplateID(ctx context.Context, arg database.UpdateWorkspacesDormantDeletingAtByTemplateIDParams) ([]database.WorkspaceTable, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpdateWorkspacesDormantDeletingAtByTemplateID", arg0, arg1) + ret := m.ctrl.Call(m, "UpdateWorkspacesDormantDeletingAtByTemplateID", ctx, arg) ret0, _ := ret[0].([]database.WorkspaceTable) ret1, _ := ret[1].(error) return ret0, ret1 } // UpdateWorkspacesDormantDeletingAtByTemplateID indicates an expected call of UpdateWorkspacesDormantDeletingAtByTemplateID. -func (mr *MockStoreMockRecorder) UpdateWorkspacesDormantDeletingAtByTemplateID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpdateWorkspacesDormantDeletingAtByTemplateID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspacesDormantDeletingAtByTemplateID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspacesDormantDeletingAtByTemplateID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkspacesDormantDeletingAtByTemplateID", reflect.TypeOf((*MockStore)(nil).UpdateWorkspacesDormantDeletingAtByTemplateID), ctx, arg) } // UpsertAnnouncementBanners mocks base method. -func (m *MockStore) UpsertAnnouncementBanners(arg0 context.Context, arg1 string) error { +func (m *MockStore) UpsertAnnouncementBanners(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertAnnouncementBanners", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertAnnouncementBanners", ctx, value) ret0, _ := ret[0].(error) return ret0 } // UpsertAnnouncementBanners indicates an expected call of UpsertAnnouncementBanners. -func (mr *MockStoreMockRecorder) UpsertAnnouncementBanners(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertAnnouncementBanners(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertAnnouncementBanners", reflect.TypeOf((*MockStore)(nil).UpsertAnnouncementBanners), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertAnnouncementBanners", reflect.TypeOf((*MockStore)(nil).UpsertAnnouncementBanners), ctx, value) } // UpsertAppSecurityKey mocks base method. -func (m *MockStore) UpsertAppSecurityKey(arg0 context.Context, arg1 string) error { +func (m *MockStore) UpsertAppSecurityKey(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertAppSecurityKey", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertAppSecurityKey", ctx, value) ret0, _ := ret[0].(error) return ret0 } // UpsertAppSecurityKey indicates an expected call of UpsertAppSecurityKey. -func (mr *MockStoreMockRecorder) UpsertAppSecurityKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertAppSecurityKey(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertAppSecurityKey", reflect.TypeOf((*MockStore)(nil).UpsertAppSecurityKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertAppSecurityKey", reflect.TypeOf((*MockStore)(nil).UpsertAppSecurityKey), ctx, value) } // UpsertApplicationName mocks base method. -func (m *MockStore) UpsertApplicationName(arg0 context.Context, arg1 string) error { +func (m *MockStore) UpsertApplicationName(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertApplicationName", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertApplicationName", ctx, value) ret0, _ := ret[0].(error) return ret0 } // UpsertApplicationName indicates an expected call of UpsertApplicationName. -func (mr *MockStoreMockRecorder) UpsertApplicationName(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertApplicationName(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertApplicationName", reflect.TypeOf((*MockStore)(nil).UpsertApplicationName), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertApplicationName", reflect.TypeOf((*MockStore)(nil).UpsertApplicationName), ctx, value) } // UpsertCoordinatorResumeTokenSigningKey mocks base method. -func (m *MockStore) UpsertCoordinatorResumeTokenSigningKey(arg0 context.Context, arg1 string) error { +func (m *MockStore) UpsertCoordinatorResumeTokenSigningKey(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertCoordinatorResumeTokenSigningKey", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertCoordinatorResumeTokenSigningKey", ctx, value) ret0, _ := ret[0].(error) return ret0 } // UpsertCoordinatorResumeTokenSigningKey indicates an expected call of UpsertCoordinatorResumeTokenSigningKey. -func (mr *MockStoreMockRecorder) UpsertCoordinatorResumeTokenSigningKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertCoordinatorResumeTokenSigningKey(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertCoordinatorResumeTokenSigningKey", reflect.TypeOf((*MockStore)(nil).UpsertCoordinatorResumeTokenSigningKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertCoordinatorResumeTokenSigningKey", reflect.TypeOf((*MockStore)(nil).UpsertCoordinatorResumeTokenSigningKey), ctx, value) } // UpsertDefaultProxy mocks base method. -func (m *MockStore) UpsertDefaultProxy(arg0 context.Context, arg1 database.UpsertDefaultProxyParams) error { +func (m *MockStore) UpsertDefaultProxy(ctx context.Context, arg database.UpsertDefaultProxyParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertDefaultProxy", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertDefaultProxy", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpsertDefaultProxy indicates an expected call of UpsertDefaultProxy. -func (mr *MockStoreMockRecorder) UpsertDefaultProxy(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertDefaultProxy(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertDefaultProxy", reflect.TypeOf((*MockStore)(nil).UpsertDefaultProxy), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertDefaultProxy", reflect.TypeOf((*MockStore)(nil).UpsertDefaultProxy), ctx, arg) } // UpsertHealthSettings mocks base method. -func (m *MockStore) UpsertHealthSettings(arg0 context.Context, arg1 string) error { +func (m *MockStore) UpsertHealthSettings(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertHealthSettings", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertHealthSettings", ctx, value) ret0, _ := ret[0].(error) return ret0 } // UpsertHealthSettings indicates an expected call of UpsertHealthSettings. -func (mr *MockStoreMockRecorder) UpsertHealthSettings(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertHealthSettings(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertHealthSettings", reflect.TypeOf((*MockStore)(nil).UpsertHealthSettings), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertHealthSettings", reflect.TypeOf((*MockStore)(nil).UpsertHealthSettings), ctx, value) } // UpsertJFrogXrayScanByWorkspaceAndAgentID mocks base method. -func (m *MockStore) UpsertJFrogXrayScanByWorkspaceAndAgentID(arg0 context.Context, arg1 database.UpsertJFrogXrayScanByWorkspaceAndAgentIDParams) error { +func (m *MockStore) UpsertJFrogXrayScanByWorkspaceAndAgentID(ctx context.Context, arg database.UpsertJFrogXrayScanByWorkspaceAndAgentIDParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertJFrogXrayScanByWorkspaceAndAgentID", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertJFrogXrayScanByWorkspaceAndAgentID", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpsertJFrogXrayScanByWorkspaceAndAgentID indicates an expected call of UpsertJFrogXrayScanByWorkspaceAndAgentID. -func (mr *MockStoreMockRecorder) UpsertJFrogXrayScanByWorkspaceAndAgentID(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertJFrogXrayScanByWorkspaceAndAgentID(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertJFrogXrayScanByWorkspaceAndAgentID", reflect.TypeOf((*MockStore)(nil).UpsertJFrogXrayScanByWorkspaceAndAgentID), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertJFrogXrayScanByWorkspaceAndAgentID", reflect.TypeOf((*MockStore)(nil).UpsertJFrogXrayScanByWorkspaceAndAgentID), ctx, arg) } // UpsertLastUpdateCheck mocks base method. -func (m *MockStore) UpsertLastUpdateCheck(arg0 context.Context, arg1 string) error { +func (m *MockStore) UpsertLastUpdateCheck(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertLastUpdateCheck", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertLastUpdateCheck", ctx, value) ret0, _ := ret[0].(error) return ret0 } // UpsertLastUpdateCheck indicates an expected call of UpsertLastUpdateCheck. -func (mr *MockStoreMockRecorder) UpsertLastUpdateCheck(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertLastUpdateCheck(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertLastUpdateCheck", reflect.TypeOf((*MockStore)(nil).UpsertLastUpdateCheck), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertLastUpdateCheck", reflect.TypeOf((*MockStore)(nil).UpsertLastUpdateCheck), ctx, value) } // UpsertLogoURL mocks base method. -func (m *MockStore) UpsertLogoURL(arg0 context.Context, arg1 string) error { +func (m *MockStore) UpsertLogoURL(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertLogoURL", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertLogoURL", ctx, value) ret0, _ := ret[0].(error) return ret0 } // UpsertLogoURL indicates an expected call of UpsertLogoURL. -func (mr *MockStoreMockRecorder) UpsertLogoURL(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertLogoURL(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertLogoURL", reflect.TypeOf((*MockStore)(nil).UpsertLogoURL), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertLogoURL", reflect.TypeOf((*MockStore)(nil).UpsertLogoURL), ctx, value) } // UpsertNotificationReportGeneratorLog mocks base method. -func (m *MockStore) UpsertNotificationReportGeneratorLog(arg0 context.Context, arg1 database.UpsertNotificationReportGeneratorLogParams) error { +func (m *MockStore) UpsertNotificationReportGeneratorLog(ctx context.Context, arg database.UpsertNotificationReportGeneratorLogParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertNotificationReportGeneratorLog", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertNotificationReportGeneratorLog", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpsertNotificationReportGeneratorLog indicates an expected call of UpsertNotificationReportGeneratorLog. -func (mr *MockStoreMockRecorder) UpsertNotificationReportGeneratorLog(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertNotificationReportGeneratorLog(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertNotificationReportGeneratorLog", reflect.TypeOf((*MockStore)(nil).UpsertNotificationReportGeneratorLog), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertNotificationReportGeneratorLog", reflect.TypeOf((*MockStore)(nil).UpsertNotificationReportGeneratorLog), ctx, arg) } // UpsertNotificationsSettings mocks base method. -func (m *MockStore) UpsertNotificationsSettings(arg0 context.Context, arg1 string) error { +func (m *MockStore) UpsertNotificationsSettings(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertNotificationsSettings", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertNotificationsSettings", ctx, value) ret0, _ := ret[0].(error) return ret0 } // UpsertNotificationsSettings indicates an expected call of UpsertNotificationsSettings. -func (mr *MockStoreMockRecorder) UpsertNotificationsSettings(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertNotificationsSettings(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertNotificationsSettings", reflect.TypeOf((*MockStore)(nil).UpsertNotificationsSettings), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertNotificationsSettings", reflect.TypeOf((*MockStore)(nil).UpsertNotificationsSettings), ctx, value) } // UpsertOAuthSigningKey mocks base method. -func (m *MockStore) UpsertOAuthSigningKey(arg0 context.Context, arg1 string) error { +func (m *MockStore) UpsertOAuthSigningKey(ctx context.Context, value string) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertOAuthSigningKey", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertOAuthSigningKey", ctx, value) ret0, _ := ret[0].(error) return ret0 } // UpsertOAuthSigningKey indicates an expected call of UpsertOAuthSigningKey. -func (mr *MockStoreMockRecorder) UpsertOAuthSigningKey(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertOAuthSigningKey(ctx, value any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertOAuthSigningKey", reflect.TypeOf((*MockStore)(nil).UpsertOAuthSigningKey), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertOAuthSigningKey", reflect.TypeOf((*MockStore)(nil).UpsertOAuthSigningKey), ctx, value) } // UpsertProvisionerDaemon mocks base method. -func (m *MockStore) UpsertProvisionerDaemon(arg0 context.Context, arg1 database.UpsertProvisionerDaemonParams) (database.ProvisionerDaemon, error) { +func (m *MockStore) UpsertProvisionerDaemon(ctx context.Context, arg database.UpsertProvisionerDaemonParams) (database.ProvisionerDaemon, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertProvisionerDaemon", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertProvisionerDaemon", ctx, arg) ret0, _ := ret[0].(database.ProvisionerDaemon) ret1, _ := ret[1].(error) return ret0, ret1 } // UpsertProvisionerDaemon indicates an expected call of UpsertProvisionerDaemon. -func (mr *MockStoreMockRecorder) UpsertProvisionerDaemon(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertProvisionerDaemon(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertProvisionerDaemon", reflect.TypeOf((*MockStore)(nil).UpsertProvisionerDaemon), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertProvisionerDaemon", reflect.TypeOf((*MockStore)(nil).UpsertProvisionerDaemon), ctx, arg) } // UpsertRuntimeConfig mocks base method. -func (m *MockStore) UpsertRuntimeConfig(arg0 context.Context, arg1 database.UpsertRuntimeConfigParams) error { +func (m *MockStore) UpsertRuntimeConfig(ctx context.Context, arg database.UpsertRuntimeConfigParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertRuntimeConfig", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertRuntimeConfig", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpsertRuntimeConfig indicates an expected call of UpsertRuntimeConfig. -func (mr *MockStoreMockRecorder) UpsertRuntimeConfig(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertRuntimeConfig(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertRuntimeConfig", reflect.TypeOf((*MockStore)(nil).UpsertRuntimeConfig), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertRuntimeConfig", reflect.TypeOf((*MockStore)(nil).UpsertRuntimeConfig), ctx, arg) } // UpsertTailnetAgent mocks base method. -func (m *MockStore) UpsertTailnetAgent(arg0 context.Context, arg1 database.UpsertTailnetAgentParams) (database.TailnetAgent, error) { +func (m *MockStore) UpsertTailnetAgent(ctx context.Context, arg database.UpsertTailnetAgentParams) (database.TailnetAgent, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertTailnetAgent", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertTailnetAgent", ctx, arg) ret0, _ := ret[0].(database.TailnetAgent) ret1, _ := ret[1].(error) return ret0, ret1 } // UpsertTailnetAgent indicates an expected call of UpsertTailnetAgent. -func (mr *MockStoreMockRecorder) UpsertTailnetAgent(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertTailnetAgent(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetAgent", reflect.TypeOf((*MockStore)(nil).UpsertTailnetAgent), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetAgent", reflect.TypeOf((*MockStore)(nil).UpsertTailnetAgent), ctx, arg) } // UpsertTailnetClient mocks base method. -func (m *MockStore) UpsertTailnetClient(arg0 context.Context, arg1 database.UpsertTailnetClientParams) (database.TailnetClient, error) { +func (m *MockStore) UpsertTailnetClient(ctx context.Context, arg database.UpsertTailnetClientParams) (database.TailnetClient, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertTailnetClient", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertTailnetClient", ctx, arg) ret0, _ := ret[0].(database.TailnetClient) ret1, _ := ret[1].(error) return ret0, ret1 } // UpsertTailnetClient indicates an expected call of UpsertTailnetClient. -func (mr *MockStoreMockRecorder) UpsertTailnetClient(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertTailnetClient(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetClient", reflect.TypeOf((*MockStore)(nil).UpsertTailnetClient), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetClient", reflect.TypeOf((*MockStore)(nil).UpsertTailnetClient), ctx, arg) } // UpsertTailnetClientSubscription mocks base method. -func (m *MockStore) UpsertTailnetClientSubscription(arg0 context.Context, arg1 database.UpsertTailnetClientSubscriptionParams) error { +func (m *MockStore) UpsertTailnetClientSubscription(ctx context.Context, arg database.UpsertTailnetClientSubscriptionParams) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertTailnetClientSubscription", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertTailnetClientSubscription", ctx, arg) ret0, _ := ret[0].(error) return ret0 } // UpsertTailnetClientSubscription indicates an expected call of UpsertTailnetClientSubscription. -func (mr *MockStoreMockRecorder) UpsertTailnetClientSubscription(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertTailnetClientSubscription(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetClientSubscription", reflect.TypeOf((*MockStore)(nil).UpsertTailnetClientSubscription), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetClientSubscription", reflect.TypeOf((*MockStore)(nil).UpsertTailnetClientSubscription), ctx, arg) } // UpsertTailnetCoordinator mocks base method. -func (m *MockStore) UpsertTailnetCoordinator(arg0 context.Context, arg1 uuid.UUID) (database.TailnetCoordinator, error) { +func (m *MockStore) UpsertTailnetCoordinator(ctx context.Context, id uuid.UUID) (database.TailnetCoordinator, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertTailnetCoordinator", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertTailnetCoordinator", ctx, id) ret0, _ := ret[0].(database.TailnetCoordinator) ret1, _ := ret[1].(error) return ret0, ret1 } // UpsertTailnetCoordinator indicates an expected call of UpsertTailnetCoordinator. -func (mr *MockStoreMockRecorder) UpsertTailnetCoordinator(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertTailnetCoordinator(ctx, id any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetCoordinator", reflect.TypeOf((*MockStore)(nil).UpsertTailnetCoordinator), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetCoordinator", reflect.TypeOf((*MockStore)(nil).UpsertTailnetCoordinator), ctx, id) } // UpsertTailnetPeer mocks base method. -func (m *MockStore) UpsertTailnetPeer(arg0 context.Context, arg1 database.UpsertTailnetPeerParams) (database.TailnetPeer, error) { +func (m *MockStore) UpsertTailnetPeer(ctx context.Context, arg database.UpsertTailnetPeerParams) (database.TailnetPeer, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertTailnetPeer", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertTailnetPeer", ctx, arg) ret0, _ := ret[0].(database.TailnetPeer) ret1, _ := ret[1].(error) return ret0, ret1 } // UpsertTailnetPeer indicates an expected call of UpsertTailnetPeer. -func (mr *MockStoreMockRecorder) UpsertTailnetPeer(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertTailnetPeer(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetPeer", reflect.TypeOf((*MockStore)(nil).UpsertTailnetPeer), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetPeer", reflect.TypeOf((*MockStore)(nil).UpsertTailnetPeer), ctx, arg) } // UpsertTailnetTunnel mocks base method. -func (m *MockStore) UpsertTailnetTunnel(arg0 context.Context, arg1 database.UpsertTailnetTunnelParams) (database.TailnetTunnel, error) { +func (m *MockStore) UpsertTailnetTunnel(ctx context.Context, arg database.UpsertTailnetTunnelParams) (database.TailnetTunnel, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertTailnetTunnel", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertTailnetTunnel", ctx, arg) ret0, _ := ret[0].(database.TailnetTunnel) ret1, _ := ret[1].(error) return ret0, ret1 } // UpsertTailnetTunnel indicates an expected call of UpsertTailnetTunnel. -func (mr *MockStoreMockRecorder) UpsertTailnetTunnel(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertTailnetTunnel(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetTunnel", reflect.TypeOf((*MockStore)(nil).UpsertTailnetTunnel), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTailnetTunnel", reflect.TypeOf((*MockStore)(nil).UpsertTailnetTunnel), ctx, arg) } // UpsertTemplateUsageStats mocks base method. -func (m *MockStore) UpsertTemplateUsageStats(arg0 context.Context) error { +func (m *MockStore) UpsertTemplateUsageStats(ctx context.Context) error { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertTemplateUsageStats", arg0) + ret := m.ctrl.Call(m, "UpsertTemplateUsageStats", ctx) ret0, _ := ret[0].(error) return ret0 } // UpsertTemplateUsageStats indicates an expected call of UpsertTemplateUsageStats. -func (mr *MockStoreMockRecorder) UpsertTemplateUsageStats(arg0 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertTemplateUsageStats(ctx any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTemplateUsageStats", reflect.TypeOf((*MockStore)(nil).UpsertTemplateUsageStats), arg0) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertTemplateUsageStats", reflect.TypeOf((*MockStore)(nil).UpsertTemplateUsageStats), ctx) } // UpsertWorkspaceAgentPortShare mocks base method. -func (m *MockStore) UpsertWorkspaceAgentPortShare(arg0 context.Context, arg1 database.UpsertWorkspaceAgentPortShareParams) (database.WorkspaceAgentPortShare, error) { +func (m *MockStore) UpsertWorkspaceAgentPortShare(ctx context.Context, arg database.UpsertWorkspaceAgentPortShareParams) (database.WorkspaceAgentPortShare, error) { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "UpsertWorkspaceAgentPortShare", arg0, arg1) + ret := m.ctrl.Call(m, "UpsertWorkspaceAgentPortShare", ctx, arg) ret0, _ := ret[0].(database.WorkspaceAgentPortShare) ret1, _ := ret[1].(error) return ret0, ret1 } // UpsertWorkspaceAgentPortShare indicates an expected call of UpsertWorkspaceAgentPortShare. -func (mr *MockStoreMockRecorder) UpsertWorkspaceAgentPortShare(arg0, arg1 any) *gomock.Call { +func (mr *MockStoreMockRecorder) UpsertWorkspaceAgentPortShare(ctx, arg any) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertWorkspaceAgentPortShare", reflect.TypeOf((*MockStore)(nil).UpsertWorkspaceAgentPortShare), arg0, arg1) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpsertWorkspaceAgentPortShare", reflect.TypeOf((*MockStore)(nil).UpsertWorkspaceAgentPortShare), ctx, arg) } // Wrappers mocks base method. diff --git a/coderd/database/models.go b/coderd/database/models.go index af0a3122f7964..d1c5c042edfd3 100644 --- a/coderd/database/models.go +++ b/coderd/database/models.go @@ -1,6 +1,6 @@ // Code generated by sqlc. DO NOT EDIT. // versions: -// sqlc v1.25.0 +// sqlc v1.27.0 package database diff --git a/coderd/database/querier.go b/coderd/database/querier.go index ba4c06ce1007a..6109eda137372 100644 --- a/coderd/database/querier.go +++ b/coderd/database/querier.go @@ -1,6 +1,6 @@ // Code generated by sqlc. DO NOT EDIT. // versions: -// sqlc v1.25.0 +// sqlc v1.27.0 package database @@ -115,10 +115,6 @@ type sqlcQuerier interface { GetAPIKeysByLoginType(ctx context.Context, loginType LoginType) ([]APIKey, error) GetAPIKeysByUserID(ctx context.Context, arg GetAPIKeysByUserIDParams) ([]APIKey, error) GetAPIKeysLastUsedAfter(ctx context.Context, lastUsed time.Time) ([]APIKey, error) - // GetAccumulatedUsersInsights returns the accumulated number of users created - // in the given timeframe. It returns the accumulated number of users for each date - // within the specified timeframe, providing a running total of user sign-ups. - GetAccumulatedUsersInsights(ctx context.Context, arg GetAccumulatedUsersInsightsParams) ([]GetAccumulatedUsersInsightsRow, error) GetActiveUserCount(ctx context.Context) (int64, error) GetActiveWorkspaceBuildsByTemplateID(ctx context.Context, templateID uuid.UUID) ([]WorkspaceBuild, error) GetAllTailnetAgents(ctx context.Context) ([]TailnetAgent, error) diff --git a/coderd/database/queries.sql.go b/coderd/database/queries.sql.go index becd4c4b4f6eb..adfda230d7a9a 100644 --- a/coderd/database/queries.sql.go +++ b/coderd/database/queries.sql.go @@ -1,6 +1,6 @@ // Code generated by sqlc. DO NOT EDIT. // versions: -// sqlc v1.25.0 +// sqlc v1.27.0 package database @@ -2154,60 +2154,6 @@ func (q *sqlQuerier) UpdateGroupByID(ctx context.Context, arg UpdateGroupByIDPar return i, err } -const getAccumulatedUsersInsights = `-- name: GetAccumulatedUsersInsights :many -WITH RECURSIVE date_series AS ( - SELECT $2::timestamptz AS date - UNION ALL - SELECT date + INTERVAL '1 day' - FROM date_series - WHERE date + INTERVAL '1 day' <= $3::timestamptz -) -SELECT - (d.date AT TIME ZONE $1::text)::date AS date, - (SELECT COUNT(*) FROM users u WHERE u.created_at <= d.date) AS total -FROM - date_series d -ORDER BY - d.date -` - -type GetAccumulatedUsersInsightsParams struct { - Timezone string `db:"timezone" json:"timezone"` - StartTime time.Time `db:"start_time" json:"start_time"` - EndTime time.Time `db:"end_time" json:"end_time"` -} - -type GetAccumulatedUsersInsightsRow struct { - Date time.Time `db:"date" json:"date"` - Total int64 `db:"total" json:"total"` -} - -// GetAccumulatedUsersInsights returns the accumulated number of users created -// in the given timeframe. It returns the accumulated number of users for each date -// within the specified timeframe, providing a running total of user sign-ups. -func (q *sqlQuerier) GetAccumulatedUsersInsights(ctx context.Context, arg GetAccumulatedUsersInsightsParams) ([]GetAccumulatedUsersInsightsRow, error) { - rows, err := q.db.QueryContext(ctx, getAccumulatedUsersInsights, arg.Timezone, arg.StartTime, arg.EndTime) - if err != nil { - return nil, err - } - defer rows.Close() - var items []GetAccumulatedUsersInsightsRow - for rows.Next() { - var i GetAccumulatedUsersInsightsRow - if err := rows.Scan(&i.Date, &i.Total); err != nil { - return nil, err - } - items = append(items, i) - } - if err := rows.Close(); err != nil { - return nil, err - } - if err := rows.Err(); err != nil { - return nil, err - } - return items, nil -} - const getTemplateAppInsights = `-- name: GetTemplateAppInsights :many WITH -- Create a list of all unique apps by template, this is used to diff --git a/coderd/database/queries/insights.sql b/coderd/database/queries/insights.sql index 2256373071575..de107bc0e80c7 100644 --- a/coderd/database/queries/insights.sql +++ b/coderd/database/queries/insights.sql @@ -771,22 +771,3 @@ SELECT FROM unique_template_params utp JOIN workspace_build_parameters wbp ON (utp.workspace_build_ids @> ARRAY[wbp.workspace_build_id] AND utp.name = wbp.name) GROUP BY utp.num, utp.template_ids, utp.name, utp.type, utp.display_name, utp.description, utp.options, wbp.value; - --- name: GetAccumulatedUsersInsights :many --- GetAccumulatedUsersInsights returns the accumulated number of users created --- in the given timeframe. It returns the accumulated number of users for each date --- within the specified timeframe, providing a running total of user sign-ups. -WITH RECURSIVE date_series AS ( - SELECT @start_time::timestamptz AS date - UNION ALL - SELECT date + INTERVAL '1 day' - FROM date_series - WHERE date + INTERVAL '1 day' <= @end_time::timestamptz -) -SELECT - (d.date AT TIME ZONE @timezone::text)::date AS date, - (SELECT COUNT(*) FROM users u WHERE u.created_at <= d.date) AS total -FROM - date_series d -ORDER BY - d.date; diff --git a/coderd/insights.go b/coderd/insights.go index e70c362964e89..7234a88d44fe9 100644 --- a/coderd/insights.go +++ b/coderd/insights.go @@ -5,7 +5,6 @@ import ( "database/sql" "fmt" "net/http" - "strconv" "strings" "time" @@ -471,67 +470,6 @@ func (api *API) insightsTemplates(rw http.ResponseWriter, r *http.Request) { httpapi.Write(ctx, rw, http.StatusOK, resp) } -// @Summary Get insights about total users -// @ID get-insights-about-total-users -// @Security CoderSessionToken -// @Produce json -// @Tags Insights -// @Param start_time query string true "Start time" format(date-time) -// @Param end_time query string true "End time" format(date-time) -// @Success 200 {object} codersdk.TotalUsersInsightResponse -// @Router /insights/total-users [get] -func (api *API) insightsTotalUsers(rw http.ResponseWriter, r *http.Request) { - ctx := r.Context() - - p := httpapi.NewQueryParamParser(). - RequiredNotEmpty("start_time"). - RequiredNotEmpty("end_time") - vals := r.URL.Query() - var ( - // The QueryParamParser does not preserve timezone, so we need - // to parse the time ourselves. - startTimeString = p.String(vals, "", "start_time") - endTimeString = p.String(vals, "", "end_time") - ) - p.ErrorExcessParams(vals) - if len(p.Errors) > 0 { - httpapi.Write(ctx, rw, http.StatusBadRequest, codersdk.Response{ - Message: "Query parameters have invalid values.", - Validations: p.Errors, - }) - return - } - - startTime, endTime, ok := parseInsightsStartAndEndTime(ctx, rw, time.Now(), startTimeString, endTimeString) - if !ok { - return - } - - _, offset := startTime.Zone() - rows, err := api.Database.GetAccumulatedUsersInsights(ctx, database.GetAccumulatedUsersInsightsParams{ - StartTime: startTime, - EndTime: endTime, - Timezone: strconv.Itoa(offset), - }) - if err != nil { - httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{ - Message: "Internal error fetching total users.", - Detail: err.Error(), - }) - return - } - - var res codersdk.TotalUsersInsightResponse - for _, row := range rows { - res = append(res, codersdk.TotalUserByDate{ - Date: row.Date.Format(time.DateOnly), - Total: uint64(row.Total), - }) - } - - httpapi.Write(r.Context(), rw, http.StatusOK, res) -} - // convertTemplateInsightsApps builds the list of builtin apps and template apps // from the provided database rows, builtin apps are implicitly a part of all // templates. diff --git a/docs/reference/api/insights.md b/docs/reference/api/insights.md index 6c34ab97f612b..d9bb2327a9517 100644 --- a/docs/reference/api/insights.md +++ b/docs/reference/api/insights.md @@ -138,57 +138,6 @@ curl -X GET http://coder-server:8080/api/v2/insights/templates?start_time=2019-0 To perform this operation, you must be authenticated. [Learn more](authentication.md). -## Get insights about total users - -### Code samples - -```shell -# Example request using curl -curl -X GET http://coder-server:8080/api/v2/insights/total-users?start_time=2019-08-24T14%3A15%3A22Z&end_time=2019-08-24T14%3A15%3A22Z \ - -H 'Accept: application/json' \ - -H 'Coder-Session-Token: API_KEY' -``` - -`GET /insights/total-users` - -### Parameters - -| Name | In | Type | Required | Description | -| ------------ | ----- | ----------------- | -------- | ----------- | -| `start_time` | query | string(date-time) | true | Start time | -| `end_time` | query | string(date-time) | true | End time | - -### Example responses - -> 200 Response - -```json -[ - { - "date": "string", - "total": 0 - } -] -``` - -### Responses - -| Status | Meaning | Description | Schema | -| ------ | ------------------------------------------------------- | ----------- | ----------------------------------------------------------------------- | -| 200 | [OK](https://tools.ietf.org/html/rfc7231#section-6.3.1) | OK | array of [codersdk.TotalUserByDate](schemas.md#codersdktotaluserbydate) | - -

Response Schema

- -Status Code **200** - -| Name | Type | Required | Restrictions | Description | -| -------------- | ------- | -------- | ------------ | ----------- | -| `[array item]` | array | false | | | -| `» date` | string | false | | | -| `» total` | integer | false | | | - -To perform this operation, you must be authenticated. [Learn more](authentication.md). - ## Get insights about user activity ### Code samples diff --git a/docs/reference/api/schemas.md b/docs/reference/api/schemas.md index 93dbfd07701b9..c0f887a3771f8 100644 --- a/docs/reference/api/schemas.md +++ b/docs/reference/api/schemas.md @@ -5791,22 +5791,6 @@ CreateWorkspaceRequest provides options for creating a new workspace. Only one o | -------------------- | ------- | -------- | ------------ | ----------- | | `max_token_lifetime` | integer | false | | | -## codersdk.TotalUserByDate - -```json -{ - "date": "string", - "total": 0 -} -``` - -### Properties - -| Name | Type | Required | Restrictions | Description | -| ------- | ------- | -------- | ------------ | ----------- | -| `date` | string | false | | | -| `total` | integer | false | | | - ## codersdk.TraceConfig ```json From 6fb880c6d3680c81d5672b648018363733722cfa Mon Sep 17 00:00:00 2001 From: Sas Swart Date: Tue, 26 Nov 2024 18:12:31 +0000 Subject: [PATCH 19/19] Remove temporarily defunct sdk functions --- coderd/insights_test.go | 105 --------------------------------- coderd/workspaces.go | 2 + codersdk/insights.go | 29 --------- site/src/api/typesGenerated.ts | 15 ----- 4 files changed, 2 insertions(+), 149 deletions(-) diff --git a/coderd/insights_test.go b/coderd/insights_test.go index caadb05d98381..ef25a9dcf626c 100644 --- a/coderd/insights_test.go +++ b/coderd/insights_test.go @@ -2351,108 +2351,3 @@ func TestGenericInsights_RBAC(t *testing.T) { }) } } - -func TestTotalUsersInsight(t *testing.T) { - t.Parallel() - - t.Run("Success", func(t *testing.T) { - t.Parallel() - - client, db := coderdtest.NewWithDatabase(t, nil) - coderdtest.CreateFirstUser(t, client) - ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort) - t.Cleanup(cancel) - - // Given: a deployment with many users - dbgen.User(t, db, database.User{ - Email: "user1@coder.com", - Username: "user1", - CreatedAt: daysAgo(2), - }) - dbgen.User(t, db, database.User{ - Email: "user2@coder.com", - Username: "user2", - CreatedAt: daysAgo(2), - }) - dbgen.User(t, db, database.User{ - Email: "user3@coder.com", - Username: "user3", - CreatedAt: daysAgo(1), - }) - dbgen.User(t, db, database.User{ - Email: "user4@coder.com", - Username: "user4", - CreatedAt: daysAgo(0), - }) - - // When: requesting the accumulated total of users by day - now := time.Now().UTC().Truncate(time.Hour).Add(time.Hour) - res, err := client.TotalUsersInsight(ctx, codersdk.TotalUsersInsightRequest{ - StartTime: daysAgo(2), - EndTime: now, - }) - require.NoError(t, err) - - // Then: expect the correct number of dates and the growth of the accumulated total of users - require.Len(t, res, 3, "expect 3 dates of data") - - // First day - require.Equal(t, res[0].Date, daysAgo(2).Format(time.DateOnly)) - require.Equal(t, res[0].Total, uint64(2)) - - // Second day - require.Equal(t, res[1].Date, daysAgo(1).Format(time.DateOnly)) - require.Equal(t, res[1].Total, uint64(3)) - - // Third day - require.Equal(t, res[2].Date, daysAgo(0).Format(time.DateOnly)) - require.Equal(t, res[2].Total, uint64(5)) - }) - - t.Run("DatesWithNoUserRegistration", func(t *testing.T) { - t.Parallel() - - // Given: a deployment with no users getting created every day - client, db := coderdtest.NewWithDatabase(t, nil) - coderdtest.CreateFirstUser(t, client) - ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort) - t.Cleanup(cancel) - dbgen.User(t, db, database.User{ - Email: "user1@coder.com", - Username: "user1", - CreatedAt: daysAgo(2), - }) - dbgen.User(t, db, database.User{ - Email: "user2@coder.com", - Username: "user2", - CreatedAt: daysAgo(1), - }) - - // When: requesting the accumulated total of users for dates with no users created - now := time.Now().UTC().Truncate(time.Hour).Add(time.Hour) - res, err := client.TotalUsersInsight(ctx, codersdk.TotalUsersInsightRequest{ - StartTime: daysAgo(2).Truncate(24 * time.Hour), - EndTime: now, - }) - require.NoError(t, err) - - // Then: expect the correct number of dates. If there are no users created, the total should be 0. - require.Len(t, res, 3, "expect 3 dates of data") - - // First day - require.Equal(t, res[0].Date, daysAgo(2).Format(time.DateOnly)) - require.Equal(t, res[0].Total, uint64(1)) - - // Second day - no users created so should contain the same total as the previous day - require.Equal(t, res[1].Date, daysAgo(1).Format(time.DateOnly)) - require.Equal(t, res[1].Total, uint64(2)) - - // Third day - require.Equal(t, res[2].Date, daysAgo(0).Format(time.DateOnly)) - require.Equal(t, res[2].Total, uint64(3)) - }) -} - -func daysAgo(days int) time.Time { - return time.Now().UTC().AddDate(0, 0, -days).Truncate(24 * time.Hour) -} diff --git a/coderd/workspaces.go b/coderd/workspaces.go index ff8a55ded775a..333341195d355 100644 --- a/coderd/workspaces.go +++ b/coderd/workspaces.go @@ -517,6 +517,8 @@ func createWorkspace( // Update audit log's organization auditReq.UpdateOrganizationID(template.OrganizationID) + // TODO (SasSwart): api.Authorize below is already done above. This seems redundant. + // Should we remove this? // Do this upfront to save work. If this fails, the rest of the work // would be wasted. if !api.Authorize(r, policy.ActionCreate, diff --git a/codersdk/insights.go b/codersdk/insights.go index cb8c81a720bf4..c9e708de8f34a 100644 --- a/codersdk/insights.go +++ b/codersdk/insights.go @@ -282,32 +282,3 @@ func (c *Client) TemplateInsights(ctx context.Context, req TemplateInsightsReque var result TemplateInsightsResponse return result, json.NewDecoder(resp.Body).Decode(&result) } - -type TotalUsersInsightRequest struct { - StartTime time.Time `json:"start_time" format:"date-time"` - EndTime time.Time `json:"end_time" format:"date-time"` -} -type TotalUserByDate struct { - Date string `json:"date"` - Total uint64 `json:"total"` -} -type TotalUsersInsightResponse []TotalUserByDate - -func (c *Client) TotalUsersInsight(ctx context.Context, req TotalUsersInsightRequest) (TotalUsersInsightResponse, error) { - qp := url.Values{} - qp.Add("start_time", req.StartTime.Format(insightsTimeLayout)) - qp.Add("end_time", req.EndTime.Format(insightsTimeLayout)) - - reqURL := fmt.Sprintf("/api/v2/insights/total-users?%s", qp.Encode()) - resp, err := c.Request(ctx, http.MethodGet, reqURL, nil) - if err != nil { - return nil, xerrors.Errorf("make request: %w", err) - } - defer resp.Body.Close() - - if resp.StatusCode != http.StatusOK { - return nil, ReadBodyAsError(resp) - } - var result TotalUsersInsightResponse - return result, json.NewDecoder(resp.Body).Decode(&result) -} diff --git a/site/src/api/typesGenerated.ts b/site/src/api/typesGenerated.ts index 30eb7fa31920a..c1b409013b6d7 100644 --- a/site/src/api/typesGenerated.ts +++ b/site/src/api/typesGenerated.ts @@ -1540,21 +1540,6 @@ export interface TokensFilter { readonly include_all: boolean; } -// From codersdk/insights.go -export interface TotalUserByDate { - readonly date: string; - readonly total: number; -} - -// From codersdk/insights.go -export interface TotalUsersInsightRequest { - readonly start_time: string; - readonly end_time: string; -} - -// From codersdk/insights.go -export type TotalUsersInsightResponse = Readonly> - // From codersdk/deployment.go export interface TraceConfig { readonly enable: boolean; pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy