Revision 08ce0e6f snf-quotaholder-app/quotaholder_django/test/simpletests.py

b/snf-quotaholder-app/quotaholder_django/test/simpletests.py
47 47
    Name, Key, Quantity, Capacity, ImportLimit, ExportLimit, Resource, Flags,
48 48
    Imported, Exported, Returned, Released)
49 49

  
50
QH_MAX_INT = 10**32
51

  
50 52
DEFAULT_HOLDING = (0, 0, 0, 0)
51 53

  
52 54
class QHAPITest(QHTestCase):
......
236 238
        resource1 = self.rand_resource()
237 239

  
238 240
        r = self.qh.set_quota(
239
            set_quota=[(e0, resource0, k0) + (5, None, 5, 6) + (0,),
241
            set_quota=[(e0, resource0, k0) + (5, QH_MAX_INT, 5, 6) + (0,),
240 242
                       (e1, resource0, k1) + (5, 5, 5, 5) + (0,)])
241 243
        self.assertEqual(r, [])
242 244

  
243 245
        r = self.qh.add_quota(clientkey=self.client,
244 246
                              serial=1,
245
                              sub_quota=[(e0, resource0, k0, 0, None, 1, 1)],
246
                              add_quota=[(e0, resource0, k0, 0, 3, None, 0),
247
                              sub_quota=[(e0, resource0, k0, 0, QH_MAX_INT, 1, 1)],
248
                              add_quota=[(e0, resource0, k0, 0, 3, QH_MAX_INT, 0),
247 249
                                         # new holding
248
                                         (e0, resource1, k0, 0, None, 5, 5)])
250
                                         (e0, resource1, k0, 0, QH_MAX_INT, 5, 5)])
249 251
        self.assertEqual(r, [])
250 252

  
251 253
        r = self.qh.get_quota(get_quota=[(e0, resource0, k0),
252 254
                                         (e0, resource1, k0)])
253
        self.assertEqual(r, [(e0, resource0, 5, 3, None, 5)
255
        self.assertEqual(r, [(e0, resource0, 5, 3, QH_MAX_INT, 5)
254 256
                             + DEFAULT_HOLDING + (0,),
255
                             (e0, resource1, 0, None, 5, 5)
257
                             (e0, resource1, 0, QH_MAX_INT, 5, 5)
256 258
                             + DEFAULT_HOLDING + (0,)])
257 259

  
258 260
        # repeated serial
259 261
        r = self.qh.add_quota(clientkey=self.client,
260 262
                              serial=1,
261
                              sub_quota=[(e0, resource1, k0, 0, None, (-5), 0)],
262
                              add_quota=[(e0, resource0, k0, 0, 2, None, 0)])
263
                              sub_quota=[(e0, resource1, k0, 0, QH_MAX_INT, (-5), 0)],
264
                              add_quota=[(e0, resource0, k0, 0, 2, QH_MAX_INT, 0)])
263 265
        self.assertEqual(r, [(e0, resource1), (e0, resource0)])
264 266

  
265 267
        r = self.qh.query_serials(clientkey=self.client, serials=[1, 2])
......
282 284
        # serial has been deleted
283 285
        r = self.qh.add_quota(clientkey=self.client,
284 286
                              serial=1,
285
                              add_quota=[(e0, resource0, k0, 0, 2, None, 0)])
287
                              add_quota=[(e0, resource0, k0, 0, 2, QH_MAX_INT, 0)])
286 288
        self.assertEqual(r, [])
287 289

  
288 290
        # none is committed
289 291
        r = self.qh.add_quota(clientkey=self.client,
290 292
                              serial=2,
291
                              add_quota=[(e1, resource0, k1, 0, (-10), None, 0),
293
                              add_quota=[(e1, resource0, k1, 0, (-10), QH_MAX_INT, 0),
292 294
                                         (e0, resource1, k0, 1, 0, 0, 0)])
293 295
        self.assertEqual(r, [(e1, resource0)])
294 296

  
......
296 298
                                         (e0, resource1, k0)])
297 299
        self.assertEqual(r, [(e1, resource0, 5, 5 , 5, 5)
298 300
                             + DEFAULT_HOLDING + (0,),
299
                             (e0, resource1, 0, None, 5, 5)
301
                             (e0, resource1, 0, QH_MAX_INT, 5, 5)
300 302
                             + DEFAULT_HOLDING + (0,)])
301 303

  
304
    def test_0082_max_quota(self):
305
        e0, k0 = self.new_entity()
306
        e1, k1 = self.new_entity()
307
        resource0 = self.rand_resource()
308
        resource1 = self.rand_resource()
309

  
310
        r = self.qh.set_quota(
311
            set_quota=[(e0, resource0, k0) + (5, QH_MAX_INT, 5, 6) + (0,)])
312
        self.assertEqual(r, [])
313

  
314
        r = self.qh.add_quota(clientkey=self.client,
315
                              serial=3,
316
                              add_quota=[(e0, resource0, k0, 0, QH_MAX_INT, 0, 0)])
317

  
318
        self.assertEqual(r, [])
319

  
320
        r = self.qh.get_quota(get_quota=[(e0, resource0, k0)])
321
        self.assertEqual(r, [(e0, resource0, 5, 2*QH_MAX_INT, 5, 6)
322
                             + DEFAULT_HOLDING + (0,)])
323

  
324

  
325

  
302 326
    def test_0090_commissions(self):
303 327
        e0, k0 = self.new_entity()
304 328
        e1, k1 = self.new_entity()

Also available in: Unified diff