summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--vio/vio/tests/test_aai_client.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/vio/vio/tests/test_aai_client.py b/vio/vio/tests/test_aai_client.py
index 6be9958..abf7618 100644
--- a/vio/vio/tests/test_aai_client.py
+++ b/vio/vio/tests/test_aai_client.py
@@ -305,3 +305,12 @@ class TestAAIClient(unittest.TestCase):
ret = self.view._get_hugepages_capabilities(extra)
self.assertIn(
"2", ret["hpa-feature-attributes"][0]["hpa-attribute-value"])
+
+ @mock.patch.object(restcall, "call_req")
+ def test_get_hpa_hugepage_small(self, mock_call):
+ extra = {
+ "hw:mem_page_size": "small"
+ }
+ ret = self.view._get_hugepages_capabilities(extra)
+ self.assertIn(
+ "4", ret["hpa-feature-attributes"][0]["hpa-attribute-value"])