s4-tests: Modified speedtest.py to use samba.tests.delete_force
authorNadezhda Ivanova <nivanova@samba.org>
Wed, 24 Nov 2010 23:16:14 +0000 (01:16 +0200)
committerNadezhda Ivanova <nivanova@samba.org>
Thu, 25 Nov 2010 00:28:19 +0000 (01:28 +0100)
Autobuild-User: Nadezhda Ivanova <nivanova@samba.org>
Autobuild-Date: Thu Nov 25 01:28:19 CET 2010 on sn-devel-104

source4/scripting/devel/speedtest.py

index d5d87afb77b16c80849601a06e291dd208928b2a..891a741b1c7a3a860805b160157f090eafe523b2 100755 (executable)
@@ -46,6 +46,7 @@ from samba import gensec
 from samba.samdb import SamDB
 from samba.credentials import Credentials
 import samba.tests
+from samba.tests import delete_force
 from subunit.run import SubunitTestRunner
 import unittest
 
@@ -76,12 +77,6 @@ creds.set_gensec_features(creds.get_gensec_features() | gensec.FEATURE_SEAL)
 
 class SpeedTest(samba.tests.TestCase):
 
-    def delete_force(self, ldb, dn):
-        try:
-            ldb.delete(dn)
-        except LdbError, (num, _):
-            self.assertEquals(num, ERR_NO_SUCH_OBJECT)
-
     def find_basedn(self, ldb):
         res = ldb.search(base="", expression="", scope=SCOPE_BASE,
                          attrs=["defaultNamingContext"])
@@ -126,13 +121,13 @@ url: www.example.com
 
     def remove_bundle(self, count):
         for i in range(count):
-            self.delete_force(self.ldb_admin, "cn=speedtestuser%d,cn=Users,%s" % (i+1, self.base_dn))
+            delete_force(self.ldb_admin, "cn=speedtestuser%d,cn=Users,%s" % (i+1, self.base_dn))
 
     def remove_test_users(self):
         res = ldb.search(base="cn=Users,%s" % self.base_dn, expression="(objectClass=user)", scope=SCOPE_SUBTREE)
         dn_list = [item.dn for item in res if "speedtestuser" in str(item.dn)]
         for dn in dn_list:
-            self.delete_force(self.ldb_admin, dn)
+            delete_force(self.ldb_admin, dn)
 
     def run_bundle(self, num):
         print "\n=== Test ADD/DEL %s user objects ===\n" % num