Fix test_upgrade.py

canonical · Jun 28, 2024 · 3af1de0 · 3af1de0
1 parent 5c6292f
commit 3af1de0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/integration/test_upgrade.py b/tests/integration/test_upgrade.py
@@ -14,6 +14,7 @@
 import tenacity
 from pytest_operator.plugin import OpsTest
 
+import snap
 from .helpers import (
     APPLICATION_DEFAULT_APP_NAME,
     MYSQL_DEFAULT_APP_NAME,
@@ -224,7 +225,8 @@ def create_valid_upgrade_charm(charm_file: typing.Union[str, pathlib.Path]) -> N
         # set rev 102 (an old edge version of the snap)
         snap_file = pathlib.Path("src/snap.py")
         content = snap_file.read_text()
-        new_snap_content = re.sub(r'REVISION = "\d+"', 'REVISION = "102"', str(content))
+        # TODO: add arm64 support or mark as amd64 only
+        new_snap_content = re.sub(f'"x86_64": "{snap.revision}"', '"x86_64": "102"', str(content))
         charm_zip.writestr("src/snap.py", new_snap_content)