def create_field(self, field_name, field_attributes):
field_attributes = [
field_attribute.strip()
for field_attribute in field_attributes.split(",")
]
if Relationship.is_relationship(field_attributes):
self.relationships.append(
Relationship(field_name, field_attributes))
else:
self.fields.append(Field(field_name, field_attributes))
def test_relationship_mapping_template(self):
post_object = DataObject('$postResponse', {
'id': 'int,primarykey',
'title': 'string',
'body': 'string'
})
user_object = DataObject('$userResponse', {'id': 'int,primarykey'})
relationship = Relationship('user', ["M2O", "$userResponse"])
relationship.related_object = user_object
self.assertTemplateEqual('relationship_mapping.j2',
'RelationshipMapping.swift', {
'data_object': post_object,
'relationship': relationship
})
def test_relationship_mapping_template(self):
post_object = DataObject('$postResponse', {
'id': 'int,primarykey',
'title': 'string',
'body': 'string'
})
user_object = DataObject('$userResponse', {
'id': 'int,primarykey'
})
relationship = Relationship('user', ["M2O", "$userResponse"])
relationship.related_object = user_object
self.assertTemplateEqual('relationship_mapping.j2', 'RelationshipMapping.swift', {
'data_object': post_object,
'relationship': relationship
})
def create_field(self, field_name, field_attributes):
field_attributes = [field_attribute.strip() for field_attribute in field_attributes.split(",")]
if Relationship.is_relationship(field_attributes):
self.relationships.append(Relationship(field_name, field_attributes))
else:
self.fields.append(Field(field_name, field_attributes))
def test_is_relationship(self):
self.assertTrue(Relationship.is_relationship(["O2M", "$messageResponse"]))
self.assertFalse(Relationship.is_relationship(["email", "string"]))
