Index: tools/telemetry/third_party/webpagereplay/proxyshaper_test.py |
diff --git a/tools/telemetry/third_party/webpagereplay/proxyshaper_test.py b/tools/telemetry/third_party/webpagereplay/proxyshaper_test.py |
deleted file mode 100755 |
index 5c2e3ae229247dffc471c75007564c6fadfd62f0..0000000000000000000000000000000000000000 |
--- a/tools/telemetry/third_party/webpagereplay/proxyshaper_test.py |
+++ /dev/null |
@@ -1,141 +0,0 @@ |
-#!/usr/bin/env python |
-# Copyright 2012 Google Inc. All Rights Reserved. |
-# |
-# 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. |
- |
-"""Unit tests for proxyshaper. |
- |
-Usage: |
-$ ./proxyshaper_test.py |
-""" |
- |
-import proxyshaper |
-import StringIO |
-import unittest |
- |
- |
-# pylint: disable=bad-whitespace |
-VALID_RATES = ( |
- # input, expected_bps |
- ( '384Kbit/s', 384000), |
- ('1536Kbit/s', 1536000), |
- ( '1Mbit/s', 1000000), |
- ( '5Mbit/s', 5000000), |
- ( '2MByte/s', 16000000), |
- ( '0', 0), |
- ( '5', 5), |
- ( 384000, 384000), |
- ) |
- |
-ERROR_RATES = ( |
- '1536KBit/s', # Older versions of dummynet used capital 'B' for bytes. |
- '1Mbyte/s', # Require capital 'B' for bytes. |
- '5bps', |
- ) |
- |
- |
-class TimedTestCase(unittest.TestCase): |
- def assertValuesAlmostEqual(self, expected, actual, tolerance=0.05): |
- """Like the following with nicer default message: |
- assertTrue(expected <= actual + tolerance && |
- expected >= actual - tolerance) |
- """ |
- delta = tolerance * expected |
- if actual > expected + delta or actual < expected - delta: |
- self.fail('%s is not equal to expected %s +/- %s%%' % ( |
- actual, expected, 100 * tolerance)) |
- |
- |
-class RateLimitedFileTest(TimedTestCase): |
- def testReadLimitedBasic(self): |
- num_bytes = 1024 |
- bps = 384000 |
- request_counter = lambda: 1 |
- f = StringIO.StringIO(' ' * num_bytes) |
- limited_f = proxyshaper.RateLimitedFile(request_counter, f, bps) |
- start = proxyshaper.TIMER() |
- self.assertEqual(num_bytes, len(limited_f.read())) |
- expected_ms = 8.0 * num_bytes / bps * 1000.0 |
- actual_ms = (proxyshaper.TIMER() - start) * 1000.0 |
- self.assertValuesAlmostEqual(expected_ms, actual_ms) |
- |
- def testReadlineLimitedBasic(self): |
- num_bytes = 1024 * 8 + 512 |
- bps = 384000 |
- request_counter = lambda: 1 |
- f = StringIO.StringIO(' ' * num_bytes) |
- limited_f = proxyshaper.RateLimitedFile(request_counter, f, bps) |
- start = proxyshaper.TIMER() |
- self.assertEqual(num_bytes, len(limited_f.readline())) |
- expected_ms = 8.0 * num_bytes / bps * 1000.0 |
- actual_ms = (proxyshaper.TIMER() - start) * 1000.0 |
- self.assertValuesAlmostEqual(expected_ms, actual_ms) |
- |
- def testReadLimitedSlowedByMultipleRequests(self): |
- num_bytes = 1024 |
- bps = 384000 |
- request_count = 2 |
- request_counter = lambda: request_count |
- f = StringIO.StringIO(' ' * num_bytes) |
- limited_f = proxyshaper.RateLimitedFile(request_counter, f, bps) |
- start = proxyshaper.TIMER() |
- num_read_bytes = limited_f.read() |
- self.assertEqual(num_bytes, len(num_read_bytes)) |
- expected_ms = 8.0 * num_bytes / (bps / float(request_count)) * 1000.0 |
- actual_ms = (proxyshaper.TIMER() - start) * 1000.0 |
- self.assertValuesAlmostEqual(expected_ms, actual_ms) |
- |
- def testWriteLimitedBasic(self): |
- num_bytes = 1024 * 10 + 350 |
- bps = 384000 |
- request_counter = lambda: 1 |
- f = StringIO.StringIO() |
- limited_f = proxyshaper.RateLimitedFile(request_counter, f, bps) |
- start = proxyshaper.TIMER() |
- limited_f.write(' ' * num_bytes) |
- self.assertEqual(num_bytes, len(limited_f.getvalue())) |
- expected_ms = 8.0 * num_bytes / bps * 1000.0 |
- actual_ms = (proxyshaper.TIMER() - start) * 1000.0 |
- self.assertValuesAlmostEqual(expected_ms, actual_ms) |
- |
- def testWriteLimitedSlowedByMultipleRequests(self): |
- num_bytes = 1024 * 10 |
- bps = 384000 |
- request_count = 2 |
- request_counter = lambda: request_count |
- f = StringIO.StringIO(' ' * num_bytes) |
- limited_f = proxyshaper.RateLimitedFile(request_counter, f, bps) |
- start = proxyshaper.TIMER() |
- limited_f.write(' ' * num_bytes) |
- self.assertEqual(num_bytes, len(limited_f.getvalue())) |
- expected_ms = 8.0 * num_bytes / (bps / float(request_count)) * 1000.0 |
- actual_ms = (proxyshaper.TIMER() - start) * 1000.0 |
- self.assertValuesAlmostEqual(expected_ms, actual_ms) |
- |
- |
-class GetBitsPerSecondTest(unittest.TestCase): |
- def testConvertsValidValues(self): |
- for dummynet_option, expected_bps in VALID_RATES: |
- bps = proxyshaper.GetBitsPerSecond(dummynet_option) |
- self.assertEqual( |
- expected_bps, bps, 'Unexpected result for %s: %s != %s' % ( |
- dummynet_option, expected_bps, bps)) |
- |
- def testRaisesOnUnexpectedValues(self): |
- for dummynet_option in ERROR_RATES: |
- self.assertRaises(proxyshaper.BandwidthValueError, |
- proxyshaper.GetBitsPerSecond, dummynet_option) |
- |
- |
-if __name__ == '__main__': |
- unittest.main() |