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