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"]
