// This file is part of Bizinikiwi.

// Copyright (C) Parity Technologies (UK) Ltd. and Dijital Kurdistan Tech Institute
// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0

// This program 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.

// This program 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 this program. If not, see <https://www.gnu.org/licenses/>.

use codec::{Decode, Encode, Joiner};
use pezframe_support::{
	dispatch::{DispatchClass, GetDispatchInfo},
	traits::Currency,
	weights::Weight,
};
use pezframe_system::{self, AccountInfo, DispatchEventInfo, EventRecord, Phase};
use pezkuwi_sdk::*;
use pezsp_core::{storage::well_known_keys, traits::Externalities};
use pezsp_runtime::{
	traits::Hash as HashT, transaction_validity::InvalidTransaction, ApplyExtrinsicResult,
};

use pez_kitchensink_runtime::{
	constants::{currency::*, time::SLOT_DURATION},
	Balances, CheckedExtrinsic, Header, Runtime, RuntimeCall, RuntimeEvent, System,
	TransactionPayment, Treasury, UncheckedExtrinsic,
};
use pez_node_primitives::{Balance, Hash};
use pez_node_testing::keyring::*;
use pretty_assertions::assert_eq;
use wat;

pub mod common;
use self::common::{sign, *};

/// The wasm runtime binary which hasn't undergone the compacting process.
///
/// The idea here is to pass it as the current runtime code to the executor so the executor will
/// have to execute provided wasm code instead of the native equivalent. This trick is used to
/// test code paths that differ between native and wasm versions.
pub fn bloaty_code_unwrap() -> &'static [u8] {
	pez_kitchensink_runtime::WASM_BINARY_BLOATY.expect(
		"Development wasm binary is not available. \
											 Testing is only supported with the flag disabled.",
	)
}

/// Default transfer fee. This will use the same logic that is implemented in transaction-payment
/// module.
///
/// Note that reads the multiplier from storage directly, hence to get the fee of `extrinsic`
/// at block `n`, it must be called prior to executing block `n` to do the calculation with the
/// correct multiplier.
fn transfer_fee(extrinsic: &UncheckedExtrinsic) -> Balance {
	let mut info = default_transfer_call().get_dispatch_info();
	info.extension_weight = extrinsic.0.extension_weight();
	TransactionPayment::compute_fee(extrinsic.encode().len() as u32, &info, 0)
}

/// Default transfer fee, same as `transfer_fee`, but with a weight refund factored in.
fn transfer_fee_with_refund(extrinsic: &UncheckedExtrinsic, weight_refund: Weight) -> Balance {
	let mut info = default_transfer_call().get_dispatch_info();
	info.extension_weight = extrinsic.0.extension_weight();
	let post_info = (Some(info.total_weight().saturating_sub(weight_refund)), info.pays_fee).into();
	TransactionPayment::compute_actual_fee(extrinsic.encode().len() as u32, &info, &post_info, 0)
}

fn xt() -> UncheckedExtrinsic {
	sign(CheckedExtrinsic {
		format: pezsp_runtime::generic::ExtrinsicFormat::Signed(alice(), tx_ext(0, 0)),
		function: RuntimeCall::Balances(default_transfer_call()),
	})
}

fn set_heap_pages<E: Externalities>(ext: &mut E, heap_pages: u64) {
	ext.place_storage(well_known_keys::HEAP_PAGES.to_vec(), Some(heap_pages.encode()));
}

fn changes_trie_block() -> (Vec<u8>, Hash) {
	let time = 42 * 1000;
	construct_block(
		&mut new_test_ext(compact_code_unwrap()),
		1,
		GENESIS_HASH.into(),
		vec![
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Bare,
				function: RuntimeCall::Timestamp(pezpallet_timestamp::Call::set { now: time }),
			},
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Signed(alice(), tx_ext(0, 0)),
				function: RuntimeCall::Balances(pezpallet_balances::Call::transfer_allow_death {
					dest: bob().into(),
					value: 69 * DOLLARS,
				}),
			},
		],
		(time / SLOT_DURATION).into(),
	)
}

