test: Support pre-release version

The "packaging" library only allows a limited set of pre-release names
[1], while the semantic versioning permits any alphanumeric charcters
[2]. Use "semver" library instead.

[1]: https://github.com/pypa/packaging/blob/7952e97c650cd70962a45018e11b760d7b1e1325/src/packaging/version.py#L122-L127
[2]: https://semver.org/#backusnaur-form-grammar-for-valid-semver-versions

Signed-off-by: Takahiro Itazuri <[email protected]>

Author: zulinx86

tests/framework/artifacts.py

@@ -75,10 +75,10 @@ def jailer(self):
 
     @property
     def version(self):
-        """Return Firecracker's version: `X.Y.Z`."""
-        # Get the filename, split on '-' and trim the leading 'v'.
-        # sample: firecracker-v1.2.0
-        return self.path.name.split("-")[1][1:]
+        """Return Firecracker's version: `X.Y.Z-prerelase`."""
+        # Get the filename, split on the first '-' and trim the leading 'v'.
+        # sample: firecracker-v1.2.0-alpha
+        return self.path.name.split("-", 1)[1][1:]
 
     @property
     def version_tuple(self):
@@ -132,7 +132,7 @@ def working_version_as_artifact():
     Return working copy of Firecracker as a release artifact
     """
     cargo_version = get_firecracker_version_from_toml()
-    return FirecrackerArtifact(current_release(cargo_version.base_version)[0])
+    return FirecrackerArtifact(current_release(str(cargo_version))[0])
 
 
 def firecracker_artifacts():

tests/framework/utils.py

@@ -16,8 +16,8 @@
 from contextlib import contextmanager
 from typing import Dict
 
-import packaging.version
 import psutil
+import semver
 from tenacity import (
     Retrying,
     retry,
@@ -377,7 +377,7 @@ def get_firecracker_version_from_toml():
     """
     cmd = "cd ../src/firecracker && cargo pkgid | cut -d# -f2 | cut -d: -f2"
     _, stdout, _ = check_output(cmd)
-    return packaging.version.parse(stdout)
+    return semver.Version.parse(stdout)
 
 
 def get_kernel_version(level=2):

tests/integration_tests/functional/test_api.py

@@ -11,8 +11,8 @@
 import time
 from pathlib import Path
 
-import packaging.version
 import pytest
+import semver
 
 import host_tools.drive as drive_tools
 import host_tools.network as net_tools
@@ -924,14 +924,12 @@ def test_api_version(uvm_plain):
     assert preboot_response.json() == postboot_response.json()
 
     cargo_version = get_firecracker_version_from_toml()
-    api_version = packaging.version.parse(
-        preboot_response.json()["firecracker_version"]
-    )
+    api_version = semver.Version.parse(preboot_response.json()["firecracker_version"])
 
     # Cargo version should match FC API version
     assert cargo_version == api_version
 
-    binary_version = packaging.version.parse(test_microvm.firecracker_version)
+    binary_version = semver.Version.parse(test_microvm.firecracker_version)
     assert api_version == binary_version
 
 

tools/devctr/Dockerfile

@@ -156,7 +156,7 @@ RUN cd /usr/include/$ARCH-linux-musl \
     && ln -s ../asm-generic asm-generic
 
 # Install static version of libseccomp
-# We need to compile from source because 
+# We need to compile from source because
 # libseccomp provided by the distribution is not
 # compiled with musl-gcc and we need this
 # for our musl builds.

tools/devctr/poetry.lock

@@ -34,6 +34,7 @@ requests-unixsocket2 = "^0.4.0"
 scipy = "^1.11.2"
 setproctitle = "^1.3.2"
 tenacity = "^8.2.2"
+semver = "^3.0.4"
 
 
 [tool.poetry.group.dev.dependencies]

tools/devctr/pyproject.toml

@@ -68,7 +68,7 @@
 DEVCTR_IMAGE_NO_TAG="public.ecr.aws/firecracker/fcuvm"
 
 # Development container tag
-DEVCTR_IMAGE_TAG=${DEVCTR_IMAGE_TAG:-v80}
+DEVCTR_IMAGE_TAG=${DEVCTR_IMAGE_TAG:-v81}
 
 # Development container image (name:tag)
 # This should be updated whenever we upgrade the development container.