def test_arch(value, expected):
# Arrange
test_api = CobblerAPI()
distro = Distro(test_api)
# Act
with expected:
distro.arch = value
# Assert
if isinstance(value, str):
assert distro.arch.value == value
else:
assert distro.arch == value
def test_calculate_grub_name(self, input_arch, result_binary_name,
expected_exception, api):
# Arrange
test_builder = buildiso.BuildIso(api)
test_distro = Distro(api)
test_distro.name = "testdistro"
test_distro.arch = input_arch
# Act
with expected_exception:
result = test_builder.calculate_grub_name(test_distro)
# Assert
assert result == result_binary_name
