make perftest: for performance testing

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

diff --git a/selftest/tests/test_samba.py b/selftest/tests/test_samba.py
index f06d846debf6f9b30120bd9375bcefb24cebbdd6..4bcfcba9e13e5441632e874d20eee5055a54ec1c 100644 (file)
--- a/selftest/tests/test_samba.py
+++ b/selftest/tests/test_samba.py
@@ -19,16 +19,13 @@
 
 """Tests for selftest.target.samba."""
 
-import os
-import sys
-
 from cStringIO import StringIO
 
-from selftest.tests import TestCase
+from samba.tests import TestCase
 
 from selftest.target.samba import (
     bindir_path,
-    cleanup_child,
+    get_interface,
     mk_realms_stanza,
     write_krb5_conf,
     )
@@ -37,15 +34,15 @@ from selftest.target.samba import (
 class BinDirPathTests(TestCase):
 
     def test_mapping(self):
-        self.assertEquals("exe4",
-            bindir_path({"exe": "exe4"}, "/some/path", "exe"))
-        self.assertEquals("/bin/ls",
-            bindir_path({"exe": "ls"}, "/bin", "exe"))
+        self.assertEquals("exe",
+            bindir_path("/some/path", "exe"))
+        self.assertEquals("/bin/exe",
+            bindir_path("/bin", "/bin/exe"))
 
     def test_no_mapping(self):
-        self.assertEqual("exe", bindir_path({}, "/some/path", "exe"))
+        self.assertEqual("exe", bindir_path("/some/path", "exe"))
         self.assertEqual("/bin/ls",
-            bindir_path({}, "/bin", "ls"))
+            bindir_path("/bin", "ls"))
 
 
 class MkRealmsStanzaTests(TestCase):
@@ -107,3 +104,13 @@ class WriteKrb5ConfTests(TestCase):
  }
 
 ''', f.getvalue())
+
+
+class GetInterfaceTests(TestCase):
+
+    def test_get_interface(self):
+        self.assertEquals(21, get_interface("localdc"))
+        self.assertEquals(4, get_interface("localshare4"))
+
+    def test_unknown(self):
+        self.assertRaises(KeyError, get_interface, "unknown")