selftest.py: Use testtools TestCase class.

[ira/wip.git] / selftest / tests / test_socket_wrapper.py

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")