Use samba TestCase so we get all compatibility functions on Python < 2.7.

[sfrench/samba-autobuild/.git] / selftest / tests / test_socket_wrapper.py

diff --git a/selftest/tests/test_socket_wrapper.py b/selftest/tests/test_socket_wrapper.py
index 5862f1edf8a910fb58623c48805084446d166007..e278fdddfa6fabeb5abf7b7828ee4a2d1d717089 100644 (file)
--- a/selftest/tests/test_socket_wrapper.py
+++ b/selftest/tests/test_socket_wrapper.py
@@ -19,7 +19,7 @@
 
 """Tests for selftest/socket_wrapper."""
 
-from unittest import TestCase
+from samba.tests import TestCase
 
 from selftest import socket_wrapper