[pbs-devel] [PATCH proxmox-backup v2 2/2] pbs2to3: add test cases for kernel version compatibility

[Maximiliano Sandoval]

Daniel Kral <d.kral at proxmox.com> writes:

Some small nits commented bellow.

> Implements test cases for the kernel version compatibility check in
> `pbs2to3` for a set of expected and unexpected kernel versions. They are
> tested for both cases, when PBS was already upgraded to v3.x and when it
> is still on v2.x.
>
> The kernel versions in the test cases were mostly hand-picked from the
> current Proxmox Backup Server repositories, excluding the kernel
> versions 5.19.* and 6.1.* because they were both not expected before
> this patch. I also added a fictional kernel version 7.1.0 as an
> unexpected case to make sure that the initial objective of only allowing
> arbitrary kernel versions above 6.2.* were marked as expected.
>
> Suggested-by: Maximiliano Sandoval <m.sandoval at proxmox.com>
> Signed-off-by: Daniel Kral <d.kral at proxmox.com>
> ---
>  src/bin/pbs2to3.rs | 61 ++++++++++++++++++++++++++++++++++++++++------
>  1 file changed, 54 insertions(+), 7 deletions(-)
>
> diff --git a/src/bin/pbs2to3.rs b/src/bin/pbs2to3.rs
> index 194a077e..2f6a8dcc 100644
> --- a/src/bin/pbs2to3.rs
> +++ b/src/bin/pbs2to3.rs
> @@ -123,18 +123,26 @@ impl Checker {
>          Ok(())
>      }
>  
> +    fn is_kernel_version_compatible(&self, running_version: &str) -> bool {
> +        let re = if self.upgraded {
> +            r"^6\.(?:2\.(?:[2-9]\d+|1[6-8]|1\d\d+)|([5-9]|\d{2,}))[^~]*$"
> +        } else {
> +            r"^(?:5\.(?:13|15)|6\.2)"
> +        };
> +        let re = Regex::new(re).expect("failed to compile kernel compat regex");
> +
> +        re.is_match(running_version)
> +    }
> +
>      fn check_kernel_compat(
>          &mut self,
>          pkg_versions: &[pbs_api_types::APTUpdateInfo],
>      ) -> Result<(), Error> {
>          self.output.log_info("Check running kernel version..")?;
> -        let (krunning, kinstalled) = if self.upgraded {
> -            (
> -                Regex::new(r"^6\.(?:2\.(?:[2-9]\d+|1[6-8]|1\d\d+)|([5-9]|\d{2,}))[^~]*$")?,
> -                "proxmox-kernel-6.2",
> -            )
> +        let kinstalled = if self.upgraded {
> +            "proxmox-kernel-6.2"
>          } else {
> -            (Regex::new(r"^(?:5\.(?:13|15)|6\.2)")?, "pve-kernel-5.15")
> +            "pve-kernel-5.15"
>          };
>  
>          let output = std::process::Command::new("uname").arg("-r").output();
> @@ -144,7 +152,7 @@ impl Checker {
>                  .log_fail("unable to determine running kernel version.")?,
>              Ok(ret) => {
>                  let running_version = std::str::from_utf8(&ret.stdout[..ret.stdout.len() - 1])?;
> -                if krunning.is_match(running_version) {
> +                if self.is_kernel_version_compatible(running_version) {
>                      if self.upgraded {
>                          self.output.log_pass(format!(
>                              "running new kernel '{running_version}' after upgrade."
> @@ -620,3 +628,42 @@ impl ConsoleOutput {
>          Ok(())
>      }
>  }
> +
> +#[cfg(test)]
> +mod tests {
> +    use super::*;
> +
> +    fn test_is_kernel_version_compatible(
> +        expected_versions: &[&str],
> +        unexpected_versions: &[&str],
> +        upgraded: bool,
> +    ) {
> +        let checker = Checker {
> +            output: ConsoleOutput::new(),
> +            upgraded,
> +        };
> +
> +        for version in expected_versions {
> +            assert!(checker.is_kernel_version_compatible(version));
> +        }
> +        for version in unexpected_versions.iter() {

The `iter()` should not be necessary here.

> +            assert!(!checker.is_kernel_version_compatible(version));
> +        }
> +    }
> +
> +    #[test]
> +    fn test_is_pve_kernel_version_compatible() {
> +        let expected_versions = vec!["5.13.19-6-pve", "5.15.158-2-pve", "6.2.16-5-pve"];
> +        let unexpected_versions = vec!["6.1.10-1-pve", "5.19.17-2-pve"];

Both of these could be slices instead of `vec!`.

> +
> +        test_is_kernel_version_compatible(&expected_versions, &unexpected_versions, false);

Which also would entail removing the `&`s here.

> +    }
> +
> +    #[test]
> +    fn test_is_proxmox_kernel_version_compatible() {
> +        let expected_versions = vec!["6.2.16-20-pve", "6.5.13-6-pve", "6.8.12-1-pve"];
> +        let unexpected_versions = vec!["5.13.19-6-pve", "6.1.15-1-pve", "7.1.0-1-pve"];

Ditto

> +
> +        test_is_kernel_version_compatible(&expected_versions, &unexpected_versions, true);
> +    }
> +}

-- 
Maximiliano