Add N2VCNotFound exception

[osm/N2VC.git] / n2vc / n2vc_juju_conn.py

diff --git a/n2vc/n2vc_juju_conn.py b/n2vc/n2vc_juju_conn.py
index 4456dd9..6b8ac09 100644 (file)
--- a/n2vc/n2vc_juju_conn.py
+++ b/n2vc/n2vc_juju_conn.py
@@ -32,7 +32,7 @@ from n2vc.n2vc_conn import N2VCConnector
 from n2vc.n2vc_conn import obj_to_dict, obj_to_yaml
 from n2vc.exceptions \
     import N2VCBadArgumentsException, N2VCException, N2VCConnectionException, \
 from n2vc.n2vc_conn import obj_to_dict, obj_to_yaml
 from n2vc.exceptions \
     import N2VCBadArgumentsException, N2VCException, N2VCConnectionException, \

-    N2VCExecutionException, N2VCInvalidCertificate
+    N2VCExecutionException, N2VCInvalidCertificate, N2VCNotFound

 from n2vc.juju_observer import JujuModelObserver
 
 from juju.controller import Controller
 from n2vc.juju_observer import JujuModelObserver
 
 from juju.controller import Controller


@@ -1294,7 +1294,7 @@ class N2VCJujuConnector(N2VCConnector):
         model = await self._juju_get_model(model_name=model_name)
 
         if not model:
         model = await self._juju_get_model(model_name=model_name)
 
         if not model:

-            raise N2VCException(
+            raise N2VCNotFound(

                 message="Model {} does not exist".format(model_name)
             )
 
                 message="Model {} does not exist".format(model_name)
             )