def test_formfields2taskargs_wrongshape(self):
task = PyKML()
color = {'id': 3,
'slowest': '#FFFF50',
'slow': '#FDD017',
'fast': '#C68E17',
'fastest': '#733C00'
}
formfields = {'start': '2013-05-14T10:11:12Z',
'end': '2013-05-15T08:33:34Z',
'trackers': [{'id': 1234, 'color': color}],
# wrong shape
'shape': 'square',
'size': 'medium',
'colorby': 'ispeed',
'speedthreshold1': 5,
'speedthreshold2': 10,
'speedthreshold3': 20,
'transparency': 100,
'altitudemode': 'absolute',
}
db_url = 'postgresql://localhost/eecology'
with self.assertRaises(Invalid) as e:
task.formfields2taskargs(formfields, db_url)
expected = {'shape': u'"square" is not one of circle, iarrow, tarrow'
}
self.assertEqual(e.exception.asdict(), expected)
def test_formfields2taskargs_noerrors(self, gpscount):
gpscount.return_value = 400
task = PyKML()
color = {'id': 3,
'slowest': '#FFFF50',
'slow': '#FDD017',
'fast': '#C68E17',
'fastest': '#733C00'
}
formfields = {'start': '2013-05-14T10:11:12Z',
'end': '2013-05-15T08:33:34Z',
'trackers': [{'id': 1234, 'color': color}],
'shape': 'circle',
'size': 'medium',
'colorby': 'ispeed',
'speedthreshold1': 5,
'speedthreshold2': 10,
'speedthreshold3': 20,
'transparency': 0,
'altitudemode': 'absolute',
}
db_url = 'postgresql://localhost/eecology'
taskargs = task.formfields2taskargs(formfields, db_url)
ecolor = {'slowest': '#FFFF50',
'slow': '#FDD017',
'fast': '#C68E17',
'fastest': '#733C00'
}
etaskargs = {'db_url': 'postgresql://localhost/eecology',
'start': '2013-05-14T10:11:12Z',
'end': '2013-05-15T08:33:34Z',
'trackers': [{'id': 1234, 'color': ecolor}],
'shape': 'circle',
'size': 'medium',
'sizebyalt': False,
'colorby': 'ispeed',
'speedthreshold1': 5,
'speedthreshold2': 10,
'speedthreshold3': 20,
'alpha': 100,
'altitudemode': 'absolute',
}
self.assertDictEqual(taskargs, etaskargs)