From 1e4abbfe69b35d335b16176dcbd9473b00b9a0c3 Mon Sep 17 00:00:00 2001
From: Uncle Stretch <uncle.stretch@ghostchain.io>
Date: Wed, 4 Jun 2025 15:35:48 +0300
Subject: [PATCH] function to trigger nullification period, tests for the new
 functionality and expansion of NetworkData trait

Signed-off-by: Uncle Stretch <uncle.stretch@ghostchain.io>
---
 pallets/networks/Cargo.toml    |  2 +-
 pallets/networks/src/lib.rs    |  9 +++++++++
 pallets/networks/src/tests.rs  | 12 ++++++++++++
 pallets/traits/Cargo.toml      |  2 +-
 pallets/traits/src/networks.rs |  1 +
 5 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/pallets/networks/Cargo.toml b/pallets/networks/Cargo.toml
index 177ceb9..30ad1a9 100644
--- a/pallets/networks/Cargo.toml
+++ b/pallets/networks/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "ghost-networks"
-version = "0.1.4"
+version = "0.1.5"
 license.workspace = true
 authors.workspace = true
 edition.workspace = true
diff --git a/pallets/networks/src/lib.rs b/pallets/networks/src/lib.rs
index cdc0c77..b43a032 100644
--- a/pallets/networks/src/lib.rs
+++ b/pallets/networks/src/lib.rs
@@ -697,4 +697,13 @@ impl<T: Config> NetworkDataMutateHandler<NetworkData, BalanceOf<T>> for Pallet<T
     fn nullify_commission() {
         AccumulatedCommission::<T>::set(Default::default());
     }
+
+    fn trigger_nullification() {
+        if NullifyNeeded::<T>::get() {
+            Self::nullify_commission();
+            NullifyNeeded::<T>::put(false);
+        } else {
+            NullifyNeeded::<T>::put(true);
+        }
+    }
 }
diff --git a/pallets/networks/src/tests.rs b/pallets/networks/src/tests.rs
index 07d1b1c..9c3a010 100644
--- a/pallets/networks/src/tests.rs
+++ b/pallets/networks/src/tests.rs
@@ -935,3 +935,15 @@ fn bridged_inflation_era_payout_triggers_need_of_nullification() {
             assert_eq!(NullifyNeeded::<Test>::get(), false);
         });
 }
+
+#[test]
+fn trigger_nullification_works_as_expected() {
+    ExtBuilder::build()
+        .execute_with(|| {
+            assert_eq!(NullifyNeeded::<Test>::get(), false);
+            GhostNetworks::trigger_nullification();
+            assert_eq!(NullifyNeeded::<Test>::get(), true);
+            GhostNetworks::trigger_nullification();
+            assert_eq!(NullifyNeeded::<Test>::get(), false);
+        });
+}
diff --git a/pallets/traits/Cargo.toml b/pallets/traits/Cargo.toml
index 170dba4..0bd089d 100644
--- a/pallets/traits/Cargo.toml
+++ b/pallets/traits/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "ghost-traits"
-version = "0.3.20"
+version = "0.3.21"
 license.workspace = true
 authors.workspace = true
 edition.workspace = true
diff --git a/pallets/traits/src/networks.rs b/pallets/traits/src/networks.rs
index 30752c6..eafb8e6 100644
--- a/pallets/traits/src/networks.rs
+++ b/pallets/traits/src/networks.rs
@@ -38,4 +38,5 @@ pub trait NetworkDataMutateHandler<Network, Balance>: NetworkDataInspectHandler<
     ) -> Result<(Balance, Balance), ()>;
     fn accumulate_commission(commission: &Balance) -> Result<Balance, ()>;
     fn nullify_commission();
+    fn trigger_nullification();
 }