selftest: make _get_attribute use parse_ldif

Signed-off-by: Rob van der Linde <rob@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>

diff --git a/python/samba/tests/samba_tool/user_virtualCryptSHA_base.py b/python/samba/tests/samba_tool/user_virtualCryptSHA_base.py
index d5b72048d4abd37abdf3aea907f138dd6a65eda1..19cd713eacf22c3e774c89ea9836f8ef2165bca5 100644 (file)
--- a/python/samba/tests/samba_tool/user_virtualCryptSHA_base.py
+++ b/python/samba/tests/samba_tool/user_virtualCryptSHA_base.py
@@ -23,7 +23,6 @@ from samba.credentials import Credentials
 from samba.samdb import SamDB
 from samba.auth import system_session
 from samba import dsdb
-import re
 
 USER_NAME = "CryptSHATestUser"
 HASH_OPTION = "password hash userPassword schemes"
@@ -38,12 +37,10 @@ class UserCmdCryptShaTestCase(SambaToolCmdTest):
     samdb = None
 
     def _get_attribute(self, out, name):
-        p = re.compile("^" + name + r":\s+(\S+)")
-        for line in out.split("\n"):
-            m = p.match(line)
-            if m:
-                return m.group(1)
-        return ""
+        parsed = list(self.ldb.parse_ldif(out))
+        self.assertEqual(len(parsed), 1)
+        changetype, msg = parsed[0]
+        return str(msg.get(name, ""))
 
     def add_user(self, hashes=""):
         self.lp = samba.tests.env_loadparm()