-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: implement meme token-whale contract and unit test
style: modify and add comments refactor: remove unused import feat: add minting function and validate with lightclient before mint and transfer style: delete unused import and fix comments refactor: delete unnecessary function refactor: change contract struct test: fix test for minting and transfer and add test for burn and double minting style: edit import style: fix for yarn format
- Loading branch information
Showing
1 changed file
with
211 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1 +1,212 @@ | ||
use near_contract_standards::fungible_token::metadata::{ | ||
FungibleTokenMetadata, FungibleTokenMetadataProvider, | ||
}; | ||
use near_contract_standards::fungible_token::{events, FungibleToken}; | ||
use near_sdk::borsh::{self, BorshDeserialize, BorshSerialize}; | ||
use near_sdk::json_types::U128; | ||
use near_sdk::{near_bindgen, AccountId, PanicOnDefault, PromiseOrValue}; | ||
|
||
#[near_bindgen] | ||
#[derive(BorshDeserialize, BorshSerialize, PanicOnDefault)] | ||
pub struct WhaleContract { | ||
token: FungibleToken, | ||
decimals: u8, | ||
name: String, | ||
symbol: String, | ||
icon: Option<String>, | ||
} | ||
|
||
near_contract_standards::impl_fungible_token_core!(WhaleContract, token); | ||
near_contract_standards::impl_fungible_token_storage!(WhaleContract, token); | ||
|
||
///TODO: replace PDAO's whale icon | ||
const PDAO_WHALE_ICON: &str = "PDAO WHALE ICON ADRESSS"; | ||
|
||
#[near_bindgen] | ||
impl WhaleContract { | ||
/// Initializes the contract with PDAO whale spec | ||
#[init] | ||
pub fn new() -> Self { | ||
Self { | ||
token: FungibleToken::new(b"w".to_vec()), | ||
decimals: 8, | ||
name: "PDAO_meme_token-Whale".to_string(), | ||
symbol: "PDAO-WHALE".to_string(), | ||
icon: Some(PDAO_WHALE_ICON.to_string()), | ||
} | ||
} | ||
|
||
pub fn mint_whale(&mut self, receiver_id: AccountId, amount: U128) { | ||
//before minting validate with lightclient | ||
self.validate_with_lightclient(); | ||
|
||
if self.token.accounts.get(&receiver_id).is_none() { | ||
self.token.internal_register_account(&receiver_id); | ||
self.token.internal_deposit(&receiver_id, amount.into()); | ||
} else { | ||
self.token.internal_deposit(&receiver_id, amount.into()); | ||
}; | ||
|
||
// log mint | ||
events::FtMint { | ||
owner_id: &receiver_id, | ||
amount: &amount, | ||
memo: Some("PDAO Whale token newly minted"), | ||
} | ||
.emit() | ||
} | ||
|
||
pub fn transfer_whale(&mut self, receiver_id: AccountId, amount: U128, memo: Option<String>) { | ||
//before transfer validate with lightclient | ||
self.validate_with_lightclient(); | ||
// use FungibleTokenCore::ft_transfer for internal transfer | ||
self.token.ft_transfer(receiver_id, amount, memo); | ||
} | ||
|
||
#[payable] | ||
pub fn transfer_whale_call( | ||
&mut self, | ||
receiver_id: AccountId, | ||
amount: U128, | ||
memo: Option<String>, | ||
msg: String, | ||
) -> PromiseOrValue<U128> { | ||
self.validate_with_lightclient(); | ||
self.token.ft_transfer_call(receiver_id, amount, memo, msg) | ||
} | ||
|
||
pub fn burn(&mut self, account_id: AccountId, amount: U128) { | ||
// validate with lightclient | ||
self.validate_with_lightclient(); | ||
self.token.internal_withdraw(&account_id, amount.into()); | ||
|
||
//log burn | ||
events::FtBurn { | ||
owner_id: &account_id, | ||
amount: &amount, | ||
memo: Some("PDAO Whale token burned"), | ||
} | ||
.emit() | ||
} | ||
} | ||
|
||
impl WhaleContract { | ||
fn validate_with_lightclient(&self) { | ||
//todo | ||
} | ||
} | ||
|
||
#[near_bindgen] | ||
impl FungibleTokenMetadataProvider for WhaleContract { | ||
fn ft_metadata(&self) -> FungibleTokenMetadata { | ||
FungibleTokenMetadata { | ||
spec: "ft-1.0.0".to_string(), | ||
reference: None, | ||
reference_hash: None, | ||
decimals: self.decimals, | ||
name: self.name.clone(), | ||
symbol: self.symbol.clone(), | ||
icon: self.icon.clone(), | ||
} | ||
} | ||
} | ||
|
||
#[cfg(all(test, not(target_arch = "wasm32")))] | ||
mod tests { | ||
use near_sdk::test_utils::{accounts, VMContextBuilder}; | ||
use near_sdk::{env, testing_env}; | ||
|
||
use super::*; | ||
|
||
fn get_context(predecessor_account_id: AccountId) -> VMContextBuilder { | ||
let mut builder = VMContextBuilder::new(); | ||
builder | ||
.current_account_id(accounts(0)) | ||
.signer_account_id(predecessor_account_id.clone()) | ||
.predecessor_account_id(predecessor_account_id); | ||
builder | ||
} | ||
|
||
#[test] | ||
#[should_panic(expected = "The contract is not initialized")] | ||
fn test_default() { | ||
let context = get_context(accounts(1)); | ||
testing_env!(context.build()); | ||
let _contract = WhaleContract::default(); | ||
} | ||
|
||
#[test] | ||
fn test_minting() { | ||
let mut context = VMContextBuilder::new(); | ||
testing_env!(context.build()); | ||
let mut contract = WhaleContract::new(); | ||
testing_env!(context | ||
.attached_deposit(125 * env::storage_byte_cost()) | ||
.build()); | ||
contract.mint_whale(accounts(0), 1_000_000.into()); | ||
assert_eq!(contract.ft_balance_of(accounts(0)), 1_000_000.into()); | ||
} | ||
|
||
#[test] | ||
fn test_transfer() { | ||
let mut context = get_context(accounts(2)); | ||
testing_env!(context.build()); | ||
let mut contract = WhaleContract::new(); | ||
testing_env!(context | ||
.storage_usage(env::storage_usage()) | ||
.attached_deposit(contract.storage_balance_bounds().min.into()) | ||
.predecessor_account_id(accounts(1)) | ||
.build()); | ||
|
||
contract.mint_whale(accounts(2), 1_000_000.into()); | ||
|
||
// Paying for account registration, aka storage deposit | ||
contract.storage_deposit(None, None); | ||
|
||
testing_env!(context | ||
.storage_usage(env::storage_usage()) | ||
.attached_deposit(1) | ||
.predecessor_account_id(accounts(2)) | ||
.build()); | ||
let transfer_amount = 1_000_000 / 3; | ||
contract.ft_transfer(accounts(1), transfer_amount.into(), None); | ||
|
||
testing_env!(context | ||
.storage_usage(env::storage_usage()) | ||
.account_balance(env::account_balance()) | ||
.is_view(true) | ||
.attached_deposit(0) | ||
.build()); | ||
assert_eq!( | ||
contract.ft_balance_of(accounts(2)).0, | ||
(1_000_000 - transfer_amount) | ||
); | ||
assert_eq!(contract.ft_balance_of(accounts(1)).0, transfer_amount); | ||
} | ||
|
||
#[test] | ||
fn test_burn() { | ||
let mut context = VMContextBuilder::new(); | ||
testing_env!(context.build()); | ||
let mut contract = WhaleContract::new(); | ||
testing_env!(context | ||
.attached_deposit(125 * env::storage_byte_cost()) | ||
.build()); | ||
contract.mint_whale(accounts(0), 1_000_000.into()); | ||
contract.burn(accounts(0), 500_000.into()); | ||
assert_eq!(contract.ft_balance_of(accounts(0)), 500_000.into()); | ||
} | ||
|
||
#[test] | ||
fn double_minting() { | ||
let mut context = VMContextBuilder::new(); | ||
testing_env!(context.build()); | ||
let mut contract = WhaleContract::new(); | ||
testing_env!(context | ||
.attached_deposit(125 * env::storage_byte_cost()) | ||
.build()); | ||
contract.mint_whale(accounts(0), 1_000_000.into()); | ||
contract.mint_whale(accounts(0), 1_000_000.into()); | ||
assert_eq!(contract.ft_balance_of(accounts(0)), 2_000_000.into()); | ||
} | ||
} |