| Index: tools/telemetry/third_party/gsutilz/third_party/protorpc/demos/quotas/backend/quotas/services_test.py
|
| diff --git a/tools/telemetry/third_party/gsutilz/third_party/protorpc/demos/quotas/backend/quotas/services_test.py b/tools/telemetry/third_party/gsutilz/third_party/protorpc/demos/quotas/backend/quotas/services_test.py
|
| deleted file mode 100755
|
| index 1f0dac64b62d0afe52fca14bda00b5e07c869df9..0000000000000000000000000000000000000000
|
| --- a/tools/telemetry/third_party/gsutilz/third_party/protorpc/demos/quotas/backend/quotas/services_test.py
|
| +++ /dev/null
|
| @@ -1,441 +0,0 @@
|
| -#!/usr/bin/env python
|
| -#
|
| -# Copyright 2010 Google Inc.
|
| -#
|
| -# Licensed under the Apache License, Version 2.0 (the "License");
|
| -# you may not use this file except in compliance with the License.
|
| -# You may obtain a copy of the License at
|
| -#
|
| -# http://www.apache.org/licenses/LICENSE-2.0
|
| -#
|
| -# Unless required by applicable law or agreed to in writing, software
|
| -# distributed under the License is distributed on an "AS IS" BASIS,
|
| -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
| -# See the License for the specific language governing permissions and
|
| -# limitations under the License.
|
| -#
|
| -
|
| -"""QuotaService tests."""
|
| -
|
| -__author__ = 'rafek@google.com (Rafe Kaplan)'
|
| -
|
| -import unittest
|
| -import time
|
| -
|
| -from protorpc import test_util
|
| -
|
| -from quotas import services
|
| -
|
| -SIMPLE_CONFIG = services.QuotaConfig(
|
| - buckets=[
|
| - services.QuotaConfig.Bucket(
|
| - name='DISK',
|
| - initial_tokens=1000),
|
| - services.QuotaConfig.Bucket(
|
| - name='EMAILS',
|
| - initial_tokens=10,
|
| - refresh_every=10)
|
| - ])
|
| -
|
| -
|
| -class QuotaTestBase(test_util.TestCase):
|
| -
|
| - def MakeState(self, disk=None, emails=None, next_refresh=2000, state=None):
|
| - state = state or {}
|
| - if disk:
|
| - state['DISK'] = disk, None
|
| - if emails:
|
| - state['EMAILS'] = emails, next_refresh
|
| - buckets = dict((bucket.name, bucket) for bucket in SIMPLE_CONFIG.buckets)
|
| - return services.QuotaState(state, buckets)
|
| -
|
| - def setUp(self):
|
| - self.current_time = 1000.5
|
| - self.original_time = time.time
|
| - time.time = self.Time
|
| -
|
| - def tearDown(self):
|
| - time.time = self.original_time
|
| -
|
| - def Time(self):
|
| - return self.current_time
|
| -
|
| -
|
| -class QuotaStateTest(QuotaTestBase):
|
| -
|
| - def setUp(self):
|
| - super(QuotaStateTest, self).setUp()
|
| - self.state = self.MakeState()
|
| -
|
| - def testGetRemainingTokens(self):
|
| - self.state.begin_transaction()
|
| - self.assertEquals(1000, self.state.get_remaining_tokens('DISK'))
|
| - self.assertEquals(10, self.state.get_remaining_tokens('EMAILS'))
|
| - self.assertEquals(None, self.state.get_remaining_tokens('UNKNOWN'))
|
| - self.state.abort_transaction()
|
| -
|
| - self.state = self.MakeState(200, 5)
|
| - self.state.begin_transaction()
|
| - self.assertEquals(200, self.state.get_remaining_tokens('DISK'))
|
| - self.assertEquals(5, self.state.get_remaining_tokens('EMAILS'))
|
| - self.assertEquals(None, self.state.get_remaining_tokens('UNKNOWN'))
|
| - self.state.abort_transaction()
|
| -
|
| - def testCheckQuota(self):
|
| - self.state.begin_transaction()
|
| - self.assertEquals(100, self.state.check_quota('DISK', 100))
|
| - self.assertEquals(1000, self.state.check_quota('DISK', 1000))
|
| - self.assertEquals(1000, self.state.check_quota('DISK', 1001))
|
| - self.assertEquals(-1001, self.state.check_quota('DISK', -1001))
|
| - self.assertEquals(1, self.state.check_quota('EMAILS', 1))
|
| - self.assertEquals(10, self.state.check_quota('EMAILS', 10))
|
| - self.assertEquals(10, self.state.check_quota('EMAILS', 11))
|
| - self.assertEquals(-11, self.state.check_quota('EMAILS', -11))
|
| - self.assertEquals(None, self.state.check_quota('UNKNOWN', 10))
|
| - self.state.abort_transaction()
|
| -
|
| - def testDeduct(self):
|
| - self.state.begin_transaction()
|
| - self.assertEquals(100, self.state.deduct_quota('DISK', 100))
|
| - self.assertEquals(900, self.state.get_remaining_tokens('DISK'))
|
| - self.state.commit_transaction()
|
| -
|
| - self.state.begin_transaction()
|
| - self.assertEquals(900, self.state.deduct_quota('DISK', 900))
|
| - self.assertEquals(0, self.state.get_remaining_tokens('DISK'))
|
| - self.state.commit_transaction()
|
| -
|
| - self.state.begin_transaction()
|
| - self.assertEquals(0, self.state.deduct_quota('DISK', 1))
|
| - self.assertEquals(0, self.state.get_remaining_tokens('DISK'))
|
| - self.state.commit_transaction()
|
| -
|
| - self.state.begin_transaction()
|
| - self.assertEquals(10, self.state.deduct_quota('EMAILS', 11))
|
| - self.assertEquals(0, self.state.get_remaining_tokens('EMAILS'))
|
| - self.state.commit_transaction()
|
| -
|
| - self.state.begin_transaction()
|
| - self.assertEquals(0, self.state.deduct_quota('DISK', 1))
|
| - self.assertEquals(0, self.state.get_remaining_tokens('EMAILS'))
|
| - self.state.commit_transaction()
|
| -
|
| - def testDeductAbort(self):
|
| - self.state.begin_transaction()
|
| - self.assertEquals(100, self.state.deduct_quota('DISK', 100))
|
| - self.assertEquals(900, self.state.get_remaining_tokens('DISK'))
|
| - self.state.abort_transaction()
|
| -
|
| - self.state.begin_transaction()
|
| - self.assertEquals(1000, self.state.get_remaining_tokens('DISK'))
|
| - self.state.abort_transaction()
|
| -
|
| - def testAutoRefresh(self):
|
| - # Use some quota.
|
| - self.state.begin_transaction()
|
| - self.state.deduct_quota('DISK', 100)
|
| - self.state.deduct_quota('EMAILS', 10)
|
| - self.state.commit_transaction()
|
| -
|
| - # Cause refresh.
|
| - self.current_time += 20
|
| -
|
| - # Emails should now be restored.
|
| - self.state.begin_transaction()
|
| - self.assertEquals(900, self.state.get_remaining_tokens('DISK'))
|
| - self.assertEquals(10, self.state.get_remaining_tokens('EMAILS'))
|
| - self.state.commit_transaction()
|
| -
|
| - def testAutoRefreshDuringTransaction(self):
|
| - # Use some quota quota.
|
| - self.state.begin_transaction()
|
| - self.state.deduct_quota('DISK', 100)
|
| - self.state.deduct_quota('EMAILS', 10)
|
| -
|
| - # Refresh is too late for transaction.
|
| - self.current_time += 20
|
| -
|
| - self.assertEquals(900, self.state.get_remaining_tokens('DISK'))
|
| - self.assertEquals(0, self.state.get_remaining_tokens('EMAILS'))
|
| - self.state.commit_transaction()
|
| -
|
| - def testDeductSameBucket(self):
|
| - self.state.begin_transaction()
|
| - self.state.deduct_quota('DISK', 100)
|
| - self.assertRaises(AssertionError,
|
| - self.state.deduct_quota, 'DISK', 100)
|
| - self.state.commit_transaction()
|
| -
|
| -
|
| -class QuotaServiceTest(QuotaTestBase):
|
| -
|
| - def setUp(self):
|
| - super(QuotaServiceTest, self).setUp()
|
| - self.states = {}
|
| - self.service = services.QuotaService(SIMPLE_CONFIG, {})
|
| -
|
| - def MakeCheckQuotaRequest(self, user, quotas):
|
| - return services.QuotaRequest(
|
| - user=user,
|
| - quotas=[
|
| - services.QuotaCheck(name=bucket,
|
| - tokens=tokens,
|
| - mode=services.QuotaCheck.Mode(mode))
|
| - for bucket, tokens, mode in quotas
|
| - ])
|
| -
|
| - def AssertRemaining(self, user, remaining_quotas):
|
| - response = self.service.get_quota_state(
|
| - services.QuotaStateRequest(user=user))
|
| - self.assertEquals(len(remaining_quotas), len(response.bucket_states))
|
| - for (name, remaining), bucket_state in zip(remaining_quotas,
|
| - response.bucket_states):
|
| - self.assertEquals(name, bucket_state.name)
|
| - self.assertEquals(remaining, bucket_state.remaining_tokens)
|
| -
|
| - def testCheckQuotaAll(self):
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 100, 'ALL'),
|
| - ('EMAILS', 10, 'ALL'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.OK, response.all_status)
|
| - self.assertFalse(response.denied)
|
| - self.assertEquals(2, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.OK, response.results[0].status)
|
| - self.assertEquals(100, response.results[0].available)
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.OK, response.results[1].status)
|
| - self.assertEquals(10, response.results[1].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 900), ('EMAILS', 0)])
|
| -
|
| - def testCheckQuotaAll_SomeAvailable(self):
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 100, 'ALL'),
|
| - ('EMAILS', 11, 'ALL'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.SOME, response.all_status)
|
| - self.assertTrue(response.denied)
|
| - self.assertEquals(2, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.OK, response.results[0].status)
|
| - self.assertEquals(100, response.results[0].available)
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.SOME, response.results[1].status)
|
| - self.assertEquals(10, response.results[1].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 1000), ('EMAILS', 10)])
|
| -
|
| - def testCheckQuotaAll_NoneAvailable(self):
|
| - # Deplete all quota.
|
| - self.service.check_quota(self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1000, 'ALL'),
|
| - ('EMAILS', 10, 'ALL'),
|
| - ]))
|
| -
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1, 'ALL'),
|
| - ('EMAILS', 1, 'ALL'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.NONE, response.all_status)
|
| - self.assertTrue(response.denied)
|
| - self.assertEquals(2, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.NONE, response.results[0].status)
|
| - self.assertEquals(0, response.results[0].available)
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.NONE, response.results[1].status)
|
| - self.assertEquals(0, response.results[1].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 0), ('EMAILS', 0)])
|
| -
|
| - def testCheckQuotaAll_SomeAndNone(self):
|
| - # Deplete some quota.
|
| - self.service.check_quota(self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1000, 'ALL'),
|
| - ]))
|
| -
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1, 'ALL'),
|
| - ('EMAILS', 11, 'ALL'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.SOME, response.all_status)
|
| - self.assertTrue(response.denied)
|
| - self.assertEquals(2, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.NONE, response.results[0].status)
|
| - self.assertEquals(0, response.results[0].available)
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.SOME, response.results[1].status)
|
| - self.assertEquals(10, response.results[1].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 0), ('EMAILS', 10)])
|
| -
|
| - def testCheckQuotaSome(self):
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 500, 'ALL'),
|
| - ('EMAILS', 5, 'SOME'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.OK, response.all_status)
|
| - self.assertFalse(response.denied)
|
| - self.assertEquals(2, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.OK, response.results[0].status)
|
| - self.assertEquals(500, response.results[0].available)
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.OK, response.results[1].status)
|
| - self.assertEquals(5, response.results[1].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 500), ('EMAILS', 5)])
|
| -
|
| - def testCheckQuotaSome_SomeAvailable(self):
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 100, 'SOME'),
|
| - ('EMAILS', 11, 'SOME'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.SOME, response.all_status)
|
| - self.assertFalse(response.denied)
|
| - self.assertEquals(2, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.OK, response.results[0].status)
|
| - self.assertEquals(100, response.results[0].available)
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.SOME, response.results[1].status)
|
| - self.assertEquals(10, response.results[1].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 900), ('EMAILS', 0)])
|
| -
|
| - def testCheckQuotaSome_NoneAvailable(self):
|
| - # Deplete all quota.
|
| - self.service.check_quota(self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1000, 'ALL'),
|
| - ('EMAILS', 10, 'ALL'),
|
| - ]))
|
| -
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1, 'SOME'),
|
| - ('EMAILS', 1, 'SOME'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.NONE, response.all_status)
|
| - self.assertTrue(response.denied)
|
| - self.assertEquals(2, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.NONE, response.results[0].status)
|
| - self.assertEquals(0, response.results[0].available)
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.NONE, response.results[1].status)
|
| - self.assertEquals(0, response.results[1].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 0), ('EMAILS', 0)])
|
| -
|
| - def testCheckQuotaSome_SomeAndNone(self):
|
| - # Deplete some quota.
|
| - self.service.check_quota(self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1000, 'ALL'),
|
| - ]))
|
| -
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1, 'SOME'),
|
| - ('EMAILS', 11, 'SOME'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.SOME, response.all_status)
|
| - self.assertTrue(response.denied)
|
| - self.assertEquals(2, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.NONE, response.results[0].status)
|
| - self.assertEquals(0, response.results[0].available)
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.SOME, response.results[1].status)
|
| - self.assertEquals(10, response.results[1].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 0), ('EMAILS', 10)])
|
| -
|
| - def testCheckQuotaCheckAll(self):
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 500, 'CHECK_ALL'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.OK, response.all_status)
|
| - self.assertFalse(response.denied)
|
| - self.assertEquals(1, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.OK, response.results[0].status)
|
| - self.assertEquals(500, response.results[0].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 1000), ('EMAILS', 10)])
|
| -
|
| - def testCheckQuotaCheckAll_SomeAvailable(self):
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1001, 'CHECK_ALL'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.SOME, response.all_status)
|
| - self.assertTrue(response.denied)
|
| - self.assertEquals(1, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.SOME, response.results[0].status)
|
| - self.assertEquals(1000, response.results[0].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 1000), ('EMAILS', 10)])
|
| -
|
| - def testCheckQuotaCheckAll_NoneAvailable(self):
|
| - # Deplete all quota.
|
| - self.service.check_quota(self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1000, 'ALL'),
|
| - ]))
|
| -
|
| - request = self.MakeCheckQuotaRequest('bob',
|
| - [('DISK', 1, 'CHECK_ALL'),
|
| - ])
|
| - response = self.service.check_quota(request)
|
| -
|
| - self.assertEquals(services.CheckResult.Status.NONE, response.all_status)
|
| - self.assertTrue(response.denied)
|
| - self.assertEquals(1, len(response.results))
|
| -
|
| - self.assertEquals(
|
| - services.CheckResult.Status.NONE, response.results[0].status)
|
| - self.assertEquals(0, response.results[0].available)
|
| -
|
| - self.AssertRemaining('bob', [('DISK', 0), ('EMAILS', 10)])
|
| -
|
| -
|
| -if __name__ == '__main__':
|
| - unittest.main()
|
|
|