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))