/// block 1 and 2 must be created together to ensure transactions are only signed once (since they
/// are not guaranteed to be deterministic) and to ensure that the correct state is propagated
/// from block1's execution to block2 to derive the correct storage_root.
fn blocks() -> ((Vec<u8>, Hash), (Vec<u8>, Hash)) {
	let mut t = new_test_ext(compact_code_unwrap());
	let time1 = 42 * 1000;
	let block1 = construct_block(
		&mut t,
		1,
		GENESIS_HASH.into(),
		vec![
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Bare,
				function: RuntimeCall::Timestamp(pezpallet_timestamp::Call::set { now: time1 }),
			},
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Signed(alice(), tx_ext(0, 0)),
				function: RuntimeCall::Balances(pezpallet_balances::Call::transfer_allow_death {
					dest: bob().into(),
					value: 69 * DOLLARS,
				}),
			},
		],
		(time1 / SLOT_DURATION).into(),
	);
	let time2 = 52 * 1000;
	let block2 = construct_block(
		&mut t,
		2,
		block1.1,
		vec![
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Bare,
				function: RuntimeCall::Timestamp(pezpallet_timestamp::Call::set { now: time2 }),
			},
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Signed(bob(), tx_ext(0, 0)),
				function: RuntimeCall::Balances(pezpallet_balances::Call::transfer_allow_death {
					dest: alice().into(),
					value: 5 * DOLLARS,
				}),
			},
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Signed(alice(), tx_ext(1, 0)),
				function: RuntimeCall::Balances(pezpallet_balances::Call::transfer_allow_death {
					dest: bob().into(),
					value: 15 * DOLLARS,
				}),
			},
		],
		(time2 / SLOT_DURATION).into(),
	);

	// session change => consensus authorities change => authorities change digest item appears
	let digest = Header::decode(&mut &block2.0[..]).unwrap().digest;
	assert_eq!(digest.logs().len(), 2 /* Just babe and BEEFY slots */);

	(block1, block2)
}

fn block_with_size(time: u64, nonce: u32, size: usize) -> (Vec<u8>, Hash) {
	construct_block(
		&mut new_test_ext(compact_code_unwrap()),
		1,
		GENESIS_HASH.into(),
		vec![
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Bare,
				function: RuntimeCall::Timestamp(pezpallet_timestamp::Call::set {
					now: time * 1000,
				}),
			},
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Signed(alice(), tx_ext(nonce, 0)),
				function: RuntimeCall::System(pezframe_system::Call::remark {
					remark: vec![0; size],
				}),
			},
		],
		(time * 1000 / SLOT_DURATION).into(),
	)
}

#[test]
fn panic_execution_with_foreign_code_gives_error() {
	let mut t = new_test_ext(bloaty_code_unwrap());
	t.insert(
		<pezframe_system::Account<Runtime>>::hashed_key_for(alice()),
		AccountInfo::<<Runtime as pezframe_system::Config>::Nonce, _> {
			providers: 1,
			data: (69u128, 0u128, 0u128, 1u128 << 127),
			..Default::default()
		}
		.encode(),
	);
	t.insert(<pezpallet_balances::TotalIssuance<Runtime>>::hashed_key().to_vec(), 69_u128.encode());
	t.insert(<pezframe_system::BlockHash<Runtime>>::hashed_key_for(0), vec![0u8; 32]);

	let r = executor_call(&mut t, "Core_initialize_block", &vec![].and(&from_block_number(1u32))).0;
	assert!(r.is_ok());
	let v = executor_call(&mut t, "BlockBuilder_apply_extrinsic", &vec![].and(&xt()))
		.0
		.unwrap();
	let r = ApplyExtrinsicResult::decode(&mut &v[..]).unwrap();
	assert_eq!(r, Err(InvalidTransaction::Payment.into()));
}

#[test]
fn bad_extrinsic_with_native_equivalent_code_gives_error() {
	let mut t = new_test_ext(compact_code_unwrap());
	t.insert(
		<pezframe_system::Account<Runtime>>::hashed_key_for(alice()),
		AccountInfo::<<Runtime as pezframe_system::Config>::Nonce, _> {
			providers: 1,
			data: (69u128, 0u128, 0u128, 1u128 << 127),
			..Default::default()
		}
		.encode(),
	);
	t.insert(<pezpallet_balances::TotalIssuance<Runtime>>::hashed_key().to_vec(), 69u128.encode());
	t.insert(<pezframe_system::BlockHash<Runtime>>::hashed_key_for(0), vec![0u8; 32]);

	let r = executor_call(&mut t, "Core_initialize_block", &vec![].and(&from_block_number(1u32))).0;
	assert!(r.is_ok());
	let v = executor_call(&mut t, "BlockBuilder_apply_extrinsic", &vec![].and(&xt()))
		.0
		.unwrap();
	let r = ApplyExtrinsicResult::decode(&mut &v[..]).unwrap();
	assert_eq!(r, Err(InvalidTransaction::Payment.into()));
}

