def test_get_default_profile(self):
"""
get_default_profile should return a default profile attached to an architecture, or raise
value error if there is none.
"""
machine = NonCallableMagicMock(spec_set=Machine)
architecture = NonCallableMagicMock(spec_set=Architecture)
architecture.default_profile = "test_profile"
machine.architecture = architecture
self.assertEqual(cobbler.get_default_profile(machine), "test_profile")
architecture.default_profile = None
self.assertRaises(ValueError, cobbler.get_default_profile, machine)
def test_get_filename(self):
"""
get_filename should return the right filename attribute.
Machine > Group > Architecture > None
"""
machine = NonCallableMagicMock(spec_set=Machine)
machine.dhcp_filename = "machine"
group = NonCallableMagicMock(spec_set=MachineGroup)
group.dhcp_filename = "group"
machine.group = group
architecture = NonCallableMagicMock(spec_set=Architecture)
architecture.dhcp_filename = "architecture"
machine.architecture = architecture
self.assertEqual(cobbler.get_filename(machine), "machine")
machine.dhcp_filename = None
self.assertEqual(cobbler.get_filename(machine), "group")
group.dhcp_filename = None
self.assertEqual(cobbler.get_filename(machine), "architecture")
architecture.dhcp_filename = None
self.assertIsNone(cobbler.get_filename(machine))
