| Index: tests/super_mox.py
|
| diff --git a/tests/super_mox.py b/tests/super_mox.py
|
| deleted file mode 100644
|
| index 62113e210b57c9b5641aecb45624ebe155906a84..0000000000000000000000000000000000000000
|
| --- a/tests/super_mox.py
|
| +++ /dev/null
|
| @@ -1,153 +0,0 @@
|
| -# Copyright (c) 2011 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."""
|
| -
|
| -import os
|
| -import random
|
| -import shutil
|
| -import string
|
| -import StringIO
|
| -import subprocess
|
| -import sys
|
| -
|
| -sys.path.append(os.path.dirname(os.path.dirname(__file__)))
|
| -from third_party.pymox import mox
|
| -
|
| -
|
| -class IsOneOf(mox.Comparator):
|
| - def __init__(self, keys):
|
| - self._keys = keys
|
| -
|
| - def equals(self, rhs):
|
| - return rhs in self._keys
|
| -
|
| - def __repr__(self):
|
| - 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.letters
|
| -
|
| - ## Some utilities for generating arbitrary arguments.
|
| - def String(self, max_length):
|
| - return ''.join([self._RANDOM_CHOICE(self._STRING_LETTERS)
|
| - for _ in xrange(self._RANDOM_RANDINT(1, max_length))])
|
| -
|
| - def Strings(self, max_arg_count, max_arg_length):
|
| - return [self.String(max_arg_length) for _ in xrange(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 SvnUrl(self, max_elt_count=4, max_elt_length=8):
|
| - return ('svn://random_host:port/a' +
|
| - self._DirElts(max_elt_count, max_elt_length
|
| - ).replace(self._OS_SEP, '/'))
|
| -
|
| - 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 >> sys.stderr, diff
|
| - # pylint: disable=E1101
|
| - self.assertEqual(actual_members, expected_members)
|
| -
|
| - def setUp(self):
|
| - self.root_dir = self.Dir()
|
| - self.args = self.Args()
|
| - self.relpath = self.String(200)
|
| -
|
| - def tearDown(self):
|
| - pass
|
| -
|
| -
|
| -class StdoutCheck(object):
|
| - def setUp(self):
|
| - # Override the mock with a StringIO, it's much less painful to test.
|
| - self._old_stdout = sys.stdout
|
| - sys.stdout = StringIO.StringIO()
|
| - sys.stdout.flush = lambda: None
|
| -
|
| - def tearDown(self):
|
| - try:
|
| - # If sys.stdout was used, self.checkstdout() must be called.
|
| - # pylint: disable=E1101
|
| - self.assertEquals('', sys.stdout.getvalue())
|
| - except AttributeError:
|
| - pass
|
| - sys.stdout = self._old_stdout
|
| -
|
| - def checkstdout(self, expected):
|
| - value = sys.stdout.getvalue()
|
| - sys.stdout.close()
|
| - # pylint: disable=E1101
|
| - self.assertEquals(expected, value)
|
| -
|
| -
|
| -class SuperMoxTestBase(TestCaseUtils, StdoutCheck, mox.MoxTestBase):
|
| - def setUp(self):
|
| - """Patch a few functions with know side-effects."""
|
| - TestCaseUtils.setUp(self)
|
| - mox.MoxTestBase.setUp(self)
|
| - os_to_mock = ('chdir', 'chown', 'close', 'closerange', 'dup', 'dup2',
|
| - 'fchdir', 'fchmod', 'fchown', 'fdopen', 'getcwd', 'getpid', 'lseek',
|
| - 'makedirs', 'mkdir', 'open', 'popen', 'popen2', 'popen3', 'popen4',
|
| - 'read', 'remove', 'removedirs', 'rename', 'renames', 'rmdir', 'symlink',
|
| - 'system', 'tmpfile', 'walk', 'write')
|
| - self.MockList(os, os_to_mock)
|
| - os_path_to_mock = ('abspath', 'exists', 'getsize', 'isdir', 'isfile',
|
| - 'islink', 'ismount', 'lexists', 'realpath', 'samefile', 'walk')
|
| - self.MockList(os.path, os_path_to_mock)
|
| - self.MockList(shutil, ('rmtree'))
|
| - self.MockList(subprocess, ('call', 'Popen'))
|
| - # Don't mock stderr since it confuses unittests.
|
| - self.MockList(sys, ('stdin'))
|
| - StdoutCheck.setUp(self)
|
| -
|
| - def tearDown(self):
|
| - StdoutCheck.tearDown(self)
|
| - TestCaseUtils.tearDown(self)
|
| - mox.MoxTestBase.tearDown(self)
|
| -
|
| - def MockList(self, parent, items_to_mock):
|
| - for item in items_to_mock:
|
| - # Skip over items not present because of OS-specific implementation,
|
| - # implemented only in later python version, etc.
|
| - if hasattr(parent, item):
|
| - try:
|
| - self.mox.StubOutWithMock(parent, item)
|
| - except TypeError, e:
|
| - raise TypeError(
|
| - 'Couldn\'t mock %s in %s: %s' % (item, parent.__name__, e))
|
| -
|
| - def UnMock(self, obj, name):
|
| - """Restore an object inside a test."""
|
| - for (parent, old_child, child_name) in self.mox.stubs.cache:
|
| - if parent == obj and child_name == name:
|
| - setattr(parent, child_name, old_child)
|
| - break
|
|
|