fee cut tests

Author: kaizu-xyz

contracts/launchpad/tests/fees.move

@@ -6,11 +6,14 @@ module ob_launchpad::test_fees {
     use sui::test_scenario::{Self, ctx};
 
     use ob_launchpad::flat_fee;
+    use ob_launchpad::fee_cut;
     use ob_launchpad::listing;
     use ob_launchpad::test_listing;
 
     const MARKETPLACE: address = @0xA123;
     const CREATOR: address = @0xA1C05;
+    const CUT_ADDR_1: address = @0x1;
+    const CUT_ADDR_2: address = @0x2;
 
     #[test]
     public fun marketplace_default_fee() {
@@ -110,4 +113,62 @@ module ob_launchpad::test_fees {
         test_scenario::return_shared(listing);
         test_scenario::end(scenario);
     }
+
+    #[test]
+    public fun test_fee_cut() {
+        let scenario = test_scenario::begin(MARKETPLACE);
+
+        // Creates `Marketplace` with default fee
+        let (marketplace, listing) = test_listing::init_listing_and_marketplace(
+            CREATOR, MARKETPLACE, 2000, &mut scenario,
+        );
+
+        fee_cut::add_fee_cut(
+            &mut marketplace,
+            &mut listing,
+            CUT_ADDR_1,
+            70,
+            ctx(&mut scenario)
+        );
+
+        fee_cut::add_fee_cut(
+            &mut marketplace,
+            &mut listing,
+            CUT_ADDR_2,
+            30,
+            ctx(&mut scenario)
+        );
+
+        listing::pay(&mut listing, balance::create_for_testing<SUI>(10_000), 1);
+
+        flat_fee::collect_proceeds_and_fees<SUI>(
+            &marketplace, &mut listing, ctx(&mut scenario),
+        );
+
+        test_scenario::next_tx(&mut scenario, MARKETPLACE);
+
+        let marketplace_proceeds =
+            test_scenario::take_from_address<Coin<SUI>>(&scenario, MARKETPLACE);
+        assert!(coin::value(&marketplace_proceeds) == 1_980, 0);
+
+        test_scenario::return_to_address(MARKETPLACE, marketplace_proceeds);
+
+        let creator_proceeds =
+            test_scenario::take_from_address<Coin<SUI>>(&scenario, CREATOR);
+        assert!(coin::value(&creator_proceeds) == 7_920, 0);
+
+        let addr1_proceeds = test_scenario::take_from_address<Coin<SUI>>(&scenario, CUT_ADDR_1);
+        assert!(coin::value(&addr1_proceeds) == 70, 0);
+
+        let addr2_proceeds = test_scenario::take_from_address<Coin<SUI>>(&scenario, CUT_ADDR_2);
+        assert!(coin::value(&addr2_proceeds) == 30, 0);
+
+        test_scenario::return_to_address(CREATOR, creator_proceeds);
+        test_scenario::return_to_address(CUT_ADDR_1, addr1_proceeds);
+        test_scenario::return_to_address(CUT_ADDR_2, addr2_proceeds);
+
+        test_scenario::return_shared(marketplace);
+        test_scenario::return_shared(listing);
+        test_scenario::end(scenario);
+    }
 }