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>changes/77/1754977/4
parent
614050efa0
commit
9cafbf4b35
@ -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)
|
||||||
Loading…
Reference in New Issue