cloud-init: minor test cleanup

canonical · Oct 4, 2024 · 9781b44 · 9781b44
1 parent a348c8b
commit 9781b44
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/subiquity/tests/test_cloudinit.py b/subiquity/tests/test_cloudinit.py
@@ -53,14 +53,12 @@ class TestCloudInitVersion(SubiTestCase):
     )
     def test_split_version(self, pkgver, expected):
         with patch("subiquity.cloudinit.run_command") as rc:
-            rc.return_value = Mock()
-            rc.return_value.stdout = pkgver
+            rc.return_value = Mock(stdout=pkgver)
             self.assertEqual(expected, cloud_init_version())
 
     def test_cloud_init_not_present(self):
         with patch("subiquity.cloudinit.run_command") as rc:
-            rc.return_value = Mock()
-            rc.return_value.stdout = ""
+            rc.return_value = Mock(stdout="")
             self.assertEqual("", cloud_init_version())
 
     def test_cloud_init_full(self):