| OLD | NEW |
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # Copyright 2012 Google Inc. All Rights Reserved. | |
| 3 # | |
| 4 # Licensed under the Apache License, Version 2.0 (the "License"); | |
| 5 # you may not use this file except in compliance with the License. | |
| 6 # You may obtain a copy of the License at | |
| 7 # | |
| 8 # http://www.apache.org/licenses/LICENSE-2.0 | |
| 9 # | |
| 10 # Unless required by applicable law or agreed to in writing, software | |
| 11 # distributed under the License is distributed on an "AS IS" BASIS, | |
| 12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| 13 # See the License for the specific language governing permissions and | |
| 14 # limitations under the License. | |
| 15 | |
| 16 """Unit tests for replay. | |
| 17 | |
| 18 Usage: | |
| 19 $ ./replay_test.py | |
| 20 """ | |
| 21 | |
| 22 import replay | |
| 23 import unittest | |
| 24 | |
| 25 | |
| 26 class MockOptions(dict): | |
| 27 """A dict with items that can be accessed as attributes.""" | |
| 28 def __getattr__(self, name): | |
| 29 return self[name] | |
| 30 | |
| 31 | |
| 32 class OptionsWrapperTest(unittest.TestCase): | |
| 33 | |
| 34 def testNoTrafficShapingByDefault(self): | |
| 35 parser = replay.GetParser() | |
| 36 options = parser.parse_args([]) | |
| 37 options = replay.OptionsWrapper(options, parser) | |
| 38 self.assertEqual({}, options.shaping_dns) | |
| 39 self.assertEqual({}, options.shaping_http) | |
| 40 self.assertEqual({}, options.shaping_dummynet) | |
| 41 | |
| 42 def testShapingProxyWithoutOptionsGivesEmptySettings(self): | |
| 43 parser = replay.GetParser() | |
| 44 options = parser.parse_args(['--shaping=proxy']) | |
| 45 options = replay.OptionsWrapper(options, parser) | |
| 46 self.assertEqual({}, options.shaping_dns) | |
| 47 self.assertEqual({}, options.shaping_http) | |
| 48 self.assertEqual({}, options.shaping_dummynet) | |
| 49 | |
| 50 def testShapingProxyWithNetOption(self): | |
| 51 parser = replay.GetParser() | |
| 52 options = parser.parse_args(['--shaping=proxy', '--net=cable']) | |
| 53 options = replay.OptionsWrapper(options, parser) | |
| 54 expected_http = { | |
| 55 'down_bandwidth': '5Mbit/s', 'delay_ms': '28', 'up_bandwidth': '1Mbit/s' | |
| 56 } | |
| 57 self.assertEqual({'delay_ms': '28'}, options.shaping_dns) | |
| 58 self.assertEqual(expected_http, options.shaping_http) | |
| 59 self.assertEqual({}, options.shaping_dummynet) | |
| 60 | |
| 61 def testNetOptionUsesDummynetByDefault(self): | |
| 62 parser = replay.GetParser() | |
| 63 options = parser.parse_args(['--net=cable']) | |
| 64 options = replay.OptionsWrapper(options, parser) | |
| 65 expected_dummynet = { | |
| 66 'down_bandwidth': '5Mbit/s', 'delay_ms': '28', 'up_bandwidth': '1Mbit/s' | |
| 67 } | |
| 68 self.assertEqual({}, options.shaping_dns) | |
| 69 self.assertEqual({}, options.shaping_http) | |
| 70 self.assertEqual(expected_dummynet, options.shaping_dummynet) | |
| 71 | |
| 72 def testPacketLossForDummynet(self): | |
| 73 parser = replay.GetParser() | |
| 74 options = parser.parse_args(['--packet_loss_rate=12']) | |
| 75 options = replay.OptionsWrapper(options, parser) | |
| 76 self.assertEqual({'packet_loss_rate': '12'}, options.shaping_dummynet) | |
| 77 | |
| 78 def testIgnoredProxyShapingOptions(self): | |
| 79 parser = replay.GetParser() | |
| 80 options = parser.parse_args( | |
| 81 ['--packet_loss_rate=12', '--init_cwnd=10', '--shaping=proxy']) | |
| 82 options = replay.OptionsWrapper(options, parser) | |
| 83 self.assertEqual({}, options.shaping_dns) | |
| 84 self.assertEqual({}, options.shaping_http) | |
| 85 self.assertEqual({}, options.shaping_dummynet) | |
| 86 | |
| 87 | |
| 88 if __name__ == '__main__': | |
| 89 unittest.main() | |
| OLD | NEW |