#[test]
fn successful_execution_with_native_equivalent_code_gives_ok() {
	let mut t = new_test_ext(compact_code_unwrap());
	t.insert(
		<pezframe_system::Account<Runtime>>::hashed_key_for(alice()),
		AccountInfo::<<Runtime as pezframe_system::Config>::Nonce, _> {
			providers: 1,
			data: (111 * DOLLARS, 0u128, 0u128, 1u128 << 127),
			..Default::default()
		}
		.encode(),
	);
	t.insert(
		<pezframe_system::Account<Runtime>>::hashed_key_for(bob()),
		AccountInfo::<
			<Runtime as pezframe_system::Config>::Nonce,
			<Runtime as pezframe_system::Config>::AccountData,
		>::default()
		.encode(),
	);
	t.insert(
		<pezpallet_balances::TotalIssuance<Runtime>>::hashed_key().to_vec(),
		(111 * DOLLARS).encode(),
	);
	t.insert(<pezframe_system::BlockHash<Runtime>>::hashed_key_for(0), vec![0u8; 32]);

	let r = executor_call(&mut t, "Core_initialize_block", &vec![].and(&from_block_number(1u32))).0;
	assert!(r.is_ok());

	let weight_refund = Weight::zero();
	let fees_after_refund = t.execute_with(|| transfer_fee_with_refund(&xt(), weight_refund));

	let r = executor_call(&mut t, "BlockBuilder_apply_extrinsic", &vec![].and(&xt())).0;
	assert!(r.is_ok());

	t.execute_with(|| {
		assert_eq!(Balances::total_balance(&alice()), 42 * DOLLARS - fees_after_refund);
		assert_eq!(Balances::total_balance(&bob()), 69 * DOLLARS);
	});
}

#[test]
fn successful_execution_with_foreign_code_gives_ok() {
	let mut t = new_test_ext(bloaty_code_unwrap());
	t.insert(
		<pezframe_system::Account<Runtime>>::hashed_key_for(alice()),
		AccountInfo::<<Runtime as pezframe_system::Config>::Nonce, _> {
			providers: 1,
			data: (111 * DOLLARS, 0u128, 0u128, 1u128 << 127),
			..Default::default()
		}
		.encode(),
	);
	t.insert(
		<pezframe_system::Account<Runtime>>::hashed_key_for(bob()),
		AccountInfo::<
			<Runtime as pezframe_system::Config>::Nonce,
			<Runtime as pezframe_system::Config>::AccountData,
		>::default()
		.encode(),
	);
	t.insert(
		<pezpallet_balances::TotalIssuance<Runtime>>::hashed_key().to_vec(),
		(111 * DOLLARS).encode(),
	);
	t.insert(<pezframe_system::BlockHash<Runtime>>::hashed_key_for(0), vec![0u8; 32]);

	let r = executor_call(&mut t, "Core_initialize_block", &vec![].and(&from_block_number(1u32))).0;
	assert!(r.is_ok());

	let weight_refund = Weight::zero();
	let fees_after_refund = t.execute_with(|| transfer_fee_with_refund(&xt(), weight_refund));

	let r = executor_call(&mut t, "BlockBuilder_apply_extrinsic", &vec![].and(&xt())).0;
	assert!(r.is_ok());

	t.execute_with(|| {
		assert_eq!(Balances::total_balance(&alice()), 42 * DOLLARS - fees_after_refund);
		assert_eq!(Balances::total_balance(&bob()), 69 * DOLLARS);
	});
}

