selftest.py: Add get_interface.

Autobuild-User: Jelmer Vernooij <jelmer@samba.org>
Autobuild-Date: Tue Mar 27 00:32:48 CEST 2012 on sn-devel-104

2 files changed, 45 insertions, 4 deletions

diff --git a/selftest/target/samba.py b/selftest/target/samba.py
index 1ea156c003..666d223ff1 100644
--- a/selftest/target/samba.py
+++ b/selftest/target/samba.py
@@ -119,3 +119,37 @@ def cleanup_child(pid, name, outf=None):
     else:
         outf.write("%s child process %d exited with value %d.\n" % (name, childpid, status >> 8))
     return childpid
+
+
+def get_interface(netbiosname):
+    """Return interface id for a particular server.
+    """
+    netbiosname = netbiosname.lower()
+
+    interfaces = {
+        "locals3dc2": 2,
+        "localmember3": 3,
+        "localshare4": 4,
+        "localserver5": 5,
+        "localktest6": 6,
+        "maptoguest": 7,
+
+        # 11-16 used by selftest.pl for client interfaces
+        "localdc": 21,
+        "localvampiredc": 22,
+        "s4member": 23,
+        "localrpcproxy": 24,
+        "dc5": 25,
+        "dc6": 26,
+        "dc7": 27,
+        "rodc": 28,
+        "localadmember": 29,
+        "plugindc": 30,
+        "localsubdc": 31,
+        "chgdcpass": 32,
+    }
+
+    # update lib/socket_wrapper/socket_wrapper.c
+    #  #define MAX_WRAPPED_INTERFACES 32
+    # if you wish to have more than 32 interfaces
+    return interfaces[netbiosname]
diff --git a/selftest/tests/test_samba.py b/selftest/tests/test_samba.py
index f06d846deb..b49463eabb 100644
--- 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 selftest.target.samba import (
     bindir_path,
-    cleanup_child,
+    get_interface,
     mk_realms_stanza,
     write_krb5_conf,
     )
@@ -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")