def test_get_form_class_returns_DeviceForm_for_update(self): user = factory.make_User() handler = DeviceHandler(user, {}) self.assertIs(DeviceForm, handler.get_form_class("update"))
def test_get_form_class_returns_DeviceWithMACsForm_for_create(self): user = factory.make_User() handler = DeviceHandler(user, {}, None) self.assertIs(DeviceWithMACsForm, handler.get_form_class("create"))