Revision 20c6de35 snf-astakos-app/astakos/quotaholder_app/tests.py
| b/snf-astakos-app/astakos/quotaholder_app/tests.py | ||
|---|---|---|
| 34 | 34 |
from django.test import TestCase |
| 35 | 35 |
|
| 36 | 36 |
from snf_django.utils.testing import assertGreater, assertIn, assertRaises |
| 37 |
from astakos.quotaholder_app import models |
|
| 37 | 38 |
import astakos.quotaholder_app.callpoint as qh |
| 38 | 39 |
from astakos.quotaholder_app.exception import ( |
| 39 | 40 |
InvalidDataError, |
| ... | ... | |
| 275 | 276 |
serial = self.issue_commission([]) |
| 276 | 277 |
r = qh.resolve_pending_commission(self.client, serial) |
| 277 | 278 |
self.assertEqual(r, True) |
| 279 |
|
|
| 280 |
def test_030_set(self): |
|
| 281 |
holder = 'h0' |
|
| 282 |
source = 'system' |
|
| 283 |
resource1 = 'r1' |
|
| 284 |
resource2 = 'r2' |
|
| 285 |
limit1 = 10 |
|
| 286 |
limit2 = 20 |
|
| 287 |
|
|
| 288 |
models.Holding.objects.create( |
|
| 289 |
holder=holder, source=source, resource=resource1, |
|
| 290 |
usage_min=1, usage_max=1, limit=2) |
|
| 291 |
models.Holding.objects.create( |
|
| 292 |
holder=holder, source=source, resource=resource2, |
|
| 293 |
usage_min=2, usage_max=2, limit=22) |
|
| 294 |
|
|
| 295 |
qh.set_quota([((holder, source, resource1), limit1), |
|
| 296 |
((holder, source, resource1), limit2)]) |
|
| 297 |
|
|
| 298 |
r = qh.get_quota(holders=[holder]) |
|
| 299 |
self.assertEqual(r, {(holder, source, resource1): (limit2, 1, 1),
|
|
| 300 |
(holder, source, resource2): (22, 2, 2)}) |
|
Also available in: Unified diff