diff options
-rw-r--r-- | source4/scripting/swig/dcerpc.i | 9 | ||||
-rwxr-xr-x | source4/scripting/swig/torture/samr.py | 8 |
2 files changed, 15 insertions, 2 deletions
diff --git a/source4/scripting/swig/dcerpc.i b/source4/scripting/swig/dcerpc.i index a90ac90b5a..bded875b10 100644 --- a/source4/scripting/swig/dcerpc.i +++ b/source4/scripting/swig/dcerpc.i @@ -1,3 +1,5 @@ +/* Tastes like -*- C -*- */ + /* Unix SMB/CIFS implementation. @@ -227,11 +229,16 @@ PyObject *string_ptr_to_python(TALLOC_CTX *mem_ctx, char *obj) %include "samba.i" +%pythoncode %{ + NTSTATUS = _dcerpc.NTSTATUS +%} + %init %{ setup_logging("python", DEBUG_STDOUT); lp_load(dyn_CONFIGFILE, True, False, False); load_interfaces(); - ntstatus_exception = PyErr_NewException("dcerpc.NTSTATUS", NULL, NULL); + ntstatus_exception = PyErr_NewException("_dcerpc.NTSTATUS", NULL, NULL); + PyDict_SetItemString(d, "NTSTATUS", ntstatus_exception); %} %typemap(in, numinputs=0) struct dcerpc_pipe **OUT (struct dcerpc_pipe *temp_dcerpc_pipe) { diff --git a/source4/scripting/swig/torture/samr.py b/source4/scripting/swig/torture/samr.py index 00e4db1958..1f3b60eed4 100755 --- a/source4/scripting/swig/torture/samr.py +++ b/source4/scripting/swig/torture/samr.py @@ -90,7 +90,13 @@ def test_LookupDomain(pipe, handle, domain): result = dcerpc.samr_LookupDomain(pipe, r) - print result + r['domain']['name'] = 'xxNODOMAINxx' + + try: + result = dcerpc.samr_LookupDomain(pipe, r) + except dcerpc.NTSTATUS, arg: + if arg[0] != 0xc00000df: + raise dcerpc.NTSTATUS(arg) def test_EnumDomains(pipe, handle): |