make gen impl

hugodutka · hugodutka · commit 3f1d43121e9b · 2025-02-22T16:03:31.000Z
diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go
@@ -1851,7 +1851,10 @@ func (q *querier) GetNotificationsSettings(ctx context.Context) (string, error)
 }
 
 func (q *querier) GetOAuth2GithubDefaultEligible(ctx context.Context) (bool, error) {
-	panic("not implemented")
+	if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceDeploymentConfig); err != nil {
+		return false, err
+	}
+	return q.db.GetOAuth2GithubDefaultEligible(ctx)
 }
 
 func (q *querier) GetOAuth2ProviderAppByID(ctx context.Context, id uuid.UUID) (database.OAuth2ProviderApp, error) {
@@ -4435,7 +4438,10 @@ func (q *querier) UpsertNotificationsSettings(ctx context.Context, value string)
 }
 
 func (q *querier) UpsertOAuth2GithubDefaultEligible(ctx context.Context, eligible bool) error {
-	panic("not implemented")
+	if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceDeploymentConfig); err != nil {
+		return err
+	}
+	return q.db.UpsertOAuth2GithubDefaultEligible(ctx, eligible)
 }
 
 func (q *querier) UpsertOAuthSigningKey(ctx context.Context, value string) error {
diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go
@@ -4404,6 +4404,12 @@ func (s *MethodTestSuite) TestSystemFunctions() {
 			Value: "value",
 		}).Asserts(rbac.ResourceSystem, policy.ActionUpdate)
 	}))
+	s.Run("GetOAuth2GithubDefaultEligible", s.Subtest(func(db database.Store, check *expects) {
+		check.Args().Asserts(rbac.ResourceDeploymentConfig, policy.ActionRead).Errors(sql.ErrNoRows)
+	}))
+	s.Run("UpsertOAuth2GithubDefaultEligible", s.Subtest(func(db database.Store, check *expects) {
+		check.Args(true).Asserts(rbac.ResourceDeploymentConfig, policy.ActionUpdate)
+	}))
 }
 
 func (s *MethodTestSuite) TestNotifications() {
diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
@@ -254,6 +254,7 @@ type data struct {
 	announcementBanners              []byte
 	healthSettings                   []byte
 	notificationsSettings            []byte
+	oauth2GithubDefaultEligible      *bool
 	applicationName                  string
 	logoURL                          string
 	appSecurityKey                   string
@@ -3528,8 +3529,14 @@ func (q *FakeQuerier) GetNotificationsSettings(_ context.Context) (string, error
 	return string(q.notificationsSettings), nil
 }
 
-func (q *FakeQuerier) GetOAuth2GithubDefaultEligible(ctx context.Context) (bool, error) {
-	panic("not implemented")
+func (q *FakeQuerier) GetOAuth2GithubDefaultEligible(_ context.Context) (bool, error) {
+	q.mutex.RLock()
+	defer q.mutex.RUnlock()
+
+	if q.oauth2GithubDefaultEligible == nil {
+		return false, sql.ErrNoRows
+	}
+	return *q.oauth2GithubDefaultEligible, nil
 }
 
 func (q *FakeQuerier) GetOAuth2ProviderAppByID(_ context.Context, id uuid.UUID) (database.OAuth2ProviderApp, error) {
@@ -11151,8 +11158,12 @@ func (q *FakeQuerier) UpsertNotificationsSettings(_ context.Context, data string
 	return nil
 }
 
-func (q *FakeQuerier) UpsertOAuth2GithubDefaultEligible(ctx context.Context, eligible bool) error {
-	panic("not implemented")
+func (q *FakeQuerier) UpsertOAuth2GithubDefaultEligible(_ context.Context, eligible bool) error {
+	q.mutex.Lock()
+	defer q.mutex.Unlock()
+
+	q.oauth2GithubDefaultEligible = &eligible
+	return nil
 }
 
 func (q *FakeQuerier) UpsertOAuthSigningKey(_ context.Context, value string) error {

Original file line number	Diff line number	Diff line change
`@@ -1851,7 +1851,10 @@ func (q *querier) GetNotificationsSettings(ctx context.Context) (string, error)`
`1851`	`1851`	`}`
`1852`	`1852`
`1853`	`1853`	`func (q *querier) GetOAuth2GithubDefaultEligible(ctx context.Context) (bool, error) {`
`1854`		`- panic("not implemented")`
	`1854`	`+ if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceDeploymentConfig); err != nil {`
	`1855`	`+ return false, err`
	`1856`	`+ }`
	`1857`	`+ return q.db.GetOAuth2GithubDefaultEligible(ctx)`
`1855`	`1858`	`}`
`1856`	`1859`
`1857`	`1860`	`func (q *querier) GetOAuth2ProviderAppByID(ctx context.Context, id uuid.UUID) (database.OAuth2ProviderApp, error) {`
`@@ -4435,7 +4438,10 @@ func (q *querier) UpsertNotificationsSettings(ctx context.Context, value string)`
`4435`	`4438`	`}`
`4436`	`4439`
`4437`	`4440`	`func (q *querier) UpsertOAuth2GithubDefaultEligible(ctx context.Context, eligible bool) error {`
`4438`		`- panic("not implemented")`
	`4441`	`+ if err := q.authorizeContext(ctx, policy.ActionUpdate, rbac.ResourceDeploymentConfig); err != nil {`
	`4442`	`+ return err`
	`4443`	`+ }`
	`4444`	`+ return q.db.UpsertOAuth2GithubDefaultEligible(ctx, eligible)`
`4439`	`4445`	`}`
`4440`	`4446`
`4441`	`4447`	`func (q *querier) UpsertOAuthSigningKey(ctx context.Context, value string) error {`