def main():
setup_logging()
args = docopt(__doc__, version="CMK v1.4.1")
if args["cluster-init"]:
clusterinit.cluster_init(args["--host-list"], args["--all-hosts"],
args["--cmk-cmd-list"], args["--cmk-img"],
args["--cmk-img-pol"], args["--conf-dir"],
args["--install-dir"],
args["--num-exclusive-cores"],
args["--num-shared-cores"],
args["--pull-secret"],
args["--saname"], args["--exclusive-mode"],
args["--shared-mode"], args["--namespace"],
args["--excl-non-isolcpus"])
return
if args["init"]:
init.init(args["--conf-dir"],
int(args["--num-exclusive-cores"]),
int(args["--num-shared-cores"]),
args["--exclusive-mode"],
args["--shared-mode"],
args["--excl-non-isolcpus"])
return
if args["discover"]:
discover.discover(args["--conf-dir"])
return
if args["describe"]:
describe.describe(args["--conf-dir"])
return
if args["isolate"]:
isolate.isolate(args["--conf-dir"],
args["--pool"],
args["--no-affinity"],
args["<command>"],
args["<args>"],
args["--socket-id"])
return
if args["reconcile"]:
reconcile.reconcile(args["--conf-dir"],
int(args["--interval"]),
args["--publish"])
return
if args["install"]:
install.install(args["--install-dir"])
return
if args["uninstall"]:
uninstall.uninstall(args["--install-dir"],
args["--conf-dir"],
args["--namespace"])
return
if args["node-report"]:
nodereport.nodereport(args["--conf-dir"],
int(args["--interval"]),
args["--publish"])
return
if args["webhook"]:
webhook.webhook(args["--conf-file"])
def test_discover_version_check2(caplog):
with patch('intel.discover.add_node_er') as mock_er, \
patch('intel.discover.add_node_oir') as mock_oir, \
patch('intel.discover.add_node_label') as mock_label, \
patch('intel.discover.add_node_taint') as mock_taint:
discover.discover()
assert not mock_er.called
assert mock_oir.called
assert mock_label.called
assert mock_taint.called
def test_discover_version_check2(caplog):
conf_dir = helpers.conf_dir("ok")
with patch('intel.discover.add_node_er') as mock_er, \
patch('intel.discover.add_node_oir') as mock_oir, \
patch('intel.discover.add_node_label') as mock_label, \
patch('intel.discover.add_node_taint') as mock_taint:
discover.discover(conf_dir)
assert not mock_er.called
assert mock_oir.called
assert mock_label.called
assert mock_taint.called
def test_discover_version_check3(caplog):
with patch('intel.discover.add_node_er') as mock_er, \
patch('intel.discover.add_node_oir') as mock_oir, \
patch('intel.discover.add_node_label') as mock_label, \
patch('intel.discover.add_node_taint') as mock_taint:
with pytest.raises(SystemExit):
discover.discover()
# no resources should be created on unsupported cluster
assert not mock_er.called
assert not mock_oir.called
assert not mock_label.called
assert not mock_taint.called
def main():
setup_logging()
args = docopt(__doc__, version="CMK v1.0.1")
if args["cluster-init"]:
clusterinit.cluster_init(args["--host-list"], args["--all-hosts"],
args["--cmk-cmd-list"], args["--cmk-img"],
args["--cmk-img-pol"], args["--conf-dir"],
args["--install-dir"], args["--num-dp-cores"],
args["--num-cp-cores"], args["--pull-secret"],
args["--saname"])
return
if args["init"]:
init.init(args["--conf-dir"], int(args["--num-dp-cores"]),
int(args["--num-cp-cores"]))
return
if args["discover"]:
discover.discover(args["--conf-dir"])
return
if args["describe"]:
describe.describe(args["--conf-dir"])
return
if args["isolate"]:
isolate.isolate(args["--conf-dir"], args["--pool"],
args["--no-affinity"], args["<command>"],
args["<args>"])
return
if args["reconcile"]:
reconcile.reconcile(args["--conf-dir"], int(args["--interval"]),
args["--publish"])
return
if args["install"]:
install.install(args["--install-dir"])
return
if args["uninstall"]:
uninstall.uninstall(args["--install-dir"], args["--conf-dir"])
return
if args["node-report"]:
nodereport.nodereport(args["--conf-dir"], int(args["--interval"]),
args["--publish"])
return
