diff --git a/test/features/steps/step_implementation.py b/test/features/steps/step_implementation.py index d2d7d7ce9..80b92d932 100644 --- a/test/features/steps/step_implementation.py +++ b/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)