selftest.py: Use testtools TestCase class.

Autobuild-User: Jelmer Vernooij <jelmer@samba.org>
Autobuild-Date: Sun Mar  4 19:38:25 CET 2012 on sn-devel-104

diff --git a/selftest/tests/__init__.py b/selftest/tests/__init__.py
index 12bcf43b9b401d4613445d469459276171a5b1b9..3de290c89c7c26a6b06517611fc228c068d159c5 100644 (file)
--- a/selftest/tests/__init__.py
+++ b/selftest/tests/__init__.py
@@ -19,6 +19,8 @@
 
 """Tests for selftest."""
 
+from testtools import TestCase
+
 import unittest
 
 def test_suite():

diff --git a/selftest/tests/test_socket_wrapper.py b/selftest/tests/test_socket_wrapper.py
index 1ebdf94de1797ca40d0d6f2c2650140f1966d2a7..81fe5b603cf8b4a0e68622b112d66b7706336565 100644 (file)
--- a/selftest/tests/test_socket_wrapper.py
+++ b/selftest/tests/test_socket_wrapper.py
@@ -19,12 +19,13 @@
 
 """Tests for selftest/socket_wrapper."""
 
+from selftest.tests import TestCase
+
 from selftest import socket_wrapper
 
 import os
-import unittest
 
-class SocketWrapperTests(unittest.TestCase):
+class SocketWrapperTests(TestCase):
 
     def test_setup_pcap(self):
         socket_wrapper.setup_pcap("somefile")

diff --git a/selftest/tests/test_target.py b/selftest/tests/test_target.py
index 139ed741d0e0b565dfd636d4f68cafb382ca7d91..4cdc02884b77ffb441f57dce1c937b39d9d38ddc 100644 (file)
--- a/selftest/tests/test_target.py
+++ b/selftest/tests/test_target.py
@@ -29,7 +29,7 @@ from selftest.target import (
     UnsupportedEnvironment,
     )
 
-import unittest
+from selftest.tests import TestCase
 
 
 class DummyEnvironment(Environment):
@@ -53,7 +53,7 @@ class DummyTarget(Target):
         return DummyEnvironment(name, prefix)
 
 
-class NoneEnvironmentTests(unittest.TestCase):
+class NoneEnvironmentTests(TestCase):
 
     def setUp(self):
         super(NoneEnvironmentTests, self).setUp()
@@ -69,7 +69,7 @@ class NoneEnvironmentTests(unittest.TestCase):
         self.assertEquals("", self.env.get_log())
 
 
-class NoneTargetTests(unittest.TestCase):
+class NoneTargetTests(TestCase):
 
     def setUp(self):
         super(NoneTargetTests, self).setUp()
@@ -80,7 +80,7 @@ class NoneTargetTests(unittest.TestCase):
             "something", "prefx")
 
 
-class EnvironmentManagerTests(unittest.TestCase):
+class EnvironmentManagerTests(TestCase):
 
     def setUp(self):
         super(EnvironmentManagerTests, self).setUp()

diff --git a/selftest/tests/test_testlist.py b/selftest/tests/test_testlist.py
index 4405e86003143132c110444d586b7906564e846f..cd44d46238159cd696d2f1314b352c922e9c9499 100644 (file)
--- a/selftest/tests/test_testlist.py
+++ b/selftest/tests/test_testlist.py
@@ -19,6 +19,8 @@
 
 """Tests for selftest.testlist."""
 
+from selftest.tests import TestCase
+
 from selftest.testlist import (
     RestrictedTestManager,
     find_in_list,
@@ -28,10 +30,8 @@ from selftest.testlist import (
 
 from cStringIO import StringIO
 
-import unittest
-
 
-class FindInListTests(unittest.TestCase):
+class FindInListTests(TestCase):
 
     def test_empty(self):
         self.assertIs(None, find_in_list([], "foo.test"))
@@ -41,7 +41,7 @@ class FindInListTests(unittest.TestCase):
             find_in_list([("foo.*bar", "because")], "foo.bla.bar"))
 
 
-class ReadTestRegexesTests(unittest.TestCase):
+class ReadTestRegexesTests(TestCase):
 
     def test_comment(self):
         f = StringIO("# I am a comment\n # I am also a comment\n")
@@ -57,7 +57,7 @@ class ReadTestRegexesTests(unittest.TestCase):
             list(read_test_regexes(f)))
 
 
-class ReadTestlistTests(unittest.TestCase):
+class ReadTestlistTests(TestCase):
 
     def test_read_list(self):
         inf = StringIO("-- TEST --\nfoo\nbar\nbla\n")
@@ -75,7 +75,7 @@ class ReadTestlistTests(unittest.TestCase):
 
 
 
-class RestrictedTestManagerTests(unittest.TestCase):
+class RestrictedTestManagerTests(TestCase):
 
     def test_unused(self):
         mgr = RestrictedTestManager(["foo.bar"])