summaryrefslogtreecommitdiff
path: root/source4/scripting/python
diff options
context:
space:
mode:
authorAndrew Tridgell <tridge@samba.org>2010-08-30 15:41:22 +1000
committerAndrew Tridgell <tridge@samba.org>2010-09-02 13:37:07 +1000
commit39599e949b9231fa5d88b1a6830b442c1faba1d9 (patch)
tree4b8e88d5aa7425773b002f600810fda491d683f2 /source4/scripting/python
parent71f2ebcae09a3581cb27966b44387a10ea89236d (diff)
downloadsamba-39599e949b9231fa5d88b1a6830b442c1faba1d9.tar.gz
samba-39599e949b9231fa5d88b1a6830b442c1faba1d9.tar.bz2
samba-39599e949b9231fa5d88b1a6830b442c1faba1d9.zip
s4-pidl: added a test for all generated rpc interfaces
this tries to instantiate all types in all generated python RPC interfaces, then checks that all attributes can be read, written and compared.
Diffstat (limited to 'source4/scripting/python')
-rw-r--r--source4/scripting/python/samba/tests/dcerpc/testrpc.py113
1 files changed, 113 insertions, 0 deletions
diff --git a/source4/scripting/python/samba/tests/dcerpc/testrpc.py b/source4/scripting/python/samba/tests/dcerpc/testrpc.py
new file mode 100644
index 0000000000..71c1341a43
--- /dev/null
+++ b/source4/scripting/python/samba/tests/dcerpc/testrpc.py
@@ -0,0 +1,113 @@
+#!/usr/bin/env python
+#
+# test generated python code from pidl
+# Andrew Tridgell August 2010
+#
+import sys
+
+sys.path.insert(0, "bin/python")
+
+import samba
+import samba.tests
+from samba.dcerpc import drsuapi
+
+samba.talloc_enable_null_tracking()
+
+class RpcTests(samba.tests.TestCase):
+ '''test type behaviour of pidl generated python RPC code'''
+
+ def check_blocks(self, object, num_expected):
+ '''check that the number of allocated blocks is correct'''
+ nblocks = samba.talloc_total_blocks(object)
+ if object is None:
+ nblocks -= self.initial_blocks
+ self.assertEquals(nblocks, num_expected)
+
+ def check_type(self, interface, typename, type):
+ print "Checking type %s" % typename
+ v = type()
+ for n in dir(v):
+ if n[0] == '_':
+ continue
+ try:
+ value = getattr(v, n)
+ except TypeError, errstr:
+ if str(errstr) == "unknown union level":
+ print "ERROR: Unknown union level in %s.%s" % (typename, n)
+ self.errcount += 1
+ continue
+ print str(errstr)[1:21]
+ if str(errstr)[0:21] == "Can not convert C Type":
+ print "ERROR: Unknown C type for %s.%s" % (typename, n)
+ self.errcount += 1
+ continue
+ else:
+ print "ERROR: Failed to instantiate %s.%s" % (typename, n)
+ raise
+ except:
+ print "ERROR: Failed to instantiate %s.%s" % (typename, n)
+ raise
+
+ # now try setting the value back
+ try:
+ print "Setting %s.%s" % (typename, n)
+ setattr(v, n, value)
+ except:
+ print "ERROR: Failed to set %s.%s" % (typename, n)
+ raise
+
+ # and try a comparison
+ try:
+ if value != getattr(v, n):
+ print "ERROR: Comparison failed for %s.%s" % (typename, n)
+ raise
+ except:
+ print "ERROR: compare exception for %s.%s" % (typename, n)
+ raise
+
+
+ def check_interface(self, interface, iname):
+ errcount = self.errcount
+ for n in dir(interface):
+ if n[0] == '_' or n == iname:
+ # skip the special ones
+ continue
+ value = getattr(interface, n)
+ if isinstance(value, str):
+ #print "%s=\"%s\"" % (n, value)
+ pass
+ elif isinstance(value, int):
+ #print "%s=%d" % (n, value)
+ pass
+ elif isinstance(value, type):
+ try:
+ initial_blocks = samba.talloc_total_blocks(None)
+ self.check_type(interface, n, value)
+ self.check_blocks(None, initial_blocks)
+ except:
+ print "ERROR: Failed to check_type %s.%s" % (iname, n)
+ self.errcount += 1
+ pass
+ else:
+ print "UNKNOWN: %s=%s" % (n, value)
+ if self.errcount - errcount != 0:
+ print "Found %d errors in %s" % (self.errcount - errcount, iname)
+
+
+ def check_all_interfaces(self):
+ for iname in dir(samba.dcerpc):
+ if iname[0] == '_':
+ continue
+ if iname == 'ClientConnection' or iname == 'base':
+ continue
+ print "Checking interface %s" % iname
+ iface = getattr(samba.dcerpc, iname)
+ initial_blocks = samba.talloc_total_blocks(None)
+ self.check_interface(iface, iname)
+ self.check_blocks(None, initial_blocks)
+
+ def test_run(self):
+ self.initial_blocks = samba.talloc_total_blocks(None)
+ self.errcount = 0
+ self.check_all_interfaces()
+ self.assertEquals(self.errcount, 0)