fix uninstall test (#7)

canonical · Feb 16, 2024 · d435174 · d435174
1 parent 464bb9e
commit d435174
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/maas-agent-charm/tests/unit/test_charm.py b/maas-agent-charm/tests/unit/test_charm.py
@@ -36,6 +36,7 @@ def test_start(self, mock_helper):
 
     @patch("charm.MaasHelper", autospec=True)
     def test_remove(self, mock_helper):
+        mock_helper.get_maas_mode.return_value = "rack"
         self.harness.begin()
         self.harness.charm.on.remove.emit()
         mock_helper.uninstall.assert_called_once()