#[test]
fn full_native_block_import_works() {
	let mut t = new_test_ext(compact_code_unwrap());

	let (block1, block2) = blocks();

	let mut alice_last_known_balance: Balance = Default::default();
	let mut fees = t.execute_with(|| transfer_fee(&xt()));
	let extension_weight = xt().0.extension_weight();
	let weight_refund = Weight::zero();
	let fees_after_refund = t.execute_with(|| transfer_fee_with_refund(&xt(), weight_refund));

	let transfer_weight = default_transfer_call().get_dispatch_info().call_weight.saturating_add(
		<Runtime as pezframe_system::Config>::BlockWeights::get()
			.get(DispatchClass::Normal)
			.base_extrinsic,
	);
	let timestamp_weight = pezpallet_timestamp::Call::set::<Runtime> { now: Default::default() }
		.get_dispatch_info()
		.call_weight
		.saturating_add(
			<Runtime as pezframe_system::Config>::BlockWeights::get()
				.get(DispatchClass::Mandatory)
				.base_extrinsic,
		);

	executor_call(&mut t, "Core_execute_block", &block1.0).0.unwrap();

	t.execute_with(|| {
		assert_eq!(Balances::total_balance(&alice()), 42 * DOLLARS - fees_after_refund);
		assert_eq!(Balances::total_balance(&bob()), 169 * DOLLARS);
		alice_last_known_balance = Balances::total_balance(&alice());
		let events = vec![
			EventRecord {
				phase: Phase::ApplyExtrinsic(0),
				event: RuntimeEvent::System(pezframe_system::Event::ExtrinsicSuccess {
					dispatch_info: DispatchEventInfo {
						weight: timestamp_weight,
						class: DispatchClass::Mandatory,
						pays_fee: Default::default(),
					},
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::Balances(pezpallet_balances::Event::Withdraw {
					who: alice().into(),
					amount: fees,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::Balances(pezpallet_balances::Event::Transfer {
					from: alice().into(),
					to: bob().into(),
					amount: 69 * DOLLARS,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::Balances(pezpallet_balances::Event::Deposit {
					who: pezpallet_treasury::Pezpallet::<Runtime>::account_id(),
					amount: fees_after_refund,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::TransactionPayment(
					pezpallet_transaction_payment::Event::TransactionFeePaid {
						who: alice().into(),
						actual_fee: fees_after_refund,
						tip: 0,
					},
				),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::System(pezframe_system::Event::ExtrinsicSuccess {
					dispatch_info: DispatchEventInfo {
						weight: transfer_weight
							.saturating_add(extension_weight.saturating_sub(weight_refund)),
						..Default::default()
					},
				}),
				topics: vec![],
			},
		];
		let filtered_events: Vec<_> = System::events()
			.into_iter()
			.filter(|ev| {
				!matches!(
					ev.event,
					RuntimeEvent::VoterList(
						pezpallet_bags_list::Event::<Runtime, _>::ScoreUpdated { .. }
					)
				)
			})
			.collect();

		assert_eq!(filtered_events, events);
	});

	fees = t.execute_with(|| transfer_fee(&xt()));
	let pot = t.execute_with(|| Treasury::pot());
	let extension_weight = xt().0.extension_weight();
	let weight_refund = Weight::zero();
	let fees_after_refund = t.execute_with(|| transfer_fee_with_refund(&xt(), weight_refund));

	executor_call(&mut t, "Core_execute_block", &block2.0).0.unwrap();

	t.execute_with(|| {
		assert_eq!(
			Balances::total_balance(&alice()),
			alice_last_known_balance - 10 * DOLLARS - fees_after_refund,
		);
		assert_eq!(Balances::total_balance(&bob()), 179 * DOLLARS - fees_after_refund);
		let events = vec![
			EventRecord {
				phase: Phase::Initialization,
				event: RuntimeEvent::Treasury(pezpallet_treasury::Event::UpdatedInactive {
					reactivated: 0,
					deactivated: pot,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(0),
				event: RuntimeEvent::System(pezframe_system::Event::ExtrinsicSuccess {
					dispatch_info: DispatchEventInfo {
						weight: timestamp_weight,
						class: DispatchClass::Mandatory,
						pays_fee: Default::default(),
					},
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::Balances(pezpallet_balances::Event::Withdraw {
					who: bob().into(),
					amount: fees,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::Balances(pezpallet_balances::Event::Transfer {
					from: bob().into(),
					to: alice().into(),
					amount: 5 * DOLLARS,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::Balances(pezpallet_balances::Event::Deposit {
					who: pezpallet_treasury::Pezpallet::<Runtime>::account_id(),
					amount: fees_after_refund,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::TransactionPayment(
					pezpallet_transaction_payment::Event::TransactionFeePaid {
						who: bob().into(),
						actual_fee: fees_after_refund,
						tip: 0,
					},
				),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(1),
				event: RuntimeEvent::System(pezframe_system::Event::ExtrinsicSuccess {
					dispatch_info: DispatchEventInfo {
						weight: transfer_weight
							.saturating_add(extension_weight.saturating_sub(weight_refund)),
						..Default::default()
					},
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(2),
				event: RuntimeEvent::Balances(pezpallet_balances::Event::Withdraw {
					who: alice().into(),
					amount: fees,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(2),
				event: RuntimeEvent::Balances(pezpallet_balances::Event::Transfer {
					from: alice().into(),
					to: bob().into(),
					amount: 15 * DOLLARS,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(2),
				event: RuntimeEvent::Balances(pezpallet_balances::Event::Deposit {
					who: pezpallet_treasury::Pezpallet::<Runtime>::account_id(),
					amount: fees_after_refund,
				}),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(2),
				event: RuntimeEvent::TransactionPayment(
					pezpallet_transaction_payment::Event::TransactionFeePaid {
						who: alice().into(),
						actual_fee: fees_after_refund,
						tip: 0,
					},
				),
				topics: vec![],
			},
			EventRecord {
				phase: Phase::ApplyExtrinsic(2),
				event: RuntimeEvent::System(pezframe_system::Event::ExtrinsicSuccess {
					dispatch_info: DispatchEventInfo {
						weight: transfer_weight
							.saturating_add(extension_weight.saturating_sub(weight_refund)),
						..Default::default()
					},
				}),
				topics: vec![],
			},
		];
		let all_events = System::events();
		// Ensure that all expected events (`events`) are present in the full event log
		// (`all_events`). We use this instead of strict equality since some events (like
		// VoterList::ScoreUpdated) may be emitted non-deterministically depending on runtime
		// internals or auto-rebagging logic.
		for expected_event in &events {
			assert!(
				all_events.contains(expected_event),
				"Expected event {:?} not found in actual events",
				expected_event
			);
		}
	});
}

#[test]
fn full_wasm_block_import_works() {
	let mut t = new_test_ext(compact_code_unwrap());

	let (block1, block2) = blocks();

	let mut alice_last_known_balance: Balance = Default::default();
	let weight_refund = Weight::zero();
	let fees_after_refund = t.execute_with(|| transfer_fee_with_refund(&xt(), weight_refund));

	executor_call(&mut t, "Core_execute_block", &block1.0).0.unwrap();

	t.execute_with(|| {
		assert_eq!(Balances::total_balance(&alice()), 42 * DOLLARS - fees_after_refund);
		assert_eq!(Balances::total_balance(&bob()), 169 * DOLLARS);
		alice_last_known_balance = Balances::total_balance(&alice());
	});

	let weight_refund = Weight::zero();
	let fees_after_refund = t.execute_with(|| transfer_fee_with_refund(&xt(), weight_refund));

	executor_call(&mut t, "Core_execute_block", &block2.0).0.unwrap();

	t.execute_with(|| {
		assert_eq!(
			Balances::total_balance(&alice()),
			alice_last_known_balance - 10 * DOLLARS - fees_after_refund,
		);
		assert_eq!(Balances::total_balance(&bob()), 179 * DOLLARS - 1 * fees_after_refund);
	});
}

const CODE_TRANSFER: &str = r#"
(module
;; seal_call(
;;    callee_ptr: u32,
;;    callee_len: u32,
;;    gas: u64,
;;    value_ptr: u32,
;;    value_len: u32,
;;    input_data_ptr: u32,
;;    input_data_len: u32,
;;    output_ptr: u32,
;;    output_len_ptr: u32
;; ) -> u32
(import "seal0" "seal_call" (func $seal_call (param i32 i32 i64 i32 i32 i32 i32 i32 i32) (result i32)))
(import "seal0" "seal_input" (func $seal_input (param i32 i32)))
(import "env" "memory" (memory 1 1))
(func (export "deploy")
)
(func (export "call")
	(block $fail
		;; Load input data to contract memory
		(call $seal_input
			(i32.const 0)
			(i32.const 52)
		)

		;; fail if the input size is not != 4
		(br_if $fail
			(i32.ne
				(i32.const 4)
				(i32.load (i32.const 52))
			)
		)

		(br_if $fail
			(i32.ne
				(i32.load8_u (i32.const 0))
				(i32.const 0)
			)
		)
		(br_if $fail
			(i32.ne
				(i32.load8_u (i32.const 1))
				(i32.const 1)
			)
		)
		(br_if $fail
			(i32.ne
				(i32.load8_u (i32.const 2))
				(i32.const 2)
			)
		)
		(br_if $fail
			(i32.ne
				(i32.load8_u (i32.const 3))
				(i32.const 3)
			)
		)

		(drop
			(call $seal_call
				(i32.const 4)  ;; Pointer to "callee" address.
				(i32.const 32)  ;; Length of "callee" address.
				(i64.const 0)  ;; How much gas to devote for the execution. 0 = all.
				(i32.const 36)  ;; Pointer to the buffer with value to transfer
				(i32.const 16)   ;; Length of the buffer with value to transfer.
				(i32.const 0)   ;; Pointer to input data buffer address
				(i32.const 0)   ;; Length of input data buffer
				(i32.const 4294967295) ;; u32 max value is the sentinel value: do not copy output
				(i32.const 0) ;; Length is ignored in this case
			)
		)

		(return)
	)
	unreachable
)
;; Destination AccountId to transfer the funds.
;; Represented by H256 (32 bytes long) in little endian.
(data (i32.const 4)
	"\09\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00"
	"\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00"
	"\00\00\00\00"
)
;; Amount of value to transfer.
;; Represented by u128 (16 bytes long) in little endian.
(data (i32.const 36)
	"\06\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00\00"
	"\00\00"
)
;; Length of the input buffer
(data (i32.const 52) "\04")
)
"#;

#[test]
fn deploying_wasm_contract_should_work() {
	let transfer_code = wat::parse_str(CODE_TRANSFER).unwrap();
	let transfer_ch = <Runtime as pezframe_system::Config>::Hashing::hash(&transfer_code);

	let addr = pezpallet_contracts::Pezpallet::<Runtime>::contract_address(
		&charlie(),
		&transfer_ch,
		&[],
		&[],
	);

	let time = 42 * 1000;
	let b = construct_block(
		&mut new_test_ext(compact_code_unwrap()),
		1,
		GENESIS_HASH.into(),
		vec![
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Bare,
				function: RuntimeCall::Timestamp(pezpallet_timestamp::Call::set { now: time }),
			},
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Signed(charlie(), tx_ext(0, 0)),
				function: RuntimeCall::Contracts(
					pezpallet_contracts::Call::instantiate_with_code::<Runtime> {
						value: 0,
						gas_limit: Weight::from_parts(500_000_000, 0),
						storage_deposit_limit: None,
						code: transfer_code,
						data: Vec::new(),
						salt: Vec::new(),
					},
				),
			},
			CheckedExtrinsic {
				format: pezsp_runtime::generic::ExtrinsicFormat::Signed(charlie(), tx_ext(1, 0)),
				function: RuntimeCall::Contracts(pezpallet_contracts::Call::call::<Runtime> {
					dest: pezsp_runtime::MultiAddress::Id(addr.clone()),
					value: 10,
					gas_limit: Weight::from_parts(500_000_000, 0),
					storage_deposit_limit: None,
					data: vec![0x00, 0x01, 0x02, 0x03],
				}),
			},
		],
		(time / SLOT_DURATION).into(),
	);

	let mut t = new_test_ext(compact_code_unwrap());

	executor_call(&mut t, "Core_execute_block", &b.0).0.unwrap();

	t.execute_with(|| {
		// Verify that the contract does exist by querying some of its storage items
		// It does not matter that the storage item itself does not exist.
		assert!(&pezpallet_contracts::Pezpallet::<Runtime>::get_storage(addr, vec![]).is_ok());
	});
}

#[test]
fn wasm_big_block_import_fails() {
	let mut t = new_test_ext(compact_code_unwrap());

	set_heap_pages(&mut t.ext(), 4);

	let result = executor_call(&mut t, "Core_execute_block", &block_with_size(42, 0, 120_000).0).0;
	assert!(result.is_err()); // Err(Wasmi(Trap(Trap { kind: Host(AllocatorOutOfSpace) })))
}

#[test]
fn native_big_block_import_succeeds() {
	let mut t = new_test_ext(compact_code_unwrap());

	executor_call(&mut t, "Core_execute_block", &block_with_size(42, 0, 120_000).0)
		.0
		.unwrap();
}

#[test]
fn native_big_block_import_fails_on_fallback() {
	let mut t = new_test_ext(compact_code_unwrap());

	// We set the heap pages to 8 because we know that should give an OOM in WASM with the given
	// block.
	set_heap_pages(&mut t.ext(), 8);

	assert!(executor_call(&mut t, "Core_execute_block", &block_with_size(42, 0, 120_000).0)
		.0
		.is_err());
}

#[test]
fn panic_execution_gives_error() {
	let mut t = new_test_ext(bloaty_code_unwrap());
	t.insert(
		<pezframe_system::Account<Runtime>>::hashed_key_for(alice()),
		AccountInfo::<<Runtime as pezframe_system::Config>::Nonce, _> {
			data: (0 * DOLLARS, 0u128, 0u128, 0u128),
			..Default::default()
		}
		.encode(),
	);
	t.insert(<pezpallet_balances::TotalIssuance<Runtime>>::hashed_key().to_vec(), 0_u128.encode());
	t.insert(<pezframe_system::BlockHash<Runtime>>::hashed_key_for(0), vec![0u8; 32]);

	let r = executor_call(&mut t, "Core_initialize_block", &vec![].and(&from_block_number(1u32))).0;
	assert!(r.is_ok());
	let r = executor_call(&mut t, "BlockBuilder_apply_extrinsic", &vec![].and(&xt()))
		.0
		.unwrap();
	let r = ApplyExtrinsicResult::decode(&mut &r[..]).unwrap();
	assert_eq!(r, Err(InvalidTransaction::Payment.into()));
}

#[test]
fn successful_execution_gives_ok() {
	let mut t = new_test_ext(compact_code_unwrap());
	t.insert(
		<pezframe_system::Account<Runtime>>::hashed_key_for(alice()),
		AccountInfo::<<Runtime as pezframe_system::Config>::Nonce, _> {
			providers: 1,
			data: (111 * DOLLARS, 0u128, 0u128, 1u128 << 127),
			..Default::default()
		}
		.encode(),
	);
	t.insert(
		<pezframe_system::Account<Runtime>>::hashed_key_for(bob()),
		AccountInfo::<
			<Runtime as pezframe_system::Config>::Nonce,
			<Runtime as pezframe_system::Config>::AccountData,
		>::default()
		.encode(),
	);
	t.insert(
		<pezpallet_balances::TotalIssuance<Runtime>>::hashed_key().to_vec(),
		(111 * DOLLARS).encode(),
	);
	t.insert(<pezframe_system::BlockHash<Runtime>>::hashed_key_for(0), vec![0u8; 32]);

	let r = executor_call(&mut t, "Core_initialize_block", &vec![].and(&from_block_number(1u32))).0;
	assert!(r.is_ok());
	t.execute_with(|| {
		assert_eq!(Balances::total_balance(&alice()), 111 * DOLLARS);
	});

	let weight_refund = Weight::zero();
	let fees_after_refund = t.execute_with(|| transfer_fee_with_refund(&xt(), weight_refund));

	let r = executor_call(&mut t, "BlockBuilder_apply_extrinsic", &vec![].and(&xt()))
		.0
		.unwrap();
	ApplyExtrinsicResult::decode(&mut &r[..])
		.unwrap()
		.expect("Extrinsic could not be applied")
		.expect("Extrinsic failed");

	t.execute_with(|| {
		assert_eq!(Balances::total_balance(&alice()), 42 * DOLLARS - fees_after_refund);
		assert_eq!(Balances::total_balance(&bob()), 69 * DOLLARS);
	});
}

#[test]
fn should_import_block_with_test_client() {
	use pez_node_testing::client::{
		pezsp_consensus::BlockOrigin, ClientBlockImportExt, TestClientBuilder, TestClientBuilderExt,
	};

	let client = TestClientBuilder::new().build();
	let block1 = changes_trie_block();
	let block_data = block1.0;
	let block = pez_node_primitives::Block::decode(&mut &block_data[..]).unwrap();

	futures::executor::block_on(client.import(BlockOrigin::Own, block)).unwrap();
}

#[test]
fn default_config_as_json_works() {
	let mut t = new_test_ext(compact_code_unwrap());
	let r = executor_call(
		&mut t,
		"GenesisBuilder_get_preset",
		&None::<&pezsp_genesis_builder::PresetId>.encode(),
	)
	.0
	.unwrap();
	let r = Option::<Vec<u8>>::decode(&mut &r[..])
		.unwrap()
		.expect("default config is there");
	let json = String::from_utf8(r.into()).expect("returned value is json. qed.");
	let expected = include_str!("res/default_genesis_config.json").to_string();

	assert_eq!(
		serde_json::from_str::<serde_json::Value>(&expected).unwrap(),
		serde_json::from_str::<serde_json::Value>(&json).unwrap()
	);
}