depot_tools: Stop using mox on gclient_utils_test.py.

Bug: 984182
Change-Id: Ie29f06c817824c009656cf81d93a605abebe948a
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/1754976
Reviewed-by: Robbie Iannucci <iannucci@chromium.org>
Commit-Queue: Edward Lesmes <ehmaldonado@chromium.org>
changes/76/1754976/2
Edward Lemur 6 years ago committed by Commit Bot
parent 26a8b9f356
commit 614050efa0

@ -9,25 +9,19 @@ from __future__ import unicode_literals
import io
import os
import sys
import time
import unittest
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from testing_support.super_mox import SuperMoxTestBase
from testing_support import trial_dir
from third_party import mock
import gclient_utils
import subprocess2
class GclientUtilBase(SuperMoxTestBase):
def setUp(self):
super(GclientUtilBase, self).setUp()
gclient_utils.sys.stdout.flush = lambda: None
self.mox.StubOutWithMock(subprocess2, 'Popen')
self.mox.StubOutWithMock(subprocess2, 'communicate')
class CheckCallAndFilterTestCase(GclientUtilBase):
class CheckCallAndFilterTestCase(unittest.TestCase):
class ProcessIdMock(object):
def __init__(self, test_string, return_code=0):
self.stdout = io.BytesIO(test_string.encode('utf-8'))
@ -37,23 +31,21 @@ class CheckCallAndFilterTestCase(GclientUtilBase):
def wait(self):
return self.return_code
def testCheckCallAndFilter(self):
def setUp(self):
super(CheckCallAndFilterTestCase, self).setUp()
mock.patch('sys.stdout', io.StringIO()).start()
mock.patch('sys.stdout.flush', lambda: None).start()
self.addCleanup(mock.patch.stopall)
@mock.patch('subprocess2.Popen')
def testCheckCallAndFilter(self, mockPopen):
cwd = 'bleh'
args = ['boo', 'foo', 'bar']
test_string = 'ahah\naccb\nallo\naddb\n'
# pylint: disable=no-member
subprocess2.Popen(
args,
cwd=cwd,
stdout=subprocess2.PIPE,
stderr=subprocess2.STDOUT,
bufsize=0).AndReturn(self.ProcessIdMock(test_string))
mockPopen.return_value = self.ProcessIdMock(test_string)
os.getcwd()
self.mox.ReplayAll()
line_list = []
result = gclient_utils.CheckCallAndFilter(
args, cwd=cwd, show_header=True, always_show_header=True,
filter_fn=line_list.append)
@ -67,29 +59,21 @@ class CheckCallAndFilterTestCase(GclientUtilBase):
'addb',
''])
def testCheckCallAndFilter_RetryOnce(self):
mockPopen.assert_called_with(
args, cwd=cwd, stdout=subprocess2.PIPE, stderr=subprocess2.STDOUT,
bufsize=0)
@mock.patch('time.sleep')
@mock.patch('subprocess2.Popen')
def testCheckCallAndFilter_RetryOnce(self, mockPopen, mockTime):
cwd = 'bleh'
args = ['boo', 'foo', 'bar']
test_string = 'ahah\naccb\nallo\naddb\n'
# pylint: disable=no-member
subprocess2.Popen(
args,
cwd=cwd,
stdout=subprocess2.PIPE,
stderr=subprocess2.STDOUT,
bufsize=0).AndReturn(self.ProcessIdMock(test_string, 1))
os.getcwd()
# pylint: disable=no-member
subprocess2.Popen(
args,
cwd=cwd,
stdout=subprocess2.PIPE,
stderr=subprocess2.STDOUT,
bufsize=0).AndReturn(self.ProcessIdMock(test_string, 0))
self.mox.ReplayAll()
mockPopen.side_effect = [
self.ProcessIdMock(test_string, 1),
self.ProcessIdMock(test_string, 0),
]
line_list = []
result = gclient_utils.CheckCallAndFilter(
@ -113,12 +97,26 @@ class CheckCallAndFilterTestCase(GclientUtilBase):
'',
])
self.checkstdout(
mockTime.assert_called_with(gclient_utils.RETRY_INITIAL_SLEEP)
self.assertEqual(
mockPopen.mock_calls,
[
mock.call(
args, cwd=cwd, stdout=subprocess2.PIPE,
stderr=subprocess2.STDOUT, bufsize=0),
mock.call(
args, cwd=cwd, stdout=subprocess2.PIPE,
stderr=subprocess2.STDOUT, bufsize=0),
])
self.assertEqual(
sys.stdout.getvalue(),
'WARNING: subprocess \'"boo" "foo" "bar"\' in bleh failed; will retry '
'after a short nap...\n')
class SplitUrlRevisionTestCase(GclientUtilBase):
class SplitUrlRevisionTestCase(unittest.TestCase):
def testSSHUrl(self):
url = "ssh://test@example.com/test.git"
rev = "ac345e52dc"

Loading…
Cancel
Save