def create_sample_component(self, type_id, type_="desktop"): """ Helper method created in order to populate the table 'Component' """ sample_component = Component() sample_component.type = type_ sample_component.type_id = type_id sample_component.save() return sample_component
def test_create_component(self): """ Tests the creation of a Component """ first_component = Component() first_component.type = "desktop" first_component.type_id = "firefox.desktop" first_component.save() second_component = Component() second_component.type = "desktop" second_component.type_id = "xchat.desktop" second_component.save() saved_components = Component.objects.all() self.assertEqual(saved_components.count(), 2)