From 2856f88fcfbb26ab007ab0f7c31b54c50ff7faa7 Mon Sep 17 00:00:00 2001
From: Eyal <eyal.bar-ilan@nokia.com>
Date: Fri, 8 May 2020 10:02:39 +0300
Subject: [PATCH] Make sure depth is an int in get topology

Change-Id: If0e2bc147a51137fbe1988f84b921ce2182bb75b
---
 vitrage/api/controllers/v1/topology.py | 2 ++
 vitrage/common/utils.py                | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/vitrage/api/controllers/v1/topology.py b/vitrage/api/controllers/v1/topology.py
index 3a030b885..c616240ee 100644
--- a/vitrage/api/controllers/v1/topology.py
+++ b/vitrage/api/controllers/v1/topology.py
@@ -25,6 +25,7 @@ from pecan.core import abort
 from vitrage.api.controllers.rest import RootRestController
 from vitrage.api.policy import enforce
 from vitrage.common.constants import VertexProperties as VProps
+from vitrage.common.utils import to_int
 
 # noinspection PyProtectedMember
 from vitrage.common.utils import decompress_obj
@@ -44,6 +45,7 @@ class TopologyController(RootRestController):
     def post(self, depth=None, graph_type='graph', query=None, root=None,
              all_tenants=False):
         all_tenants = bool_from_string(all_tenants)
+        depth = to_int(depth)
         if all_tenants:
             enforce('get topology:all_tenants', pecan.request.headers,
                     pecan.request.enforcer, {})
diff --git a/vitrage/common/utils.py b/vitrage/common/utils.py
index 123f9d276..e66c75378 100644
--- a/vitrage/common/utils.py
+++ b/vitrage/common/utils.py
@@ -155,3 +155,7 @@ def decompress_obj(blob):
     del decoded_blob
     del str_data
     return obj
+
+
+def to_int(i):
+    return int(i) if i is not None else i