Index: testing_support/trial_dir.py |
diff --git a/testing_support/trial_dir.py b/testing_support/trial_dir.py |
index 16fab8045c9610cbcfd204adb1cdac2a78a7a24b..1f2c9285a33c858404fc395b8419e5d8059405fb 100644 |
--- a/testing_support/trial_dir.py |
+++ b/testing_support/trial_dir.py |
@@ -1,4 +1,3 @@ |
-#!/usr/bin/env python |
# 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. |
@@ -9,7 +8,8 @@ import logging |
import os |
import sys |
import tempfile |
-import unittest |
+ |
+from testing_support import auto_stub |
import gclient_utils |
@@ -79,15 +79,15 @@ class TrialDirMixIn(object): |
return self.trial.root_dir |
-class TestCase(unittest.TestCase, TrialDirMixIn): |
+class TestCase(auto_stub.TestCase, TrialDirMixIn): |
"""Base unittest class that cleans off a trial directory in tearDown().""" |
def setUp(self): |
- unittest.TestCase.setUp(self) |
+ auto_stub.TestCase.setUp(self) |
TrialDirMixIn.setUp(self) |
def tearDown(self): |
TrialDirMixIn.tearDown(self) |
- unittest.TestCase.tearDown(self) |
+ auto_stub.TestCase.tearDown(self) |
if '-l' in sys.argv: |