summaryrefslogtreecommitdiffstats
path: root/apex/tests/test_apex_inventory.py
diff options
context:
space:
mode:
Diffstat (limited to 'apex/tests/test_apex_inventory.py')
-rw-r--r--apex/tests/test_apex_inventory.py17
1 files changed, 10 insertions, 7 deletions
diff --git a/apex/tests/test_apex_inventory.py b/apex/tests/test_apex_inventory.py
index 87e7d50..7197946 100644
--- a/apex/tests/test_apex_inventory.py
+++ b/apex/tests/test_apex_inventory.py
@@ -15,7 +15,7 @@ from nose.tools import (
assert_raises)
from apex import Inventory
-from apex.inventory.inventory import InventoryException
+from apex.inventory.inventory import ApexInventoryException
from apex.tests.constants import (
TEST_CONFIG_DIR,
TEST_DUMMY_CONFIG
@@ -47,14 +47,17 @@ class TestInventory:
for f in inventory_files:
i = Inventory(os.path.join(files_dir, f))
assert_equal(i.dump_instackenv_json(), None)
+ i['nodes'][0]['arch'] = 'aarch64'
+ i = Inventory(i)
+ assert_equal(i['nodes'][0]['arch'], 'aarch64')
def test_inventory_invalid_ha_count(self):
- assert_raises(InventoryException, Inventory,
+ assert_raises(ApexInventoryException, Inventory,
os.path.join(TEST_DUMMY_CONFIG, 'inventory-virt.yaml'),
virtual=True, ha=True)
def test_inventory_invalid_noha_count(self):
- assert_raises(InventoryException, Inventory,
+ assert_raises(ApexInventoryException, Inventory,
os.path.join(TEST_DUMMY_CONFIG,
'inventory-virt-1-node.yaml'),
virtual=True, ha=False)
@@ -64,7 +67,7 @@ class TestInventory:
virtual=True, ha=False)
assert_equal(i.dump_instackenv_json(), None)
- def test_exception(self):
- e = InventoryException("test")
- print(e)
- assert_is_instance(e, InventoryException)
+ def test_get_node_counts(self):
+ i = Inventory(os.path.join(TEST_DUMMY_CONFIG, 'inventory-virt.yaml'),
+ virtual=True, ha=False)
+ assert_equal(i.get_node_counts(), (1, 1))