Move max_memory_sandbox test to layout module

Move the test from integration tests into the layout module where it belongs, and rewrite it to test SandboxMemoryLayout::get_memory_size directly instead of going through UninitializedSandbox::new.

Signed-off-by: Ludvig Liljenberg <[email protected]>

Author: ludfjig

src/hyperlight_host/src/mem/layout.rs

@@ -662,4 +662,13 @@ mod tests {
             get_expected_memory_size(&sbox_mem_layout)
         );
     }
+
+    #[test]
+    fn test_max_memory_sandbox() {
+        let mut cfg = SandboxConfiguration::default();
+        cfg.set_input_data_size(0x40000000);
+        let layout = SandboxMemoryLayout::new(cfg, 4096, 2048, 4096, 4096, None).unwrap();
+        let result = layout.get_memory_size();
+        assert!(matches!(result.unwrap_err(), MemoryRequestTooBig(..)));
+    }
 }

src/hyperlight_host/tests/sandbox_host_tests.rs

@@ -207,24 +207,6 @@ fn incorrect_parameter_num() {
     });
 }
 
-#[test]
-fn max_memory_sandbox() {
-    use hyperlight_host::{GuestBinary, UninitializedSandbox};
-    use hyperlight_testing::simple_guest_as_string;
-
-    let mut cfg = hyperlight_host::sandbox::SandboxConfiguration::default();
-    cfg.set_input_data_size(0x40000000);
-    let a = UninitializedSandbox::new(
-        GuestBinary::FilePath(simple_guest_as_string().unwrap()),
-        Some(cfg),
-    );
-
-    assert!(matches!(
-        a.unwrap_err(),
-        HyperlightError::MemoryRequestTooBig(..)
-    ));
-}
-
 #[test]
 fn iostack_is_working() {
     with_all_sandboxes(|mut sandbox| {