1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-24 21:34:56 +03:00
samba-mirror/python/samba/tests/dcerpc/testrpc.py
Douglas Bagnall aecb2b779b python: remove all 'from __future__ import print_function'
This made Python 2's print behave like Python 3's print().

In some cases, where we had:

   from __future__ import print_function
   """Intended module documentation..."""

this will have the side effect of making the intended module documentation
work as the actual module documentation (i.e. becoming __doc__), because
it is once again the first statement in the module.

Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
2021-04-28 03:43:34 +00:00

144 lines
5.2 KiB
Python

# test generated python code from pidl
# Copyright (C) Andrew Tridgell August 2010
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
import sys
sys.path.insert(0, "bin/python")
import samba
import samba.tests
from samba.dcerpc import drsuapi
import talloc
talloc.enable_null_tracking()
class RpcTests(object):
'''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 = talloc.total_blocks(object)
if object is None:
nblocks -= self.initial_blocks
leaked_blocks = (nblocks - num_expected)
if leaked_blocks != 0:
print("Leaked %d blocks" % leaked_blocks)
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 as 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))
self.errcount += 1
continue
except Exception:
print("ERROR: Failed to instantiate %s.%s" % (typename, n))
self.errcount += 1
continue
# now try setting the value back
try:
print("Setting %s.%s" % (typename, n))
setattr(v, n, value)
except Exception as e:
if isinstance(e, AttributeError) and str(e).endswith("is read-only"):
# readonly, ignore
continue
else:
print("ERROR: Failed to set %s.%s: %r: %s" % (typename, n, e.__class__, e))
self.errcount += 1
continue
# and try a comparison
try:
if value != getattr(v, n):
print("ERROR: Comparison failed for %s.%s: %r != %r" % (typename, n, value, getattr(v, n)))
continue
except Exception as e:
print("ERROR: compare exception for %s.%s: %r: %s" % (typename, n, e.__class__, e))
continue
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 = talloc.total_blocks(None)
self.check_type(interface, n, value)
self.check_blocks(None, initial_blocks)
except Exception as e:
print("ERROR: Failed to check_type %s.%s: %r: %s" % (iname, n, e.__class__, e))
self.errcount += 1
elif callable(value):
pass # Method
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 = talloc.total_blocks(None)
self.check_interface(iface, iname)
self.check_blocks(None, initial_blocks)
def run(self):
self.initial_blocks = talloc.total_blocks(None)
self.errcount = 0
self.check_all_interfaces()
return self.errcount
tests = RpcTests()
errcount = tests.run()
if errcount == 0:
sys.exit(0)
else:
print("%d failures" % errcount)
sys.exit(1)