Skip to content

Commit

Permalink
Merge pull request #84 from horuslabsio/feat/refactor-pub
Browse files Browse the repository at this point in the history
chore: refactor test
  • Loading branch information
codeWhizperer authored Jul 10, 2024
2 parents 24140b6 + 944036b commit bdd57ba
Showing 1 changed file with 15 additions and 51 deletions.
66 changes: 15 additions & 51 deletions tests/test_publication.cairo
Original file line number Diff line number Diff line change
Expand Up @@ -62,13 +62,11 @@ fn __setup__() -> (
// declare account
let account_class_hash = declare("Account").unwrap();

// ///// Deploying karst account for USER AND USE
// ///// Deploying karst Profile for USER
let publication_dispatcher = IComposableDispatcher {
contract_address: publication_contract_address
};
start_prank(
CheatTarget::Multiple(array![publication_contract_address]), USER_ONE.try_into().unwrap()
);
start_prank(CheatTarget::One(publication_contract_address), USER_ONE.try_into().unwrap());
let user_one_profile_address = publication_dispatcher
.create_profile(
nft_contract_address,
Expand All @@ -77,19 +75,12 @@ fn __setup__() -> (
2478,
USER_ONE.try_into().unwrap()
);
publication_dispatcher
.set_profile_metadata_uri(
user_one_profile_address.try_into().unwrap(),
"ipfs://QmSkDCsS32eLpcymxtn1cEn7Rc5hfefLBgfvZyjaYXr4ga/"
);
let contentURI: ByteArray = "ipfs://helloworld";
let user_one_first_post_pointed_pub_id = publication_dispatcher
.post(contentURI, user_one_profile_address, publication_contract_address);
stop_prank(CheatTarget::Multiple(array![publication_contract_address]),);
stop_prank(CheatTarget::One(publication_contract_address),);

start_prank(
CheatTarget::Multiple(array![publication_contract_address]), USER_TWO.try_into().unwrap()
);
start_prank(CheatTarget::One(publication_contract_address), USER_TWO.try_into().unwrap());
let user_two_profile_address = publication_dispatcher
.create_profile(
nft_contract_address,
Expand All @@ -98,18 +89,11 @@ fn __setup__() -> (
2479,
USER_TWO.try_into().unwrap()
);
publication_dispatcher
.set_profile_metadata_uri(
user_two_profile_address.try_into().unwrap(),
"ipfs://QmSkDCsS32eLpcymxtn1cEn7Rc5hfefLBgfvZyjaYXr4ga/"
);
let contentURI: ByteArray = "ipfs://helloworld";
publication_dispatcher.post(contentURI, user_two_profile_address, publication_contract_address);
stop_prank(CheatTarget::Multiple(array![publication_contract_address]),);
stop_prank(CheatTarget::One(publication_contract_address),);

start_prank(
CheatTarget::Multiple(array![publication_contract_address,]), USER_THREE.try_into().unwrap()
);
start_prank(CheatTarget::One(publication_contract_address), USER_THREE.try_into().unwrap());
let user_three_profile_address = publication_dispatcher
.create_profile(
nft_contract_address,
Expand All @@ -118,15 +102,10 @@ fn __setup__() -> (
2480,
USER_THREE.try_into().unwrap()
);
publication_dispatcher
.set_profile_metadata_uri(
user_three_profile_address.try_into().unwrap(),
"ipfs://QmSkDCsS32eLpcymxtn1cEn7Rc5hfefLBgfvZyjaYXr4ga/"
);
let contentURI: ByteArray = "ipfs://helloworld";
publication_dispatcher
.post(contentURI, user_three_profile_address, publication_contract_address);
stop_prank(CheatTarget::Multiple(array![publication_contract_address]),);
stop_prank(CheatTarget::One(publication_contract_address));

return (
nft_contract_address,
Expand Down Expand Up @@ -162,15 +141,13 @@ fn test_post() {
let publication_dispatcher = IComposableDispatcher {
contract_address: publication_contract_address
};
start_prank(
CheatTarget::Multiple(array![publication_contract_address]), USER_ONE.try_into().unwrap()
);
start_prank(CheatTarget::One(publication_contract_address), USER_ONE.try_into().unwrap());

let publication_type = publication_dispatcher
.get_publication_type(user_one_profile_address, user_one_first_post_pointed_pub_id);
assert(publication_type == PublicationType::Post, 'invalid pub_type');

stop_prank(CheatTarget::Multiple(array![publication_contract_address]),);
stop_prank(CheatTarget::One(publication_contract_address));
}

#[test]
Expand All @@ -190,9 +167,7 @@ fn test_comment() {
let publication_dispatcher = IComposableDispatcher {
contract_address: publication_contract_address
};
start_prank(
CheatTarget::Multiple(array![publication_contract_address]), USER_ONE.try_into().unwrap()
);
start_prank(CheatTarget::One(publication_contract_address), USER_ONE.try_into().unwrap());
let user_one_comment_on_his_post_content_URI =
"ipfs://QmSkDCsS32eLpcymxtn1cEn7Rc5hfefLBgfvZyjaryrga/";
let user_two_comment_one_user_one_post_content_URI =
Expand Down Expand Up @@ -229,7 +204,7 @@ fn test_comment() {
assert(publication_type == PublicationType::Comment, 'invalid pub_type');
assert(user_one_publication_root_id == user_two_comment_publication_root_id, 'Invalid root_id');

stop_prank(CheatTarget::Multiple(array![publication_contract_address]),);
stop_prank(CheatTarget::One(publication_contract_address),);
}

#[test]
Expand Down Expand Up @@ -286,9 +261,7 @@ fn test_quote() {
let publication_dispatcher = IComposableDispatcher {
contract_address: publication_contract_address
};
start_prank(
CheatTarget::Multiple(array![publication_contract_address]), USER_ONE.try_into().unwrap()
);
start_prank(CheatTarget::One(publication_contract_address), USER_ONE.try_into().unwrap());
let quote_content_URI = "ipfs://QmSkDCsS32eLpcymxtn1cEn7Rc5hfefLBgfvZysddefzp/";

let quote_params = QuoteParams {
Expand All @@ -307,9 +280,7 @@ fn test_quote() {

stop_prank(CheatTarget::Multiple(array![publication_contract_address]),);

start_prank(
CheatTarget::Multiple(array![publication_contract_address]), USER_TWO.try_into().unwrap()
);
start_prank(CheatTarget::One(publication_contract_address), USER_TWO.try_into().unwrap());
let user_two_quote_content_URI = "ipfs://QmSkDCsS32eLpcymxtn1cEn7Rc5hfefLBgfvZysdjbezo/";

let user_two_quote_params = QuoteParams {
Expand All @@ -334,7 +305,7 @@ fn test_quote() {
.root_profile_address;
assert(user_one_publication_root_id == user_two_publication_root_id, 'Invalid root_id');

stop_prank(CheatTarget::Multiple(array![publication_contract_address]),);
stop_prank(CheatTarget::One(publication_contract_address));
}

#[test]
Expand Down Expand Up @@ -390,9 +361,7 @@ fn test_quote_pointed_profile_address() {
let publication_dispatcher = IComposableDispatcher {
contract_address: publication_contract_address
};
start_prank(
CheatTarget::Multiple(array![publication_contract_address]), USER_ONE.try_into().unwrap()
);
start_prank(CheatTarget::One(publication_contract_address), USER_ONE.try_into().unwrap());
let quote_content_URI = "ipfs://QmSkDCsS32eLpcymxtn1cEn7Rc5hfefLBgfvZysddefzp/";

let quote_params = QuoteParams {
Expand Down Expand Up @@ -586,7 +555,6 @@ fn test_mirror_root_profile_address() {
};

start_prank(CheatTarget::One(publication_contract_address), USER_ONE.try_into().unwrap());
// let componentDispatcher = IComposableDispatcher { contract_address: publication_contract_address };

publication_dispatcher.mirror(mirror_params, publication_contract_address);

Expand All @@ -598,7 +566,3 @@ fn test_mirror_root_profile_address() {

stop_prank(CheatTarget::One(publication_contract_address),);
}

fn to_address(name: felt252) -> ContractAddress {
name.try_into().unwrap()
}

0 comments on commit bdd57ba

Please sign in to comment.