def get_quota(user_id):
with session_query() as session:
user = User.load(user_id, session)
quota = user.quota
remaining = None
if quota is not None:
remaining = user.remaining_quota(session)
return (remaining, quota)
def test_user_quota_update(self):
# test update quota
(before, _) = user_ctrl.get_quota(self.userid)
scanid = scan_ctrl.new(self.scanid, self.userid, 2)
for i in xrange(0, 10):
for probe in ['probe1', 'probe2']:
job_ctrl.new(scanid, "file-{0}".format(i), probe, 1)
scan_ctrl.launched(scanid)
(after, _) = user_ctrl.get_quota(self.userid)
self.assertEqual(before - after, 20)
with session_query() as session:
user = User.load(self.userid, session)
scan_ids = [scan.id for scan in user.scans]
self.assertNotEqual(scan_ids, [])
self.assertTrue(scanid in scan_ids)
def get_ftpuser(user_id):
with session_query() as session:
user = User.load(user_id, session)
return user.ftpuser
