Add quality of life feature ensure application removal at return

fixes #656

Author: cderici

juju/model.py

@@ -875,6 +875,21 @@ async def reset(self, force=False):
             lambda: len(self.machines) == 0
         )
 
+    async def remove_application(self, app_name, block_until_done=False):
+        """Removes the given application from the model.
+
+        :param str app_name: Name of the application
+        :param bool block_until_done: Ensure the app is removed from the
+        model when returned
+        """
+        if app_name not in self.applications:
+            raise JujuError("Given application doesn't seem to appear in the\
+             model: %s\nCurrent applications are: %s" %
+                            (app_name, self.applications))
+        await self.applications[app_name].remove()
+        if block_until_done:
+            await self.block_until(lambda: app_name not in self.applications)
+
     async def block_until(self, *conditions, timeout=None, wait_period=0.5):
         """Return only after all conditions are true.
 

tests/integration/test_application.py

@@ -235,3 +235,15 @@ async def test_trusted(event_loop):
         await ubuntu_app.set_trusted(False)
         trusted = await ubuntu_app.get_trusted()
         assert trusted is False
+
+
+@base.bootstrapped
+@pytest.mark.asyncio
+async def test_app_remove_wait_flag(event_loop):
+    async with base.CleanModel() as model:
+        app = await model.deploy('cs:ubuntu')
+        a_name = app.name
+        await model.wait_for_idle(status="active")
+
+        await model.remove_application(app.name, block_until_done=True)
+        assert a_name not in model.applications