Index: tests/gsutil_test.py |
diff --git a/tests/gsutil_test.py b/tests/gsutil_test.py |
index 30648279d363408f5528f3f4cbb096fa027b3b5f..76570dd31034727b4c38346ea6e84fe383f9b234 100755 |
--- a/tests/gsutil_test.py |
+++ b/tests/gsutil_test.py |
@@ -7,16 +7,17 @@ |
import __builtin__ |
-import unittest |
+import base64 |
import hashlib |
-import zipfile |
+import json |
+import os |
import shutil |
+import subprocess |
import sys |
-import base64 |
import tempfile |
-import json |
-import os |
+import unittest |
import urllib2 |
+import zipfile |
# Add depot_tools to path |
@@ -62,8 +63,6 @@ class FakeCall(object): |
message = 'Expected:\n args: %s\n kwargs: %s\n' % (exp_args, exp_kwargs) |
message += 'Got:\n args: %s\n kwargs: %s\n' % (args, kwargs) |
raise TestError(message) |
- if isinstance(exp_returns, Exception): |
- raise exp_returns |
return exp_returns |
@@ -72,15 +71,15 @@ class GsutilUnitTests(unittest.TestCase): |
self.fake = FakeCall() |
self.tempdir = tempfile.mkdtemp() |
self.old_urlopen = getattr(urllib2, 'urlopen') |
- self.old_call = getattr(gsutil, 'call') |
+ self.old_call = getattr(subprocess, 'call') |
setattr(urllib2, 'urlopen', self.fake) |
- setattr(gsutil, 'call', self.fake) |
+ setattr(subprocess, 'call', self.fake) |
def tearDown(self): |
self.assertEqual(self.fake.expectations, []) |
shutil.rmtree(self.tempdir) |
setattr(urllib2, 'urlopen', self.old_urlopen) |
- setattr(gsutil, 'call', self.old_call) |
+ setattr(subprocess, 'call', self.old_call) |
def test_download_gsutil(self): |
version = '4.2' |
@@ -126,8 +125,8 @@ class GsutilUnitTests(unittest.TestCase): |
os.makedirs(gsutil_dir) |
self.fake.add_expectation( |
- [sys.executable, gsutil_bin, 'version'], verbose=False, |
- _returns=gsutil.SubprocessError()) |
+ [sys.executable, gsutil_bin, 'version'], stdout=subprocess.PIPE, |
+ stderr=subprocess.STDOUT, _returns=1) |
with open(gsutil_bin, 'w') as f: |
f.write('Foobar') |
@@ -140,8 +139,8 @@ class GsutilUnitTests(unittest.TestCase): |
with open(tempzip, 'rb') as f: |
self.fake.add_expectation(url, _returns=Buffer(f.read())) |
self.fake.add_expectation( |
- [sys.executable, gsutil_bin, 'version'], verbose=False, |
- _returns=gsutil.SubprocessError()) |
+ [sys.executable, gsutil_bin, 'version'], stdout=subprocess.PIPE, |
+ stderr=subprocess.STDOUT, _returns=1) |
# This should delete the old bin and rewrite it with 'Fake gsutil' |
self.assertRaises( |
@@ -160,7 +159,8 @@ class GsutilUnitTests(unittest.TestCase): |
# Mock out call(). |
self.fake.add_expectation( |
- [sys.executable, gsutil_bin, 'version'], verbose=False, _returns=True) |
+ [sys.executable, gsutil_bin, 'version'], |
+ stdout=subprocess.PIPE, stderr=subprocess.STDOUT, _returns=0) |
with open(gsutil_bin, 'w') as f: |
f.write('Foobar') |