[CONTRACTS] Implement claim_bounty escrow release for #2

contracts/bounty/src/lib.rs

@@ -29,9 +29,38 @@ impl BountyContract {
         env.storage().persistent().set(&key, &bounty);
     }
 
+    pub fn claim_bounty(
+        env: Env,
+        token: Address,
+        maintainer: Address,
+        contributor: Address,
+        issue_id: Symbol,
+    ) {
+        maintainer.require_auth();
+
+        let key = (symbol_short!("bounty"), issue_id);
+        let mut bounty: Map<Address, i128> = env
+            .storage()
+            .persistent()
+            .get(&key)
+            .unwrap_or(Map::new(&env));
+        let amount = bounty
+            .get(maintainer.clone())
+            .unwrap_or_else(|| panic!("No bounty exists for this issue and maintainer"));
+
+        let client = token::Client::new(&env, &token);
+        client.transfer(&env.current_contract_address(), &contributor, &amount);
+
+        bounty.remove(maintainer);
+        env.storage().persistent().set(&key, &bounty);
+    }
+
     pub fn cancel_bounty(env: Env, token: Address, maintainer: Address, amount: i128) {
         maintainer.require_auth();
         let client = token::Client::new(&env, &token);
         client.transfer(&env.current_contract_address(), &maintainer, &amount);
     }
 }
+
+#[cfg(test)]
+mod test;

contracts/bounty/src/test.rs

@@ -1,21 +1,60 @@
 #![cfg(test)]
 
-use super::*;
-use soroban_sdk::{vec, Env, String};
+use super::{BountyContract, BountyContractClient};
+use soroban_sdk::{symbol_short, token, Address, Env};
+use soroban_sdk::testutils::Address as _;
 
 #[test]
-fn test() {
+fn claim_bounty_releases_locked_balance_and_clears_bounty() {
     let env = Env::default();
-    let contract_id = env.register(Contract, ());
-    let client = ContractClient::new(&env, &contract_id);
-
-    let words = client.hello(&String::from_str(&env, "Dev"));
-    assert_eq!(
-        words,
-        vec![
-            &env,
-            String::from_str(&env, "Hello"),
-            String::from_str(&env, "Dev"),
-        ]
-    );
+    env.mock_all_auths();
+
+    let maintainer = Address::generate(&env);
+    let contributor = Address::generate(&env);
+    let token_admin = Address::generate(&env);
+
+    let token_address = env.register_stellar_asset_contract_v2(token_admin);
+    let token = token::Client::new(&env, &token_address.address());
+    let token_admin_client = token::StellarAssetClient::new(&env, &token_address.address());
+
+    token_admin_client.mint(&maintainer, &1_000);
+
+    let contract_id = env.register(BountyContract, ());
+    let client = BountyContractClient::new(&env, &contract_id);
+
+    let issue_id = symbol_short!("iss1");
+    client.create_bounty(&token_address.address(), &maintainer, &issue_id, &300);
+
+    assert_eq!(token.balance(&maintainer), 700);
+    assert_eq!(token.balance(&contract_id), 300);
+
+    client.claim_bounty(&token_address.address(), &maintainer, &contributor, &issue_id);
+
+    assert_eq!(token.balance(&contract_id), 0);
+    assert_eq!(token.balance(&contributor), 300);
+}
+
+#[test]
+#[should_panic(expected = "No bounty exists for this issue and maintainer")]
+fn claim_bounty_cannot_be_executed_twice() {
+    let env = Env::default();
+    env.mock_all_auths();
+
+    let maintainer = Address::generate(&env);
+    let contributor = Address::generate(&env);
+    let token_admin = Address::generate(&env);
+
+    let token_address = env.register_stellar_asset_contract_v2(token_admin);
+    let token_admin_client = token::StellarAssetClient::new(&env, &token_address.address());
+
+    token_admin_client.mint(&maintainer, &500);
+
+    let contract_id = env.register(BountyContract, ());
+    let client = BountyContractClient::new(&env, &contract_id);
+
+    let issue_id = symbol_short!("iss2");
+    client.create_bounty(&token_address.address(), &maintainer, &issue_id, &200);
+    client.claim_bounty(&token_address.address(), &maintainer, &contributor, &issue_id);
+
+    client.claim_bounty(&token_address.address(), &maintainer, &contributor, &issue_id);
 }