mirror of
https://github.com/zed-industries/zed.git
synced 2024-10-26 08:31:04 +00:00
Don't assign invite code when updating count from 0 to 0
Co-Authored-By: Max Brunsfeld <maxbrunsfeld@gmail.com>
This commit is contained in:
parent
4b8f24c84e
commit
51a61cc485
1 changed files with 14 additions and 7 deletions
|
@ -228,6 +228,7 @@ impl Db for PostgresDb {
|
|||
|
||||
async fn set_invite_count(&self, id: UserId, count: u32) -> Result<()> {
|
||||
let mut tx = self.pool.begin().await?;
|
||||
if count > 0 {
|
||||
sqlx::query(
|
||||
"
|
||||
UPDATE users
|
||||
|
@ -239,6 +240,8 @@ impl Db for PostgresDb {
|
|||
.bind(id)
|
||||
.execute(&mut tx)
|
||||
.await?;
|
||||
}
|
||||
|
||||
sqlx::query(
|
||||
"
|
||||
UPDATE users
|
||||
|
@ -1394,6 +1397,10 @@ pub mod tests {
|
|||
// Initially, user 1 has no invite code
|
||||
assert_eq!(db.get_invite_code_for_user(user1).await.unwrap(), None);
|
||||
|
||||
// Setting invite count to 0 when no code is assigned does not assign a new code
|
||||
db.set_invite_count(user1, 0).await.unwrap();
|
||||
assert!(db.get_invite_code_for_user(user1).await.unwrap().is_none());
|
||||
|
||||
// User 1 creates an invite code that can be used twice.
|
||||
db.set_invite_count(user1, 2).await.unwrap();
|
||||
let (invite_code, invite_count) =
|
||||
|
|
Loading…
Reference in a new issue