test(balloon): enable secret_free on all balloon tests

Add the secret_free fixture parameter to all balloon functional tests,
so they run with both SF_OFF and SF_ON variants. This exercises the new
fallocate(PUNCH_HOLE) discard path for guest_memfd-backed memory during
balloon inflate/deflate.

Signed-off-by: Jack Thomson <jackabt@amazon.com>

Author: JackThomson2

tests/integration_tests/functional/test_balloon.py

@@ -93,13 +93,13 @@ def _test_rss_memory_lower(test_microvm):
 
 
 # pylint: disable=C0103
-def test_rss_memory_lower(uvm_plain_any):
+def test_rss_memory_lower(uvm_plain_any, secret_free):
     """
     Test that inflating the balloon makes guest use less rss memory.
     """
     test_microvm = uvm_plain_any
     test_microvm.spawn()
-    test_microvm.basic_config()
+    test_microvm.basic_config(secret_free=secret_free)
     test_microvm.add_net_iface()
 
     # Add a memory balloon.
@@ -114,13 +114,13 @@ def test_rss_memory_lower(uvm_plain_any):
 
 
 # pylint: disable=C0103
-def test_inflate_reduces_free(uvm_plain_any):
+def test_inflate_reduces_free(uvm_plain_any, secret_free):
     """
     Check that the output of free in guest changes with inflate.
     """
     test_microvm = uvm_plain_any
     test_microvm.spawn()
-    test_microvm.basic_config()
+    test_microvm.basic_config(secret_free=secret_free)
     test_microvm.add_net_iface()
 
     # Install deflated balloon.
@@ -150,7 +150,7 @@ def test_inflate_reduces_free(uvm_plain_any):
 
 # pylint: disable=C0103
 @pytest.mark.parametrize("deflate_on_oom", [True, False])
-def test_deflate_on_oom(uvm_plain_any, deflate_on_oom):
+def test_deflate_on_oom(uvm_plain_any, secret_free, deflate_on_oom):
     """
     Verify that setting the `deflate_on_oom` option works correctly.
 
@@ -167,7 +167,7 @@ def test_deflate_on_oom(uvm_plain_any, deflate_on_oom):
 
     test_microvm = uvm_plain_any
     test_microvm.spawn()
-    test_microvm.basic_config()
+    test_microvm.basic_config(secret_free=secret_free)
     test_microvm.add_net_iface()
 
     # Add a deflated memory balloon.
@@ -215,13 +215,13 @@ def test_deflate_on_oom(uvm_plain_any, deflate_on_oom):
 
 
 # pylint: disable=C0103
-def test_reinflate_balloon(uvm_plain_any):
+def test_reinflate_balloon(uvm_plain_any, secret_free):
     """
     Verify that repeatedly inflating and deflating the balloon works.
     """
     test_microvm = uvm_plain_any
     test_microvm.spawn()
-    test_microvm.basic_config()
+    test_microvm.basic_config(secret_free=secret_free)
     test_microvm.add_net_iface()
 
     # Add a deflated memory balloon.
@@ -280,13 +280,13 @@ def test_reinflate_balloon(uvm_plain_any):
 
 
 # pylint: disable=C0103
-def test_stats(uvm_plain_any):
+def test_stats(uvm_plain_any, secret_free):
     """
     Verify that balloon stats work as expected.
     """
     test_microvm = uvm_plain_any
     test_microvm.spawn()
-    test_microvm.basic_config()
+    test_microvm.basic_config(secret_free=secret_free)
     test_microvm.add_net_iface()
 
     # Add a memory balloon with stats enabled.
@@ -351,13 +351,13 @@ def test_stats(uvm_plain_any):
     check_guest_dmesg_for_stalls(test_microvm.ssh)
 
 
-def test_stats_update(uvm_plain_any):
+def test_stats_update(uvm_plain_any, secret_free):
     """
     Verify that balloon stats update correctly.
     """
     test_microvm = uvm_plain_any
     test_microvm.spawn()
-    test_microvm.basic_config()
+    test_microvm.basic_config(secret_free=secret_free)
     test_microvm.add_net_iface()
 
     # Add a memory balloon with stats enabled.
@@ -403,7 +403,7 @@ def test_stats_update(uvm_plain_any):
     check_guest_dmesg_for_stalls(test_microvm.ssh)
 
 
-def test_balloon_snapshot(uvm_plain_any, microvm_factory):
+def test_balloon_snapshot(uvm_plain_any, secret_free, microvm_factory):
     """
     Test that the balloon works after pause/resume.
     """
@@ -415,6 +415,7 @@ def test_balloon_snapshot(uvm_plain_any, microvm_factory):
     vm.basic_config(
         vcpu_count=2,
         mem_size_mib=256,
+        secret_free=secret_free,
     )
     vm.add_net_iface()
 
@@ -488,7 +489,9 @@ def test_balloon_snapshot(uvm_plain_any, microvm_factory):
 
 
 @pytest.mark.parametrize("method", ["reporting", "hinting"])
-def test_hinting_reporting_snapshot(uvm_plain_any, microvm_factory, method):
+def test_hinting_reporting_snapshot(
+    uvm_plain_any, secret_free, microvm_factory, method
+):
     """
     Test that the balloon hinting and reporting works after pause/resume.
     """
@@ -500,6 +503,7 @@ def test_hinting_reporting_snapshot(uvm_plain_any, microvm_factory, method):
     vm.basic_config(
         vcpu_count=2,
         mem_size_mib=256,
+        secret_free=secret_free,
     )
     vm.add_net_iface()
 
@@ -575,13 +579,13 @@ def test_hinting_reporting_snapshot(uvm_plain_any, microvm_factory, method):
 
 
 @pytest.mark.parametrize("method", ["traditional", "hinting", "reporting"])
-def test_memory_scrub(uvm_plain_any, method):
+def test_memory_scrub(uvm_plain_any, secret_free, method):
     """
     Test that the memory is zeroed after deflate.
     """
     microvm = uvm_plain_any
     microvm.spawn()
-    microvm.basic_config(vcpu_count=2, mem_size_mib=256)
+    microvm.basic_config(vcpu_count=2, mem_size_mib=256, secret_free=secret_free)
     microvm.add_net_iface()
 
     free_page_reporting = method == "reporting"