behave

test/features/steps/step_implementation.py

@@ -7,7 +7,8 @@
 import behave
 from behave import given, when, then
 import behave_agent
-from crmsh import corosync, sbd, userdir, bootstrap
+from crmsh import corosync, userdir, bootstrap
+from crmsh.sbd import SBDTimeout, SBDManager
 from crmsh import utils as crmutils
 from crmsh.sh import ShellUtils
 from utils import check_cluster_state, check_service_state, online, run_command, me, \
@@ -439,13 +440,13 @@ def step_impl(context, res_id, node):
 
 @then('SBD option "{key}" value is "{value}"')
 def step_impl(context, key, value):
-    res = sbd.SBDManager.get_sbd_value_from_config(key)
+    res = SBDManager.get_sbd_value_from_config(key)
     assert_eq(value, res)
 
 
 @then('SBD option "{key}" value for "{dev}" is "{value}"')
 def step_impl(context, key, dev, value):
-    res = sbd.SBDTimeout.get_sbd_msgwait(dev)
+    res = SBDTimeout.get_sbd_msgwait(dev)
     assert_eq(int(value), res)