Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(494)

Unified Diff: tools/telemetry/third_party/gsutil/third_party/protorpc/demos/quotas/backend/quotas/services_test.py

Issue 1260493004: Revert "Add gsutil 4.13 to telemetry/third_party" (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: tools/telemetry/third_party/gsutil/third_party/protorpc/demos/quotas/backend/quotas/services_test.py
diff --git a/tools/telemetry/third_party/gsutil/third_party/protorpc/demos/quotas/backend/quotas/services_test.py b/tools/telemetry/third_party/gsutil/third_party/protorpc/demos/quotas/backend/quotas/services_test.py
deleted file mode 100644
index 1f0dac64b62d0afe52fca14bda00b5e07c869df9..0000000000000000000000000000000000000000
--- a/tools/telemetry/third_party/gsutil/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()

Powered by Google App Engine
This is Rietveld 408576698