summaryrefslogtreecommitdiff
path: root/selftest/tests/test_socket_wrapper.py
diff options
context:
space:
mode:
Diffstat (limited to 'selftest/tests/test_socket_wrapper.py')
-rw-r--r--selftest/tests/test_socket_wrapper.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/selftest/tests/test_socket_wrapper.py b/selftest/tests/test_socket_wrapper.py
index 1ebdf94de1..81fe5b603c 100644
--- 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")