2 # -*- coding: utf-8 -*-
4 # Unix SMB/CIFS implementation.
5 # Copyright © Jelmer Vernooij <jelmer@samba.org> 2008
7 # This program is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 3 of the License, or
10 # (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
21 from samba
.dcerpc
import samr
, security
22 from samba
.tests
import RpcInterfaceTestCase
24 # FIXME: Pidl should be doing this for us
25 def toArray((handle
, array
, num_entries
)):
27 for x
in range(num_entries
):
28 ret
.append((array
.entries
[x
].idx
, array
.entries
[x
].name
))
32 class SamrTests(RpcInterfaceTestCase
):
34 self
.conn
= samr
.samr("ncalrpc:", self
.get_loadparm())
36 def test_connect5(self
):
37 (level
, info
, handle
) = self
.conn
.Connect5(None, 0, 1, samr
.ConnectInfo1())
39 def test_connect2(self
):
40 handle
= self
.conn
.Connect2(None, security
.SEC_FLAG_MAXIMUM_ALLOWED
)
42 def test_EnumDomains(self
):
43 handle
= self
.conn
.Connect2(None, security
.SEC_FLAG_MAXIMUM_ALLOWED
)
44 domains
= toArray(self
.conn
.EnumDomains(handle
, 0, -1))
45 self
.conn
.Close(handle
)