浏览代码

depot_tools: Separate TestCaseUtils into its own file.

This will make it easier to get rid of super_mox.py

Bug: 984182
Change-Id: Ic6c9646b8a96ef05ac59293a62f68ad7db7dc2d0
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/1754977
Commit-Queue: Edward Lesmes <ehmaldonado@chromium.org>
Reviewed-by: Robbie Iannucci <iannucci@chromium.org>
Edward Lemur 6 年之前
父节点
当前提交
9cafbf4b35
共有 3 个文件被更改,包括 66 次插入54 次删除
  1. 1 51
      testing_support/super_mox.py
  2. 62 0
      testing_support/test_case_utils.py
  3. 3 3
      tests/gclient_scm_test.py

+ 1 - 51
testing_support/super_mox.py

@@ -16,6 +16,7 @@ import sys
 sys.path.append(os.path.dirname(os.path.dirname(__file__)))
 sys.path.append(os.path.dirname(os.path.dirname(__file__)))
 import mox
 import mox
 from third_party.six.moves import StringIO
 from third_party.six.moves import StringIO
+from testing_support.test_case_utils import TestCaseUtils
 
 
 
 
 class IsOneOf(mox.Comparator):
 class IsOneOf(mox.Comparator):
@@ -29,57 +30,6 @@ class IsOneOf(mox.Comparator):
     return '<sequence or map containing \'%s\'>' % str(self._keys)
     return '<sequence or map containing \'%s\'>' % str(self._keys)
 
 
 
 
-class TestCaseUtils(object):
-  """Base class with some additional functionalities. People will usually want
-  to use SuperMoxTestBase instead."""
-  # Backup the separator in case it gets mocked
-  _OS_SEP = os.sep
-  _RANDOM_CHOICE = random.choice
-  _RANDOM_RANDINT = random.randint
-  _STRING_LETTERS = string.ascii_letters
-
-  ## Some utilities for generating arbitrary arguments.
-  def String(self, max_length):
-    return ''.join([self._RANDOM_CHOICE(self._STRING_LETTERS)
-                    for _ in range(self._RANDOM_RANDINT(1, max_length))])
-
-  def Strings(self, max_arg_count, max_arg_length):
-    return [self.String(max_arg_length) for _ in range(max_arg_count)]
-
-  def Args(self, max_arg_count=8, max_arg_length=16):
-    return self.Strings(max_arg_count,
-                        self._RANDOM_RANDINT(1, max_arg_length))
-
-  def _DirElts(self, max_elt_count=4, max_elt_length=8):
-    return self._OS_SEP.join(self.Strings(max_elt_count, max_elt_length))
-
-  def Dir(self, max_elt_count=4, max_elt_length=8):
-    return (self._RANDOM_CHOICE((self._OS_SEP, '')) +
-            self._DirElts(max_elt_count, max_elt_length))
-
-  def RootDir(self, max_elt_count=4, max_elt_length=8):
-    return self._OS_SEP + self._DirElts(max_elt_count, max_elt_length)
-
-  def compareMembers(self, obj, members):
-    """If you add a member, be sure to add the relevant test!"""
-    # Skip over members starting with '_' since they are usually not meant to
-    # be for public use.
-    actual_members = [x for x in sorted(dir(obj))
-                      if not x.startswith('_')]
-    expected_members = sorted(members)
-    if actual_members != expected_members:
-      diff = ([i for i in actual_members if i not in expected_members] +
-              [i for i in expected_members if i not in actual_members])
-      print(diff, file=sys.stderr)
-    # pylint: disable=no-member
-    self.assertEqual(actual_members, expected_members)
-
-  def setUp(self):
-    self.root_dir = self.Dir()
-    self.args = self.Args()
-    self.relpath = self.String(200)
-
-
 class StdoutCheck(object):
 class StdoutCheck(object):
   def setUp(self):
   def setUp(self):
     # Override the mock with a StringIO, it's much less painful to test.
     # Override the mock with a StringIO, it's much less painful to test.

+ 62 - 0
testing_support/test_case_utils.py

