Add comment and fix on functions that contains bugs in pdp module 77/49677/3
authorMohamedAssem <rfawzy.ext@orange.com>
Wed, 27 Dec 2017 09:22:37 +0000 (11:22 +0200)
committerReemMahmoud <rfawzy.ext@orange.com>
Tue, 2 Jan 2018 07:31:53 +0000 (09:31 +0200)
Change-Id: Idc88a54104d296c1f4ea9747953a32370485cdd3
Signed-off-by: MohamedAssem <rfawzy.ext@orange.com>
python_moondb/python_moondb/api/pdp.py

index 5fb7aa7..07feab7 100644 (file)
@@ -7,7 +7,7 @@ from uuid import uuid4
 from oslo_log import log as logging
 from python_moonutilities.security_functions import filter_input, enforce
 from python_moondb.api.managers import Managers
-
+from python_moonutilities import exceptions
 
 LOG = logging.getLogger("moon.db.api.pdp")
 
@@ -24,10 +24,14 @@ class PDPManager(Managers):
 
     @enforce(("read", "write"), "pdp")
     def delete_pdp(self, user_id, pdp_id):
+        if pdp_id not in self.driver.get_pdp(pdp_id=pdp_id):
+            raise exceptions.PdpUnknown
         return self.driver.delete_pdp(pdp_id=pdp_id)
 
     @enforce(("read", "write"), "pdp")
     def add_pdp(self, user_id, pdp_id=None, value=None):
+        if pdp_id in self.driver.get_pdp(pdp_id=pdp_id):
+            raise exceptions.PdpExisting
         if not pdp_id:
             pdp_id = uuid4().hex
         return self.driver.add_pdp(pdp_id=pdp_id, value=value)
@@ -35,4 +39,3 @@ class PDPManager(Managers):
     @enforce("read", "pdp")
     def get_pdp(self, user_id, pdp_id=None):
         return self.driver.get_pdp(pdp_id=pdp_id)
-