def test_modules_can_be_added(self): v = Variant("variant name") v.modules(["enterprise", "enterprise2"]) obj = v.to_map() assert "enterprise" in obj["modules"] assert "enterprise2" in obj["modules"]
def test_tasks_can_be_added(self): v = Variant("variant name") v.task(TaskSpec("task 0")).tasks( [TaskSpec("task 1"), TaskSpec("task 2")]) obj = v.to_map() assert "task 0" == obj["tasks"][0]["name"] assert "task 1" == obj["tasks"][1]["name"] assert "task 2" == obj["tasks"][2]["name"]
def test_expansions_can_be_added(self): v = Variant("variant name") v.expansion("k0", "v0") v.expansions({"k1": "v1", "k2": "v2"}) obj = v.to_map() assert "v0" in obj["expansions"]["k0"] assert "v1" in obj["expansions"]["k1"] assert "v2" in obj["expansions"]["k2"]
def test_flat_values_in_map(self): v = Variant("variant name") v.display_name("display name").run_on("distro").batch_time(100) obj = v.to_map() assert "variant name" == obj["name"] assert "display name" == obj["display_name"] assert ["distro"] == obj["run_on"] assert 100 == obj["batchtime"]
def test_display_tasks_can_be_added(self): v = Variant("variant name") v.display_task(DisplayTaskDefinition("display task 0"))\ .display_tasks([DisplayTaskDefinition("display task 1"), DisplayTaskDefinition("display task 2")]) obj = v.to_map() assert "display task 0" == obj["display_tasks"][0]["name"] assert "display task 1" == obj["display_tasks"][1]["name"] assert "display task 2" == obj["display_tasks"][2]["name"]
def test_empty_command_definition(self): v = Variant("variant name") assert "variant name" == v.get_name() assert {"name": "variant name"} == v.to_map()
def test_module_can_be_added(self): v = Variant("variant name") v.module("enterprise") obj = v.to_map() assert ["enterprise"] == obj["modules"]