def test_load_signage(self):
filename = os.path.join(os.path.dirname(__file__), 'data', 'signage.shp')
signage = SignageFactory(name="name", implantation_year=2010)
call_command('loadinfrastructure', filename, '--signage', type_default='label', name_default='name',
condition_default='condition', structure_default='structure',
description_default='description', year_default=2010, verbosity=0)
value = Signage.objects.all()
self.assertEquals(signage.name, value[1].name)
self.assertEquals(signage.implantation_year, value[1].implantation_year)
self.assertEquals(value.count(), 3)
def test_update_same_eid(self):
output = StringIO()
filename = os.path.join(os.path.dirname(__file__), 'data',
'signage.shp')
SignageFactory(name="name", eid="eid_2")
call_command('loadinfrastructure',
filename,
'--signage',
eid_field='eid',
type_default='label',
name_default='name',
verbosity=2,
stdout=output)
self.assertIn("Update : name with eid eid1", output.getvalue())
self.assertEqual(Signage.objects.count(), 2)
