1 files changed, 48 insertions, 49 deletions

diff --git a/src/python/grpcio_tests/tests/unit/framework/common/test_control.py b/src/python/grpcio_tests/tests/unit/framework/common/test_control.py
index 088e2f8b88..af08731b1e 100644
--- a/src/python/grpcio_tests/tests/unit/framework/common/test_control.py
+++ b/src/python/grpcio_tests/tests/unit/framework/common/test_control.py
@@ -26,7 +26,6 @@
 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-
 """Code for instructing systems under test to block or fail."""
 
 import abc
@@ -37,7 +36,7 @@ import six
 
 
 class Defect(Exception):
-  """Simulates a programming defect raised into in a system under test.
+    """Simulates a programming defect raised into in a system under test.
 
   Use of a standard exception type is too easily misconstrued as an actual
   defect in either the test infrastructure or the system under test.
@@ -45,7 +44,7 @@ class Defect(Exception):
 
 
 class Control(six.with_metaclass(abc.ABCMeta)):
-  """An object that accepts program control from a system under test.
+    """An object that accepts program control from a system under test.
 
   Systems under test passed a Control should call its control() method
   frequently during execution. The control() method may block, raise an
@@ -53,61 +52,61 @@ class Control(six.with_metaclass(abc.ABCMeta)):
   the system under test to simulate hanging, failing, or functioning.
   """
 
-  @abc.abstractmethod
-  def control(self):
-    """Potentially does anything."""
-    raise NotImplementedError()
+    @abc.abstractmethod
+    def control(self):
+        """Potentially does anything."""
+        raise NotImplementedError()
 
 
 class PauseFailControl(Control):
-  """A Control that can be used to pause or fail code under control.
+    """A Control that can be used to pause or fail code under control.
 
   This object is only safe for use from two threads: one of the system under
   test calling control and the other from the test system calling pause,
   block_until_paused, and fail.
   """
 
-  def __init__(self):
-    self._condition = threading.Condition()
-    self._pause = False
-    self._paused = False
-    self._fail = False
-
-  def control(self):
-    with self._condition:
-      if self._fail:
-        raise Defect()
-
-      while self._pause:
-        self._paused = True
-        self._condition.notify_all()
-        self._condition.wait()
-      self._paused = False
-
-  @contextlib.contextmanager
-  def pause(self):
-    """Pauses code under control while controlling code is in context."""
-    with self._condition:
-      self._pause = True
-    yield
-    with self._condition:
-      self._pause = False
-      self._condition.notify_all()
-
-  def block_until_paused(self):
-    """Blocks controlling code until code under control is paused.
+    def __init__(self):
+        self._condition = threading.Condition()
+        self._pause = False
+        self._paused = False
+        self._fail = False
+
+    def control(self):
+        with self._condition:
+            if self._fail:
+                raise Defect()
+
+            while self._pause:
+                self._paused = True
+                self._condition.notify_all()
+                self._condition.wait()
+            self._paused = False
+
+    @contextlib.contextmanager
+    def pause(self):
+        """Pauses code under control while controlling code is in context."""
+        with self._condition:
+            self._pause = True
+        yield
+        with self._condition:
+            self._pause = False
+            self._condition.notify_all()
+
+    def block_until_paused(self):
+        """Blocks controlling code until code under control is paused.
 
     May only be called within the context of a pause call.
     """
-    with self._condition:
-      while not self._paused:
-        self._condition.wait()
-
-  @contextlib.contextmanager
-  def fail(self):
-    """Fails code under control while controlling code is in context."""
-    with self._condition:
-      self._fail = True
-    yield
-    with self._condition:
-      self._fail = False
+        with self._condition:
+            while not self._paused:
+                self._condition.wait()
+
+    @contextlib.contextmanager
+    def fail(self):
+        """Fails code under control while controlling code is in context."""
+        with self._condition:
+            self._fail = True
+        yield
+        with self._condition:
+            self._fail = False