def test_groups(mock_driver): inventory = NodeInventory(driver=mock_driver, restrict_to_groups={ "TEST1": ["198.168.1.1"], "TEST2": ["198.168.1.1", "198.168.2.1"], }) inventory.sync() assert inventory.get_groups() == ["TEST1", "TEST2"]
def test_find(nodes, mock_driver, query, expected_indices): inventory = NodeInventory(driver=mock_driver, restrict_to_groups={ "TEST1": ["198.168.1.1"], "TEST2": ["198.168.1.1", "198.168.2.1"], }) inventory.sync() assert list( inventory.find_nodes(query)) == [nodes[x] for x in expected_indices]
def test_sync(nodes, mock_driver): inventory = NodeInventory(driver=mock_driver, restrict_to_groups={ "TEST1": ["198.168.1.1"], "TEST2": ["198.168.1.1", "198.168.2.1"], }) inventory.sync() assert inventory.groups == {"TEST1": nodes[0:1], "TEST2": nodes[0:2]} assert inventory.get_azs() == ["AZ1", "AZ2"]
def test_get_by_ip(nodes, mock_driver, ip, should_find, index): inventory = NodeInventory(driver=mock_driver, restrict_to_groups={ "TEST1": ["198.168.1.1"], "TEST2": ["198.168.1.1", "198.168.2.1"], }) inventory.sync() if should_find: assert inventory.get_node_by_ip(ip) is nodes[index] else: assert inventory.get_node_by_ip(ip) == None