Revision aba462a2 snf-astakos-app/astakos/quotaholder_app/tests.py
b/snf-astakos-app/astakos/quotaholder_app/tests.py | ||
---|---|---|
33 | 33 |
|
34 | 34 |
from django.test import TestCase |
35 | 35 |
|
36 |
from snf_django.utils.testing import assertGreater, assertIn, assertRaises |
|
36 | 37 |
import astakos.quotaholder_app.callpoint as qh |
37 | 38 |
from astakos.quotaholder_app.exception import ( |
38 | 39 |
InvalidDataError, |
... | ... | |
77 | 78 |
s1 = self.issue_commission([((holder, source, resource1), limit1/2), |
78 | 79 |
((holder, source, resource2), limit2)], |
79 | 80 |
name="initial") |
80 |
self.assertGreater(s1, 0)
|
|
81 |
assertGreater(s1, 0) |
|
81 | 82 |
|
82 | 83 |
r = qh.get_commission(self.client, s1) |
83 | 84 |
provisions = [ |
... | ... | |
95 | 96 |
self.assertEqual(r['serial'], s1) |
96 | 97 |
ps = r['provisions'] |
97 | 98 |
for p in ps: |
98 |
self.assertIn(p, provisions)
|
|
99 |
assertIn(p, provisions) |
|
99 | 100 |
|
100 |
with self.assertRaises(NoCommissionError):
|
|
101 |
with assertRaises(NoCommissionError): |
|
101 | 102 |
qh.get_commission(self.client, s1+1) |
102 | 103 |
|
103 | 104 |
# commission exceptions |
104 | 105 |
|
105 |
with self.assertRaises(NoCapacityError) as cm:
|
|
106 |
with assertRaises(NoCapacityError) as cm: |
|
106 | 107 |
self.issue_commission([((holder, source, resource1), 1), |
107 | 108 |
((holder, source, resource2), 1)]) |
108 | 109 |
|
... | ... | |
115 | 116 |
self.assertEqual(e.data['usage'], limit2) |
116 | 117 |
self.assertEqual(e.data['limit'], limit2) |
117 | 118 |
|
118 |
with self.assertRaises(NoQuantityError) as cm:
|
|
119 |
with assertRaises(NoQuantityError) as cm: |
|
119 | 120 |
self.issue_commission([((holder, source, resource1), -1)]) |
120 | 121 |
|
121 | 122 |
e = cm.exception |
... | ... | |
127 | 128 |
self.assertEqual(e.data['usage'], 0) |
128 | 129 |
self.assertEqual(e.data['limit'], 0) |
129 | 130 |
|
130 |
with self.assertRaises(NoHoldingError) as cm:
|
|
131 |
with assertRaises(NoHoldingError) as cm: |
|
131 | 132 |
self.issue_commission([((holder, source, resource1), 1), |
132 | 133 |
(('nonex', source, resource1), 1)]) |
133 | 134 |
|
... | ... | |
138 | 139 |
self.assertEqual(provision['resource'], resource1) |
139 | 140 |
self.assertEqual(provision['quantity'], 1) |
140 | 141 |
|
141 |
with self.assertRaises(DuplicateError) as cm:
|
|
142 |
with assertRaises(DuplicateError) as cm: |
|
142 | 143 |
self.issue_commission([((holder, source, resource1), 1), |
143 | 144 |
((holder, source, resource1), 2)]) |
144 | 145 |
|
... | ... | |
149 | 150 |
self.assertEqual(provision['resource'], resource1) |
150 | 151 |
self.assertEqual(provision['quantity'], 2) |
151 | 152 |
|
152 |
with self.assertRaises(InvalidDataError):
|
|
153 |
with assertRaises(InvalidDataError): |
|
153 | 154 |
self.issue_commission([((holder, source, resource1), 1), |
154 | 155 |
((holder, source, resource1), "nan")]) |
155 | 156 |
|
... | ... | |
217 | 218 |
} |
218 | 219 |
self.assertEqual(r, quotas) |
219 | 220 |
|
220 |
with self.assertRaises(NoCapacityError):
|
|
221 |
with assertRaises(NoCapacityError): |
|
221 | 222 |
self.issue_commission( |
222 | 223 |
[((holder, source, resource1), limit1/2+1)]) |
223 | 224 |
|
Also available in: Unified diff