update claimant balance on claim

reflector-network · Jul 31, 2024 · 2dd9d82 · 2dd9d82
1 parent db187b4
commit 2dd9d82
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -151,6 +151,9 @@ impl DAOContract {
         }
         // transfer tokens to the destination address
         token(&e).transfer(&e.current_contract_address(), &to, &amount);
+
+        // update available balance
+        update_available_balance(&e, &claimant, &(-amount));
     }
 
     /// Create a new ballot

diff --git a/src/test.rs b/src/test.rs
@@ -140,4 +140,14 @@ fn test() {
         let balance = env.get_available_balance(&operators.first().unwrap());
         assert!(balance > 0);
     });
+
+    let available = client.available(&developer);
+    assert!(available > 0);
+
+    client.claim(&developer, &developer, &available);
+
+    env.as_contract(&client.address, || {
+        let balance = env.get_available_balance(&developer);
+        assert_eq!(balance, 0);
+    });
 }