Add tests for Democracy Pallet weights

runtime/polkadot/tests/weights_and_fees.rs

@@ -28,6 +28,8 @@ use primitives::AccountId;
 use polkadot_runtime::{self, Runtime};
 use polkadot_runtime::constants::{currency::*, fee::*};
 use runtime_common::{MaximumBlockWeight, ExtrinsicBaseWeight};
+
+use democracy::Call as DemocracyCall;
 use session::Call as SessionCall;
 use staking::Call as StakingCall;
 use system::Call as SystemCall;
@@ -264,3 +266,36 @@ fn weight_of_session_purge_keys_is_correct() {
 
 	assert_eq!(weight, expected_weight);
 }
+
+#[test]
+fn weight_of_democracy_propose_is_correct() {
+	// #[weight = 5_000_000_000]
+	let expected_weight = 5_000_000_000;
+	let weight = DemocracyCall::propose::<Runtime>(Default::default(), Default::default()).get_dispatch_info().weight;
+
+	assert_eq!(weight, expected_weight);
+}
+
+#[test]
+fn weight_of_democracy_vote_is_correct() {
+	use democracy::AccountVote;
+	let vote = AccountVote::Standard {
+		vote: Default::default(),
+		balance: Default::default(),
+	};
+
+	// #[weight = 200_000_000]
+	let expected_weight = 200_000_000;
+	let weight = DemocracyCall::vote::<Runtime>(Default::default(), vote).get_dispatch_info().weight;
+
+	assert_eq!(weight, expected_weight);
+}
+
+#[test]
+fn weight_of_democracy_enact_proposal_is_correct() {
+	// #[weight = T::MaximumBlockWeight::get()]
+	let expected_weight = MaximumBlockWeight::get();
+	let weight = DemocracyCall::enact_proposal::<Runtime>(Default::default(), Default::default()).get_dispatch_info().weight;
+
+	assert_eq!(weight, expected_weight);
+}