"""Tests for selftest."""
+from testtools import TestCase
+
import unittest
def test_suite():
"""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")
UnsupportedEnvironment,
)
-import unittest
+from selftest.tests import TestCase
class DummyEnvironment(Environment):
return DummyEnvironment(name, prefix)
-class NoneEnvironmentTests(unittest.TestCase):
+class NoneEnvironmentTests(TestCase):
def setUp(self):
super(NoneEnvironmentTests, self).setUp()
self.assertEquals("", self.env.get_log())
-class NoneTargetTests(unittest.TestCase):
+class NoneTargetTests(TestCase):
def setUp(self):
super(NoneTargetTests, self).setUp()
"something", "prefx")
-class EnvironmentManagerTests(unittest.TestCase):
+class EnvironmentManagerTests(TestCase):
def setUp(self):
super(EnvironmentManagerTests, self).setUp()
"""Tests for selftest.testlist."""
+from selftest.tests import TestCase
+
from selftest.testlist import (
RestrictedTestManager,
find_in_list,
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"))
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")
list(read_test_regexes(f)))
-class ReadTestlistTests(unittest.TestCase):
+class ReadTestlistTests(TestCase):
def test_read_list(self):
inf = StringIO("-- TEST --\nfoo\nbar\nbla\n")
-class RestrictedTestManagerTests(unittest.TestCase):
+class RestrictedTestManagerTests(TestCase):
def test_unused(self):
mgr = RestrictedTestManager(["foo.bar"])