def test_should_get_field_from_field_set(self):
field1 = TextField('text',
'text',
'text label',
parent_field_code='field_set_code')
field2 = TextField('text1',
'text1',
'text label1',
parent_field_code='field_set_code')
fieldset_field = FieldSet('field_set',
'field_set_code',
'field set label',
field_set=[field1, field2])
form_model = FormModel(Mock(spec=DatabaseManager))
form_model._form_fields = [fieldset_field]
self.assertEqual(
field1,
form_model.get_field_by_code_in_fieldset('text', 'field_set_code'))
