Check the port exists in set_port_is_locked

Change-Id: Ie8bedafc0a84bdab537b4152d73718f1b3a399fd

1 files changed, 3 insertions, 0 deletions

diff --git a/db/db.py b/db/db.py
index 13d3773..c5db8b3 100644
--- a/db/db.py
+++ b/db/db.py
@@ -330,6 +330,9 @@ class VlanDB:
     # settings on the port. Use this to lock down ports that are used
     # for PDUs and other core infrastructure
     def set_port_is_locked(self, port_id, is_locked):
+        port = self.get_port_by_id(port_id)
+        if port is None:
+            raise InputError("Port %s does not exist" % port_id)
         try:
             sql = "UPDATE port SET is_locked=%s WHERE port_id=%s"
             data = (is_locked, port_id)