diff --git a/sugondat-chain/pallets/blobs/src/tests.rs b/sugondat-chain/pallets/blobs/src/tests.rs index 1ca5c3c9..182318e2 100644 --- a/sugondat-chain/pallets/blobs/src/tests.rs +++ b/sugondat-chain/pallets/blobs/src/tests.rs @@ -57,24 +57,19 @@ fn test_no_extrinsic_index() { }); } -macro_rules! submit_blobs { - ([blob_size] $blob_size: expr, [blobs_number] $n_blobs: expr) => { - let blob = get_blob($blob_size); - for i in 0..$n_blobs { +#[test] +fn test_max_blobs_reached() { + let max_blobs: u32 = ::MaxBlobs::get(); + + new_test_ext().execute_with(|| { + let blob = get_blob(1); + for i in 0..max_blobs { assert_ok!(Blobs::submit_blob( RuntimeOrigin::signed(alice()), i, blob.clone() )); } - }; -} - -#[test] -fn test_max_blobs_reached() { - let max_blobs: u32 = ::MaxBlobs::get(); - new_test_ext().execute_with(|| { - submit_blobs!([blob_size] 1, [blobs_number] max_blobs); }); } @@ -82,8 +77,18 @@ fn test_max_blobs_reached() { #[should_panic = "Maximum blob limit exceeded"] fn test_max_blobs_exceeded() { let max_blobs: u32 = ::MaxBlobs::get(); + new_test_ext().execute_with(|| { - submit_blobs!([blob_size] 1, [blobs_number] max_blobs + 1); + let blob = get_blob(1); + for i in 0..max_blobs { + assert_ok!(Blobs::submit_blob( + RuntimeOrigin::signed(alice()), + i, + blob.clone() + )); + } + + let _ = Blobs::submit_blob(RuntimeOrigin::signed(alice()), 0, blob.clone()); }); } @@ -94,7 +99,14 @@ fn test_max_total_blob_size_reached() { let blobs_needed = max_total_blobs_size / max_blob_size; new_test_ext().execute_with(|| { - submit_blobs!([blob_size] max_blob_size, [blobs_number] blobs_needed); + let blob = get_blob(max_blob_size); + for i in 0..blobs_needed { + assert_ok!(Blobs::submit_blob( + RuntimeOrigin::signed(alice()), + i, + blob.clone() + )); + } }); } @@ -106,7 +118,15 @@ fn test_max_total_blob_size_exceeded() { let blobs_needed = max_total_blobs_size / max_blob_size; new_test_ext().execute_with(|| { - submit_blobs!([blob_size] max_blob_size, [blobs_number] blobs_needed + 1); + let blob = get_blob(max_blob_size); + for i in 0..blobs_needed { + assert_ok!(Blobs::submit_blob( + RuntimeOrigin::signed(alice()), + i, + blob.clone() + )); + } + let _ = Blobs::submit_blob(RuntimeOrigin::signed(alice()), 0, blob.clone()); }); }