test-runner: Fix accidental use of generated entry args

Author: nicholasbishop

uefi-test-runner/examples/hello_world.rs

@@ -6,6 +6,7 @@
 
 // ANCHOR: use
 use log::info;
+use uefi::boot;
 use uefi::prelude::*;
 // ANCHOR_END: use
 
@@ -18,7 +19,7 @@ fn main() -> Status {
     // ANCHOR_END: services
     // ANCHOR: log
     info!("Hello world!");
-    system_table.boot_services().stall(10_000_000);
+    boot::stall(10_000_000);
     // ANCHOR_END: log
     // ANCHOR: return
     Status::SUCCESS

uefi-test-runner/examples/shell_params.rs

@@ -7,8 +7,8 @@
 use log::error;
 // ANCHOR: use
 use uefi::prelude::*;
-use uefi::println;
 use uefi::proto::shell_params::ShellParameters;
+use uefi::{boot, println};
 
 extern crate alloc;
 use alloc::string::{String, ToString};
@@ -21,12 +21,11 @@ fn main() -> Status {
     // ANCHOR_END: entry
     // ANCHOR: services
     uefi::helpers::init().unwrap();
-    let boot_services = system_table.boot_services();
     // ANCHOR_END: services
 
     // ANCHOR: params
     let shell_params =
-        boot_services.open_protocol_exclusive::<ShellParameters>(image_handle);
+        boot::open_protocol_exclusive::<ShellParameters>(boot::image_handle());
     let shell_params = match shell_params {
         Ok(s) => s,
         Err(e) => {