sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-08-03 04:22:09 +03:00
Code Activity

python/dcerpc: Custom implementations of policy_handle.__init__ and policy_handle.__repr__

Browse Source 
pair-programmed with Jelmer
This commit is contained in:
Andrew Bartlett
2009-04-21 11:53:00 +02:00
parent 1856b999c8
commit 6f60a6e71a
2 changed files with 49 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
source4/librpc/ndr/py_misc.c
View File

@ -90,3 +90,41 @@ static void py_GUID_patch(PyTypeObject *type)
#define PY_GUID_PATCH py_GUID_patch
static int py_policy_handle_init(PyObject *self, PyObject *args, PyObject *kwargs)
{
char *str = NULL;
NTSTATUS status;
struct policy_handle *handle = py_talloc_get_ptr(self);
const char *kwnames[] = { "uuid", "type", NULL };
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|si", discard_const_p(char *, kwnames), &str, &handle->handle_type))
return -1;
if (str != NULL) {
status = GUID_from_string(str, &handle->uuid);
if (!NT_STATUS_IS_OK(status)) {
PyErr_SetNTSTATUS(status);
return -1;
}
}
return 0;
}
static PyObject *py_policy_handle_repr(PyObject *py_self)
{
struct policy_handle *self = py_talloc_get_ptr(py_self);
char *uuid_str = GUID_string(NULL, &self->uuid);
PyObject *ret = PyString_FromFormat("policy_handle('%s', %d)", uuid_str, self->handle_type);
talloc_free(uuid_str);
return ret;
}
static void py_policy_handle_patch(PyTypeObject *type)
{
type->tp_init = py_policy_handle_init;
type->tp_repr = py_policy_handle_repr;
}
#define PY_POLICY_HANDLE_PATCH py_policy_handle_patch

11
source4/scripting/python/samba/tests/dcerpc/misc.py
View File

@ -46,3 +46,14 @@ class GUIDTests(unittest.TestCase):
class PolicyHandleTests(unittest.TestCase):
def test_init(self):
x = misc.policy_handle(text1, 1)
self.assertEquals(1, x.handle_type)
self.assertEquals(text1, str(x.uuid))
def test_repr(self):
x = misc.policy_handle(text1, 42)
self.assertEquals("policy_handle('%s', %d)" % (text1, 42), repr(x))