From e9d8e1d31702e201f2b9c206b3dcc909ca3018b9 Mon Sep 17 00:00:00 2001
From: rvcas <x@rvcas.dev>
Date: Thu, 8 Dec 2022 10:57:13 -0500
Subject: [PATCH] feat: print budget consumed by test

---
 crates/project/src/error.rs        |  2 +-
 crates/project/src/lib.rs          | 34 ++++++++++++++++++++++++------
 examples/sample/validators/swap.ak |  2 +-
 3 files changed, 29 insertions(+), 9 deletions(-)

diff --git a/crates/project/src/error.rs b/crates/project/src/error.rs
index 46548e41..a1bc12f9 100644
--- a/crates/project/src/error.rs
+++ b/crates/project/src/error.rs
@@ -293,7 +293,7 @@ impl Diagnostic for Warning {
 
     fn code<'a>(&'a self) -> Option<Box<dyn Display + 'a>> {
         match self {
-            Warning::Type { .. } => Some(Box::new("aiken::typecheck")),
+            Warning::Type { .. } => Some(Box::new("aiken::check")),
         }
     }
 }
diff --git a/crates/project/src/lib.rs b/crates/project/src/lib.rs
index a1f9552a..0138bc49 100644
--- a/crates/project/src/lib.rs
+++ b/crates/project/src/lib.rs
@@ -25,7 +25,10 @@ use pallas::{
 use pallas_traverse::ComputeHash;
 use script::Script;
 use serde_json::json;
-use uplc::ast::{DeBruijn, Program};
+use uplc::{
+    ast::{DeBruijn, Program},
+    machine::cost_model::ExBudget,
+};
 
 use crate::{
     config::Config,
@@ -481,14 +484,31 @@ impl Project {
 
     fn run_tests(&self, tests: Vec<Script>) {
         for test in tests {
-            let result = test.program.eval();
+            // TODO: in the future we probably just want to be able to
+            // tell the machine to not explode on budget consumption.
+            let initial_budget = ExBudget {
+                mem: i64::MAX,
+                cpu: i64::MAX,
+            };
+
+            let result = test.program.eval(initial_budget);
+
             match result {
-                (Ok(..), _, _) => {
-                    println!("{}::{} ✓", test.module, test.name);
+                (Ok(..), remaining_budget, _) => {
+                    let ExBudget { mem, cpu } = initial_budget - remaining_budget;
+
+                    println!(
+                        "    [PASS] [mem: {}, cpu: {}] {}::{}",
+                        mem, cpu, test.module, test.name
+                    );
                 }
-                (Err(e), _, _) => {
-                    println!("{}::{} x", test.module, test.name);
-                    println!("{}", e);
+                (Err(_), remaining_budget, _) => {
+                    let ExBudget { mem, cpu } = initial_budget - remaining_budget;
+
+                    println!(
+                        "    [FAIL] [mem: {}, cpu: {}] {}::{}",
+                        mem, cpu, test.module, test.name
+                    );
                 }
             }
         }
diff --git a/examples/sample/validators/swap.ak b/examples/sample/validators/swap.ak
index 8764f3b6..aedd877e 100644
--- a/examples/sample/validators/swap.ak
+++ b/examples/sample/validators/swap.ak
@@ -1,7 +1,7 @@
 use sample
 
 pub fn spend(datum: sample.Datum, rdmr: sample.Redeemer, _ctx: Nil) -> Bool {
-  let x = #(datum,  #[244])
+  let _x = #(datum, rdmr, #[244])
 
   let y = [#(#[222], #[222]), #(#[233], #[52])]