@@ -0,0 +1,62 @@
+# Copyright (c) 2019 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""Simplify unit tests based on pymox."""
+
+from __future__ import print_function
+
+import os
+import random
+import string
+
+
+class TestCaseUtils(object):
+  """Base class with some additional functionalities. People will usually want
+  to use SuperMoxTestBase instead."""
+  # Backup the separator in case it gets mocked
+  _OS_SEP = os.sep
+  _RANDOM_CHOICE = random.choice
+  _RANDOM_RANDINT = random.randint
+  _STRING_LETTERS = string.ascii_letters
+
+  ## Some utilities for generating arbitrary arguments.
+  def String(self, max_length):
+    return ''.join([self._RANDOM_CHOICE(self._STRING_LETTERS)
+                    for _ in range(self._RANDOM_RANDINT(1, max_length))])
+
+  def Strings(self, max_arg_count, max_arg_length):
+    return [self.String(max_arg_length) for _ in range(max_arg_count)]
+
+  def Args(self, max_arg_count=8, max_arg_length=16):
+    return self.Strings(max_arg_count,
+                        self._RANDOM_RANDINT(1, max_arg_length))
+
+  def _DirElts(self, max_elt_count=4, max_elt_length=8):
+    return self._OS_SEP.join(self.Strings(max_elt_count, max_elt_length))
+
+  def Dir(self, max_elt_count=4, max_elt_length=8):
+    return (self._RANDOM_CHOICE((self._OS_SEP, '')) +
+            self._DirElts(max_elt_count, max_elt_length))
+
+  def RootDir(self, max_elt_count=4, max_elt_length=8):
+    return self._OS_SEP + self._DirElts(max_elt_count, max_elt_length)
+
+  def compareMembers(self, obj, members):
+    """If you add a member, be sure to add the relevant test!"""
+    # Skip over members starting with '_' since they are usually not meant to
+    # be for public use.
+    actual_members = [x for x in sorted(dir(obj))
+                      if not x.startswith('_')]
+    expected_members = sorted(members)
+    if actual_members != expected_members:
+      diff = ([i for i in actual_members if i not in expected_members] +
+              [i for i in expected_members if i not in actual_members])
+      print(diff, file=sys.stderr)
+    # pylint: disable=no-member
+    self.assertEqual(actual_members, expected_members)
+
+  def setUp(self):
+    self.root_dir = self.Dir()
+    self.args = self.Args()
+    self.relpath = self.String(200)

+ 3 - 3
tests/gclient_scm_test.py

@@ -28,7 +28,7 @@ sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 
 from third_party import mock
 from third_party import mock
 from testing_support import fake_repos
 from testing_support import fake_repos
-from testing_support.super_mox import TestCaseUtils
+from testing_support import test_case_utils
 
 
 import gclient_scm
 import gclient_scm
 import git_cache
 import git_cache
@@ -73,7 +73,7 @@ class BasicTests(unittest.TestCase):
     self.assertEqual(mockCapture.mock_calls, expected_calls)
     self.assertEqual(mockCapture.mock_calls, expected_calls)
 
 
 
 
-class BaseGitWrapperTestCase(unittest.TestCase, TestCaseUtils):
+class BaseGitWrapperTestCase(unittest.TestCase, test_case_utils.TestCaseUtils):
   """This class doesn't use pymox."""
   """This class doesn't use pymox."""
   class OptionsObject(object):
   class OptionsObject(object):
     def __init__(self, verbose=False, revision=None):
     def __init__(self, verbose=False, revision=None):
@@ -199,7 +199,7 @@ from :3
 
 
   def setUp(self):
   def setUp(self):
     unittest.TestCase.setUp(self)
     unittest.TestCase.setUp(self)
-    TestCaseUtils.setUp(self)
+    test_case_utils.TestCaseUtils.setUp(self)
     self.url = 'git://foo'
     self.url = 'git://foo'
     # The .git suffix allows gclient_scm to recognize the dir as a git repo
     # The .git suffix allows gclient_scm to recognize the dir as a git repo
     # when cloning it locally
     # when cloning it locally