test_utils: Await application state after Error

Error state during VFIO enablement is discovered by
'wait_for_application_state' function. So that's the one
that should be re-tried after error recovery.

Fixes: e4e1d18624
Signed-off-by: Martin Kalcok <martin.kalcok@canonical.com>

zaza/openstack/charm_tests/test_utils.py

@@ -757,8 +757,9 @@ class BaseCharmTest(unittest.TestCase):
                              ' recover. Possible cause:'
                              ' https://bugs.launchpad.net/juju/+bug/2077936')
                 zaza.model.resolve_units()
-                zaza.utilities.machine_os.enable_vfio_unsafe_noiommu_mode(
-                    unit, model_name=self.model_name)
+                model.wait_for_application_states(
+                    model_name=self.model_name,
+                    states=self.test_config.get('target_deploy_status', {}))
 
     def disable_hugepages_vfio_on_hvs_in_vms(self):
         """Disable hugepages and unsafe VFIO NOIOMMU on virtual hypervisors."""