waalge/aiken
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow to set initial budget

Browse Source
This commit is contained in:
alessandrokonrad
2022-09-19 12:02:41 +02:00
parent 83bac598de
commit 1148863a43
4 changed files with 103 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
crates/uplc/src/tx/eval.rs
View File

@@ -575,6 +575,7 @@ pub fn eval_redeemer(
redeemer: &Redeemer,
lookup_table: &DataLookupTable,
cost_mdls_opt: Option<&CostMdls>,
initial_budget: Option<&ExBudget>,
) -> Result<Redeemer, Error> {
let purpose = get_script_purpose(
redeemer,
@@ -612,7 +613,7 @@ pub fn eval_redeemer(
return Err(Error::V1CostModelNotFound);
};
program.eval_as(&Language::PlutusV1, costs)
program.eval_as(&Language::PlutusV1, costs, initial_budget)
} else {
program.eval_v1()
};
@@ -658,7 +659,7 @@ pub fn eval_redeemer(
return Err(Error::V2CostModelNotFound);
};
program.eval_as(&Language::PlutusV2, costs)
program.eval_as(&Language::PlutusV2, costs, initial_budget)
} else {
program.eval()
};
@@ -706,7 +707,7 @@ pub fn eval_redeemer(
return Err(Error::V1CostModelNotFound);
};
program.eval_as(&Language::PlutusV1, costs)
program.eval_as(&Language::PlutusV1, costs, initial_budget)
} else {
program.eval_v1()
};
@@ -751,7 +752,7 @@ pub fn eval_redeemer(
return Err(Error::V2CostModelNotFound);
};
program.eval_as(&Language::PlutusV2, costs)
program.eval_as(&Language::PlutusV2, costs, initial_budget)
} else {
program.eval()
};

80
crates/uplc/src/tx/tests.rs
View File

@@ -5,6 +5,8 @@ use pallas_primitives::{
};
use pallas_traverse::{Era, MultiEraTx};
use crate::machine::cost_model::ExBudget;
use super::{eval_phase_two, ResolvedInput, SlotConfig};
#[test]
@@ -226,13 +228,25 @@ fn test_eval() {
plutus_v2: Some(costs),
};
let initial_budget = ExBudget {
cpu: 10000000000,
mem: 14000000,
};
let multi_era_tx = MultiEraTx::decode(Era::Babbage, &tx_bytes)
.or_else(|_| MultiEraTx::decode(Era::Alonzo, &tx_bytes))
.unwrap();
match multi_era_tx {
MultiEraTx::Babbage(tx) => {
let redeemers =
eval_phase_two(&tx, &utxos, Some(&cost_mdl), &slot_config, false).unwrap();
let redeemers = eval_phase_two(
&tx,
&utxos,
Some(&cost_mdl),
Some(&initial_budget),
&slot_config,
false,
)
.unwrap();
assert_eq!(redeemers.len(), 1)
}
@@ -461,13 +475,25 @@ fn test_eval_1() {
plutus_v2: Some(costs),
};
let initial_budget = ExBudget {
cpu: 10000000000,
mem: 14000000,
};
let multi_era_tx = MultiEraTx::decode(Era::Babbage, &tx_bytes)
.or_else(|_| MultiEraTx::decode(Era::Alonzo, &tx_bytes))
.unwrap();
match multi_era_tx {
MultiEraTx::Babbage(tx) => {
let redeemers =
eval_phase_two(&tx, &utxos, Some(&cost_mdl), &slot_config, false).unwrap();
let redeemers = eval_phase_two(
&tx,
&utxos,
Some(&cost_mdl),
Some(&initial_budget),
&slot_config,
false,
)
.unwrap();
println!("{:?}", redeemers.len());
}
@@ -531,13 +557,25 @@ fn test_eval_2() {
plutus_v2: None,
};
let initial_budget = ExBudget {
cpu: 10000000000,
mem: 14000000,
};
let multi_era_tx = MultiEraTx::decode(Era::Babbage, &tx_bytes)
.or_else(|_| MultiEraTx::decode(Era::Alonzo, &tx_bytes))
.unwrap();
match multi_era_tx {
MultiEraTx::Babbage(tx) => {
let redeemers =
eval_phase_two(&tx, &utxos, Some(&cost_mdl), &slot_config, false).unwrap();
let redeemers = eval_phase_two(
&tx,
&utxos,
Some(&cost_mdl),
Some(&initial_budget),
&slot_config,
false,
)
.unwrap();
println!("{:?}", redeemers.len());
}
@@ -598,13 +636,26 @@ fn eval_missing_redeemer() {
plutus_v2: None,
};
let initial_budget = ExBudget {
cpu: 10000000000,
mem: 14000000,
};
let multi_era_tx = MultiEraTx::decode(Era::Babbage, &tx_bytes)
.or_else(|_| MultiEraTx::decode(Era::Alonzo, &tx_bytes))
.unwrap();
match multi_era_tx {
MultiEraTx::Babbage(tx) => {
eval_phase_two(&tx, &utxos, Some(&cost_mdl), &slot_config, false).unwrap();
eval_phase_two(
&tx,
&utxos,
Some(&cost_mdl),
Some(&initial_budget),
&slot_config,
false,
)
.unwrap();
}
_ => unreachable!(),
};
@@ -663,13 +714,26 @@ fn eval_extraneous_redeemer() {
plutus_v2: None,
};
let initial_budget = ExBudget {
cpu: 10000000000,
mem: 14000000,
};
let multi_era_tx = MultiEraTx::decode(Era::Babbage, &tx_bytes)
.or_else(|_| MultiEraTx::decode(Era::Alonzo, &tx_bytes))
.unwrap();
match multi_era_tx {
MultiEraTx::Babbage(tx) => {
eval_phase_two(&tx, &utxos, Some(&cost_mdl), &slot_config, false).unwrap();
eval_phase_two(
&tx,
&utxos,
Some(&cost_mdl),
Some(&initial_budget),
&slot_config,
false,
)
.unwrap();
}
_ => unreachable!(),
};