def test_get_machine_edit_form_returns_MachineForm_if_non_admin(self): user = factory.make_User() self.assertEqual(MachineForm, get_machine_edit_form(user))
def test_get_machine_edit_form_returns_AdminMachineForm_if_admin(self): admin = factory.make_admin() self.assertEqual(AdminMachineForm, get_machine_edit_form(admin))