create deleted users

johnstcn · johnstcn · commit 49b9d90f006e · 2023-09-15T11:36:59.000+01:00
diff --git a/coderd/database/dbgen/dbgen.go b/coderd/database/dbgen/dbgen.go
@@ -240,6 +240,14 @@ func User(t testing.TB, db database.Store, orig database.User) database.User {
 		})
 		require.NoError(t, err, "user last seen")
 	}
+
+	if orig.Deleted {
+		err = db.UpdateUserDeletedByID(genCtx, database.UpdateUserDeletedByIDParams{
+			ID:      user.ID,
+			Deleted: orig.Deleted,
+		})
+		require.NoError(t, err, "set user as deleted")
+	}
 	return user
 }
 
diff --git a/enterprise/cli/server_dbcrypt_test.go b/enterprise/cli/server_dbcrypt_test.go
@@ -209,15 +209,19 @@ func genData(t *testing.T, db database.Store, n int) []database.User {
 	var users []database.User
 	// Make some users
 	for i := 0; i < n; i++ {
+		var deleted bool
 		status := database.UserStatusActive
 		if i%2 == 0 {
 			status = database.UserStatusSuspended
 		} else if i%3 == 0 {
 			status = database.UserStatusDormant
+		} else if i%5 == 0 {
+			deleted = true
 		}
 		usr := dbgen.User(t, db, database.User{
 			LoginType: database.LoginTypeOIDC,
 			Status:    status,
+			Deleted:   deleted,
 		})
 		_ = dbgen.UserLink(t, db, database.UserLink{
 			UserID:            usr.ID,

Original file line number	Diff line number	Diff line change
`@@ -240,6 +240,14 @@ func User(t testing.TB, db database.Store, orig database.User) database.User {`
`240`	`240`	`})`
`241`	`241`	`require.NoError(t, err, "user last seen")`
`242`	`242`	`}`
	`243`	`+`
	`244`	`+ if orig.Deleted {`
	`245`	`+ err = db.UpdateUserDeletedByID(genCtx, database.UpdateUserDeletedByIDParams{`
	`246`	`+ ID: user.ID,`
	`247`	`+ Deleted: orig.Deleted,`
	`248`	`+ })`
	`249`	`+ require.NoError(t, err, "set user as deleted")`
	`250`	`+ }`
`243`	`251`	`return user`
`244`	`252`	`}`
`245`	`253`