feat: initialize Kurdistan SDK - independent fork of Polkadot SDK

2025-12-13 15:44:15 +03:00
commit e4778b4576
6838 changed files with 1847450 additions and 0 deletions
@@ -0,0 +1,52 @@
+// Copyright (C) Parity Technologies (UK) Ltd.
+// This file is part of Pezkuwi.
+
+// Pezkuwi is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// Pezkuwi is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with Pezkuwi.  If not, see <http://www.gnu.org/licenses/>.
+
+use super::*;
+use frame_benchmarking::v2::*;
+use frame_system::RawOrigin;
+use pezkuwi_primitives::ConsensusLog;
+use sp_runtime::DigestItem;
+
+// Random large number for the digest
+const DIGEST_MAX_LEN: u32 = 65536;
+
+#[benchmarks]
+mod benchmarks {
+	use super::*;
+
+	#[benchmark]
+	fn force_approve(d: Linear<0, DIGEST_MAX_LEN>) -> Result<(), BenchmarkError> {
+		for _ in 0..d {
+			frame_system::Pallet::<T>::deposit_log(ConsensusLog::ForceApprove(d).into());
+		}
+
+		#[extrinsic_call]
+		_(RawOrigin::Root, d + 1);
+
+		assert_eq!(
+			frame_system::Pallet::<T>::digest().logs.last().unwrap(),
+			&DigestItem::from(ConsensusLog::ForceApprove(d + 1)),
+		);
+
+		Ok(())
+	}
+
+	impl_benchmark_test_suite!(
+		Pallet,
+		crate::mock::new_test_ext(Default::default()),
+		crate::mock::Test
+	);
+}
@@ -0,0 +1,145 @@
+// Copyright (C) Parity Technologies (UK) Ltd.
+// This file is part of Pezkuwi.
+
+// Pezkuwi is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// Pezkuwi is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with Pezkuwi.  If not, see <http://www.gnu.org/licenses/>.
+
+use super::*;
+use crate::{
+	mock::{new_test_ext, Dmp, Initializer, MockGenesisConfig, Paras, System, Test},
+	paras::ParaKind,
+	session_info,
+};
+use pezkuwi_primitives::{HeadData, Id as ParaId};
+use pezkuwi_primitives_test_helpers::dummy_validation_code;
+
+use frame_support::{
+	assert_ok,
+	traits::{OnFinalize, OnInitialize},
+};
+
+#[test]
+fn session_0_is_instantly_applied() {
+	new_test_ext(Default::default()).execute_with(|| {
+		Initializer::on_new_session(false, 0, Vec::new().into_iter(), Some(Vec::new().into_iter()));
+
+		let v = BufferedSessionChanges::<Test>::get();
+		assert!(v.is_empty());
+
+		assert_eq!(session_info::EarliestStoredSession::<Test>::get(), 0);
+		assert!(session_info::Sessions::<Test>::get(0).is_some());
+	});
+}
+
+#[test]
+fn session_change_before_initialize_is_still_buffered_after() {
+	new_test_ext(Default::default()).execute_with(|| {
+		Initializer::on_new_session(false, 1, Vec::new().into_iter(), Some(Vec::new().into_iter()));
+
+		let now = System::block_number();
+		Initializer::on_initialize(now);
+
+		let v = BufferedSessionChanges::<Test>::get();
+		assert_eq!(v.len(), 1);
+	});
+}
+
+#[test]
+fn session_change_applied_on_finalize() {
+	new_test_ext(Default::default()).execute_with(|| {
+		Initializer::on_initialize(1);
+		Initializer::on_new_session(false, 1, Vec::new().into_iter(), Some(Vec::new().into_iter()));
+
+		Initializer::on_finalize(1);
+
+		assert!(BufferedSessionChanges::<Test>::get().is_empty());
+	});
+}
+
+#[test]
+fn sets_flag_on_initialize() {
+	new_test_ext(Default::default()).execute_with(|| {
+		Initializer::on_initialize(1);
+
+		assert!(HasInitialized::<Test>::get().is_some());
+	})
+}
+
+#[test]
+fn clears_flag_on_finalize() {
+	new_test_ext(Default::default()).execute_with(|| {
+		Initializer::on_initialize(1);
+		Initializer::on_finalize(1);
+
+		assert!(HasInitialized::<Test>::get().is_none());
+	})
+}
+
+#[test]
+fn scheduled_cleanup_performed() {
+	let a = ParaId::from(1312);
+	let b = ParaId::from(228);
+	let c = ParaId::from(123);
+
+	let mock_genesis = crate::paras::ParaGenesisArgs {
+		para_kind: ParaKind::Teyrchain,
+		genesis_head: HeadData(vec![4, 5, 6]),
+		validation_code: dummy_validation_code(),
+	};
+
+	new_test_ext(MockGenesisConfig {
+		configuration: crate::configuration::GenesisConfig {
+			config: crate::configuration::HostConfiguration {
+				max_downward_message_size: 1024,
+				..Default::default()
+			},
+		},
+		paras: crate::paras::GenesisConfig {
+			paras: vec![
+				(a, mock_genesis.clone()),
+				(b, mock_genesis.clone()),
+				(c, mock_genesis.clone()),
+			],
+			..Default::default()
+		},
+		..Default::default()
+	})
+	.execute_with(|| {
+		// enqueue downward messages to A, B and C.
+		assert_ok!(Dmp::queue_downward_message(
+			&configuration::ActiveConfig::<Test>::get(),
+			a,
+			vec![1, 2, 3]
+		));
+		assert_ok!(Dmp::queue_downward_message(
+			&configuration::ActiveConfig::<Test>::get(),
+			b,
+			vec![4, 5, 6]
+		));
+		assert_ok!(Dmp::queue_downward_message(
+			&configuration::ActiveConfig::<Test>::get(),
+			c,
+			vec![7, 8, 9]
+		));
+
+		assert_ok!(Paras::schedule_para_cleanup(a));
+		assert_ok!(Paras::schedule_para_cleanup(b));
+
+		// Apply session 2 in the future
+		Initializer::apply_new_session(2, vec![], vec![]);
+
+		assert!(Dmp::dmq_contents(a).is_empty());
+		assert!(Dmp::dmq_contents(b).is_empty());
+		assert!(!Dmp::dmq_contents(c).is_empty());
+	});
+}