import mock
import pyipmi
import pyipmi.chassis
-
-from apex import clean_nodes
from mock import patch
from nose import tools
+from apex import clean_nodes
+
class TestClean(object):
@classmethod
with mock.patch.object(pyipmi.Session, 'establish') as mock_method:
with patch.object(pyipmi.chassis.Chassis,
'chassis_control_power_down') as mock_method2:
- clean_nodes('config/inventory.yaml')
+ clean_nodes('apex/tests/config/inventory.yaml')
tools.assert_equal(mock_method.call_count, 5)
tools.assert_equal(mock_method2.call_count, 5)