def test_getAccelerationCount(self, dbsession):
db_url = 'postgresql://localhost/db1?ssl=true'
device_info_serial = 1234
start = datetime(2013, 1, 1, tzinfo=UTC)
end = datetime(2013, 12, 12, tzinfo=UTC)
query = Mock(Query)
session = Mock(Session)
session.query.return_value = query
dbsession.return_value = session
models.getAccelerationCount(db_url, device_info_serial, start, end)
query.count.assert_called()
def formfields2taskargs(self, fields, db_url):
schema = Schema()
taskargs = schema.deserialize(fields)
# Test if selection will give results
validateRange(getAccelerationCount(db_url,
taskargs['tracker_id'],
taskargs['start'],
taskargs['end'],
), 0, self.MAX_ACC_COUNT)
taskargs['db_url'] = db_url
return taskargs