| OLD | NEW |
| (Empty) |
| 1 # -*- coding: utf-8 -*- | |
| 2 # Copyright 2013 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 """Integration tests for perfdiag command.""" | |
| 16 | |
| 17 from __future__ import absolute_import | |
| 18 | |
| 19 import os | |
| 20 import socket | |
| 21 | |
| 22 import gslib.tests.testcase as testcase | |
| 23 from gslib.tests.util import ObjectToURI as suri | |
| 24 from gslib.tests.util import unittest | |
| 25 from gslib.util import IS_WINDOWS | |
| 26 | |
| 27 | |
| 28 class TestPerfDiag(testcase.GsUtilIntegrationTestCase): | |
| 29 """Integration tests for perfdiag command.""" | |
| 30 | |
| 31 # We want to test that perfdiag works both when connecting to the standard gs | |
| 32 # endpoint, and when connecting to a specific IP or host while setting the | |
| 33 # host header. For the 2nd case we resolve storage.googleapis.com to a | |
| 34 # specific IP and connect to that explicitly. | |
| 35 _gs_ip = socket.gethostbyname('storage.googleapis.com') | |
| 36 _custom_endpoint_flags = [ | |
| 37 '-o', 'Credentials:gs_host=' + _gs_ip, | |
| 38 '-o', 'Credentials:gs_host_header=storage.googleapis.com', | |
| 39 # TODO: gsutil-beta: Add host header support for JSON | |
| 40 '-o', 'Boto:https_validate_certificates=False'] | |
| 41 | |
| 42 def _should_run_with_custom_endpoints(self): | |
| 43 # Host headers are only supported for XML, and not when | |
| 44 # using environment variables for proxies. | |
| 45 return self.test_api == 'XML' and not (os.environ.get('http_proxy') or | |
| 46 os.environ.get('https_proxy') or | |
| 47 os.environ.get('HTTPS_PROXY')) | |
| 48 | |
| 49 def test_latency(self): | |
| 50 bucket_uri = self.CreateBucket() | |
| 51 cmd = ['perfdiag', '-n', '1', '-t', 'lat', suri(bucket_uri)] | |
| 52 self.RunGsUtil(cmd) | |
| 53 if self._should_run_with_custom_endpoints(): | |
| 54 self.RunGsUtil(self._custom_endpoint_flags + cmd) | |
| 55 self.AssertNObjectsInBucket(bucket_uri, 0, versioned=True) | |
| 56 | |
| 57 def _run_basic_wthru_or_rthru(self, test_name, num_processes, num_threads): | |
| 58 bucket_uri = self.CreateBucket() | |
| 59 cmd = ['perfdiag', '-n', str(num_processes * num_threads), | |
| 60 '-s', '1024', '-c', str(num_processes), | |
| 61 '-k', str(num_threads), '-t', test_name, suri(bucket_uri)] | |
| 62 self.RunGsUtil(cmd) | |
| 63 if self._should_run_with_custom_endpoints(): | |
| 64 self.RunGsUtil(self._custom_endpoint_flags + cmd) | |
| 65 self.AssertNObjectsInBucket(bucket_uri, 0, versioned=True) | |
| 66 | |
| 67 def test_write_throughput_single_process_single_thread(self): | |
| 68 self._run_basic_wthru_or_rthru('wthru', 1, 1) | |
| 69 | |
| 70 def test_write_throughput_single_process_multi_thread(self): | |
| 71 self._run_basic_wthru_or_rthru('wthru', 1, 2) | |
| 72 | |
| 73 @unittest.skipIf(IS_WINDOWS, 'Multiprocessing is not supported on Windows') | |
| 74 def test_write_throughput_multi_process_single_thread(self): | |
| 75 self._run_basic_wthru_or_rthru('wthru', 2, 1) | |
| 76 | |
| 77 @unittest.skipIf(IS_WINDOWS, 'Multiprocessing is not supported on Windows') | |
| 78 def test_write_throughput_multi_process_multi_thread(self): | |
| 79 self._run_basic_wthru_or_rthru('wthru', 2, 2) | |
| 80 | |
| 81 def test_read_throughput_single_process_single_thread(self): | |
| 82 self._run_basic_wthru_or_rthru('rthru', 1, 1) | |
| 83 | |
| 84 def test_read_throughput_single_process_multi_thread(self): | |
| 85 self._run_basic_wthru_or_rthru('rthru', 1, 2) | |
| 86 | |
| 87 @unittest.skipIf(IS_WINDOWS, 'Multiprocessing is not supported on Windows') | |
| 88 def test_read_throughput_multi_process_single_thread(self): | |
| 89 self._run_basic_wthru_or_rthru('rthru', 2, 1) | |
| 90 | |
| 91 @unittest.skipIf(IS_WINDOWS, 'Multiprocessing is not supported on Windows') | |
| 92 def test_read_throughput_multi_process_multi_thread(self): | |
| 93 self._run_basic_wthru_or_rthru('rthru', 2, 2) | |
| 94 | |
| 95 def test_input_output(self): | |
| 96 outpath = self.CreateTempFile() | |
| 97 bucket_uri = self.CreateBucket() | |
| 98 self.RunGsUtil(['perfdiag', '-o', outpath, '-n', '1', '-t', 'lat', | |
| 99 suri(bucket_uri)]) | |
| 100 self.RunGsUtil(['perfdiag', '-i', outpath]) | |
| 101 | |
| 102 def test_invalid_size(self): | |
| 103 stderr = self.RunGsUtil( | |
| 104 ['perfdiag', '-n', '1', '-s', 'foo', '-t', 'wthru', 'gs://foobar'], | |
| 105 expected_status=1, return_stderr=True) | |
| 106 self.assertIn('Invalid -s', stderr) | |
| 107 | |
| 108 def test_toobig_size(self): | |
| 109 stderr = self.RunGsUtil( | |
| 110 ['perfdiag', '-n', '1', '-s', '3pb', '-t', 'wthru', 'gs://foobar'], | |
| 111 expected_status=1, return_stderr=True) | |
| 112 self.assertIn('Maximum throughput file size', stderr) | |
| 113 | |
| 114 def test_listing(self): | |
| 115 bucket_uri = self.CreateBucket() | |
| 116 stdout = self.RunGsUtil( | |
| 117 ['perfdiag', '-n', '1', '-t', 'list', suri(bucket_uri)], | |
| 118 return_stdout=True) | |
| 119 self.assertIn('Number of listing calls made:', stdout) | |
| 120 self.AssertNObjectsInBucket(bucket_uri, 0, versioned=True) | |
| OLD | NEW |