pricing config

MystenLabs · Oct 22, 2024 · dec6acc · dec6acc
1 parent 3e90918
commit dec6acc
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/packages/suins/sources/suins.move b/packages/suins/sources/suins.move
@@ -22,6 +22,7 @@
 /// the registry and the balance.
 module suins::suins {
     use sui::{balance::{Self, Balance}, coin::{Self, Coin}, dynamic_field as df, sui::SUI};
+    use suins::pricing::{Self, new_range};
 
     /// Trying to withdraw from an empty balance.
     const ENoProfits: u64 = 0;
@@ -201,6 +202,19 @@ module suins::suins {
             200 * suins::constants::mist_per_sui(),
             50 * suins::constants::mist_per_sui(),
         ));
+
+        let range1 = new_range(vector[3, 3]);
+        let range2 = new_range(vector[4, 4]);
+        let range3 = new_range(vector[5, 63]);
+        let prices = vector[
+            1200 * suins::constants::mist_per_sui(),
+            200 * suins::constants::mist_per_sui(),
+            50 * suins::constants::mist_per_sui()
+        ];
+
+        let pricing_config = pricing::new<SUI>(vector[range1, range2, range3], prices);
+        add_config(&admin_cap, &mut suins, pricing_config);
+
         transfer::transfer(admin_cap, tx_context::sender(ctx));
         suins
     }

diff --git a/packages/suins/tests/register_tests.move b/packages/suins/tests/register_tests.move
@@ -61,7 +61,7 @@ module suins::register_sample_tests {
         let mut clock = test_scenario::take_shared<Clock>(scenario);
 
         clock.increment_for_testing(clock_tick);
-        let nft = register(&mut suins, domain_name, no_years, payment, &clock, ctx(scenario));
+        let nft = register<SUI>(&mut suins, domain_name, no_years, payment, &clock, ctx(scenario));
 
         test_scenario::return_shared(clock);
         test_scenario::return_shared(suins);

diff --git a/packages/suins/tests/v2/register.move b/packages/suins/tests/v2/register.move
@@ -43,7 +43,7 @@ module suins::register_sample {
         assert!(0 < no_years && no_years <= 5, EInvalidYearsArgument);
 
         let label = domain::sld(&domain);
-        let price = pricing::calculate_price(config, string::length(label));
+        let price = pricing::calculate_price(config, string::length(label)) * (no_years as u64);
 
         assert!(coin::value(&payment) == price, EIncorrectAmount);