tests:samba_tool: pass stdout and stderr to assertCmdSuccess()
[nivanova/samba-autobuild/.git] / python / samba / tests / samba_tool / gpo.py
1 # Unix SMB/CIFS implementation.
2 # Copyright (C) Andrew Bartlett 2012
3 #
4 # based on time.py:
5 # Copyright (C) Sean Dague <sdague@linux.vnet.ibm.com> 2011
6 #
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.
11 #
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.
16 #
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/>.
19 #
20
21 import os
22 from samba.tests.samba_tool.base import SambaToolCmdTest
23 import shutil
24
25 class GpoCmdTestCase(SambaToolCmdTest):
26     """Tests for samba-tool time subcommands"""
27
28     gpo_name = "testgpo"
29
30     def test_gpo_list(self):
31         """Run gpo list against the server and make sure it looks accurate"""
32         (result, out, err) = self.runsubcmd("gpo", "listall", "-H", "ldap://%s" % os.environ["SERVER"])
33         self.assertCmdSuccess(result, out, err, "Ensuring gpo listall ran successfully")
34
35     def test_fetchfail(self):
36         """Run against a non-existent GPO, and make sure it fails (this hard-coded UUID is very unlikely to exist"""
37         (result, out, err) = self.runsubcmd("gpo", "fetch", "c25cac17-a02a-4151-835d-fae17446ee43", "-H", "ldap://%s" % os.environ["SERVER"])
38         self.assertEquals(result, -1, "check for result code")
39
40     def test_fetch(self):
41         """Run against a real GPO, and make sure it passes"""
42         (result, out, err) = self.runsubcmd("gpo", "fetch", self.gpo_guid, "-H", "ldap://%s" % os.environ["SERVER"], "--tmpdir", self.tempdir)
43         self.assertCmdSuccess(result, out, err, "Ensuring gpo fetched successfully")
44         shutil.rmtree(os.path.join(self.tempdir, "policy"))
45
46     def test_show(self):
47         """Show a real GPO, and make sure it passes"""
48         (result, out, err) = self.runsubcmd("gpo", "show", self.gpo_guid, "-H", "ldap://%s" % os.environ["SERVER"])
49         self.assertCmdSuccess(result, out, err, "Ensuring gpo fetched successfully")
50
51     def test_show_as_admin(self):
52         """Show a real GPO, and make sure it passes"""
53         (result, out, err) = self.runsubcmd("gpo", "show", self.gpo_guid, "-H", "ldap://%s" % os.environ["SERVER"], "-U%s%%%s" % (os.environ["USERNAME"], os.environ["PASSWORD"]))
54         self.assertCmdSuccess(result, out, err, "Ensuring gpo fetched successfully")
55
56     def test_aclcheck(self):
57         """Check all the GPOs on the remote server have correct ACLs"""
58         (result, out, err) = self.runsubcmd("gpo", "aclcheck", "-H", "ldap://%s" % os.environ["SERVER"], "-U%s%%%s" % (os.environ["USERNAME"], os.environ["PASSWORD"]))
59         self.assertCmdSuccess(result, out, err, "Ensuring gpo checked successfully")
60
61     def setUp(self):
62         """set up a temporary GPO to work with"""
63         super(GpoCmdTestCase, self).setUp()
64         (result, out, err) = self.runsubcmd("gpo", "create", self.gpo_name,
65                                             "-H", "ldap://%s" % os.environ["SERVER"],
66                                             "-U%s%%%s" % (os.environ["USERNAME"], os.environ["PASSWORD"]),
67                                             "--tmpdir", self.tempdir)
68         shutil.rmtree(os.path.join(self.tempdir, "policy"))
69         self.assertCmdSuccess(result, out, err, "Ensuring gpo created successfully")
70         try:
71             self.gpo_guid = "{%s}" % out.split("{")[1].split("}")[0]
72         except IndexError:
73             self.fail("Failed to find GUID in output: %s" % out)
74
75     def tearDown(self):
76         """remove the temporary GPO to work with"""
77         (result, out, err) = self.runsubcmd("gpo", "del", self.gpo_guid, "-H", "ldap://%s" % os.environ["SERVER"], "-U%s%%%s" % (os.environ["USERNAME"], os.environ["PASSWORD"]))
78         self.assertCmdSuccess(result, out, err, "Ensuring gpo deleted successfully")
79         super(GpoCmdTestCase, self).tearDown()