Add assertion

CygnusNetworks · Nov 3, 2018 · d119c30 · d119c30
1 parent 33df4aa
commit d119c30
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pypureomapi.py b/pypureomapi.py
@@ -1239,7 +1239,9 @@ def lookup(self, ltype, rvalues=[], ip=None, mac=None, name=None):
 				could be found or the object lacks an ip address or mac
 		@raises socket.error:
 		"""
-		msg = OmapiMessage.open(ltype.encode('utf-8'))
+		ltype_utf = ltype.encode("utf-8")
+		assert ltype_utf in [b"host", b"lease"]
+		msg = OmapiMessage.open(ltype_utf)
 		if ip:
 			msg.obj.append((b"ip-address", pack_ip(ip)))
 		if mac: