# Copyright 2012, 2013 GRNET S.A. All rights reserved. # # Redistribution and use in source and binary forms, with or # without modification, are permitted provided that the following # conditions are met: # # 1. Redistributions of source code must retain the above # copyright notice, this list of conditions and the following # disclaimer. # # 2. Redistributions in binary form must reproduce the above # copyright notice, this list of conditions and the following # disclaimer in the documentation and/or other materials # provided with the distribution. # # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE # POSSIBILITY OF SUCH DAMAGE. # # The views and conclusions contained in the software and # documentation are those of the authors and should not be # interpreted as representing official policies, either expressed # or implied, of GRNET S.A. from django.test import TestCase from snf_django.utils.testing import assertGreater, assertIn, assertRaises from astakos.quotaholder_app import models import astakos.quotaholder_app.callpoint as qh from astakos.quotaholder_app.exception import ( InvalidDataError, NoCommissionError, NoQuantityError, NoCapacityError, NoHoldingError, DuplicateError) class QuotaholderTest(TestCase): def issue_commission(self, provisions, name="", force=False): return qh.issue_commission(clientkey=self.client, name=name, force=force, provisions=provisions) def test_010_qh(self): holder = 'h0' source = 'system' resource1 = 'r1' resource2 = 'r2' limit1 = 10 limit2 = 20 qh.set_quota([((holder, source, resource1), limit1), ((holder, source, resource1), limit2)]) r = qh.get_quota(holders=[holder], sources=[source], resources=[resource1, resource2]) self.assertEqual(r, {(holder, source, resource1): (limit2, 0, 0)}) r = qh.get_quota() self.assertEqual(r, {(holder, source, resource1): (limit2, 0, 0)}) # issuing commissions qh.set_quota([((holder, source, resource1), limit1), ((holder, source, resource2), limit2)]) s1 = self.issue_commission([((holder, source, resource1), limit1/2), ((holder, source, resource2), limit2)], name="initial") assertGreater(s1, 0) r = qh.get_commission(self.client, s1) provisions = [ {'holder': holder, 'source': source, 'resource': resource1, 'quantity': limit1/2, }, {'holder': holder, 'source': source, 'resource': resource2, 'quantity': limit2, } ] self.assertEqual(r['serial'], s1) ps = r['provisions'] for p in ps: assertIn(p, provisions) with assertRaises(NoCommissionError): qh.get_commission(self.client, s1+1) # commission exceptions with assertRaises(NoCapacityError) as cm: self.issue_commission([((holder, source, resource1), 1), ((holder, source, resource2), 1)]) e = cm.exception provision = e.data['provision'] self.assertEqual(provision['holder'], holder) self.assertEqual(provision['source'], source) self.assertEqual(provision['resource'], resource2) self.assertEqual(provision['quantity'], 1) self.assertEqual(e.data['usage'], limit2) self.assertEqual(e.data['limit'], limit2) with assertRaises(NoQuantityError) as cm: self.issue_commission([((holder, source, resource1), -1)]) e = cm.exception provision = e.data['provision'] self.assertEqual(provision['holder'], holder) self.assertEqual(provision['source'], source) self.assertEqual(provision['resource'], resource1) self.assertEqual(provision['quantity'], -1) self.assertEqual(e.data['usage'], 0) self.assertEqual(e.data['limit'], 0) with assertRaises(NoHoldingError) as cm: self.issue_commission([((holder, source, resource1), 1), (('nonex', source, resource1), 1)]) e = cm.exception provision = e.data['provision'] self.assertEqual(provision['holder'], 'nonex') self.assertEqual(provision['source'], source) self.assertEqual(provision['resource'], resource1) self.assertEqual(provision['quantity'], 1) with assertRaises(DuplicateError) as cm: self.issue_commission([((holder, source, resource1), 1), ((holder, source, resource1), 2)]) e = cm.exception provision = e.data['provision'] self.assertEqual(provision['holder'], holder) self.assertEqual(provision['source'], source) self.assertEqual(provision['resource'], resource1) self.assertEqual(provision['quantity'], 2) with assertRaises(InvalidDataError): self.issue_commission([((holder, source, resource1), 1), ((holder, source, resource1), "nan")]) r = qh.get_quota(holders=[holder]) quotas = {(holder, source, resource1): (limit1, 0, limit1/2), (holder, source, resource2): (limit2, 0, limit2), } self.assertEqual(r, quotas) # resolve commission r = qh.get_pending_commissions(clientkey=self.client) self.assertEqual(len(r), 1) serial = r[0] r = qh.resolve_pending_commission(self.client, serial) self.assertEqual(r, True) r = qh.get_pending_commissions(clientkey=self.client) self.assertEqual(r, []) r = qh.resolve_pending_commission(self.client, serial) self.assertEqual(r, False) r = qh.get_quota(holders=[holder]) quotas = {(holder, source, resource1): (limit1, limit1/2, limit1/2), (holder, source, resource2): (limit2, limit2, limit2), } self.assertEqual(r, quotas) # resolve commissions serial = self.issue_commission([((holder, source, resource1), 1), ((holder, source, resource2), -1)]) r = qh.get_quota(holders=[holder]) quotas = {(holder, source, resource1): (limit1, limit1/2, limit1/2+1), (holder, source, resource2): (limit2, limit2-1, limit2), } self.assertEqual(r, quotas) r = qh.resolve_pending_commission(self.client, serial, accept=False) self.assertEqual(r, True) serial1 = self.issue_commission([((holder, source, resource1), 1), ((holder, source, resource2), -1)]) serial2 = self.issue_commission([((holder, source, resource1), 1), ((holder, source, resource2), -1)]) serial3 = self.issue_commission([((holder, source, resource1), 1), ((holder, source, resource2), -1)]) r = qh.resolve_pending_commissions(clientkey=self.client, accept_set=[serial1, serial2, 0], reject_set=[serial2, serial3]) self.assertEqual(r, ([serial1], [serial3], [0], [serial2])) r = qh.get_pending_commissions(clientkey=self.client) self.assertEqual(r, [serial2]) # forced commission r = qh.get_quota(holders=[holder]) quotas = { (holder, source, resource1): (limit1, limit1/2+1, limit1/2+2), (holder, source, resource2): (limit2, limit2-2, limit2-1), } self.assertEqual(r, quotas) with assertRaises(NoCapacityError): self.issue_commission( [((holder, source, resource1), limit1/2+1)]) serial = self.issue_commission( [((holder, source, resource1), limit1/2+1)], force=True) r = qh.resolve_pending_commission(self.client, serial, accept=True) self.assertEqual(r, True) r = qh.get_quota(holders=[holder]) quotas = { (holder, source, resource1): (limit1, limit1+2, limit1+3), (holder, source, resource2): (limit2, limit2-2, limit2-1), } self.assertEqual(r, quotas) with assertRaises(NoQuantityError): self.issue_commission( [((holder, source, resource1), -2*limit1)], force=True) # release off upper limit serial = self.issue_commission([((holder, source, resource1), -1)]) r = qh.resolve_pending_commission(self.client, serial) self.assertEqual(r, True) r = qh.get_quota(holders=[holder], resources=[resource1]) quotas = { (holder, source, resource1): (limit1, limit1+1, limit1+2), } self.assertEqual(r, quotas) def test_020_empty_provisions(self): serial = self.issue_commission([]) r = qh.resolve_pending_commission(self.client, serial) self.assertEqual(r, True) def test_030_set(self): holder = 'h0' source = 'system' resource1 = 'r1' resource2 = 'r2' limit1 = 10 limit2 = 20 models.Holding.objects.create( holder=holder, source=source, resource=resource1, usage_min=1, usage_max=1, limit=2) models.Holding.objects.create( holder=holder, source=source, resource=resource2, usage_min=2, usage_max=2, limit=22) qh.set_quota([((holder, source, resource1), limit1), ((holder, source, resource1), limit2)]) r = qh.get_quota(holders=[holder]) self.assertEqual(r, {(holder, source, resource1): (limit2, 1, 1), (holder, source, resource2): (22, 2, 2)})