source4/scripting/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, "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" %
  38 os.environ["SERVER"])
  39         self.assertEquals(result, -1, "check for result code")
  40
  41     def test_fetch(self):
  42         """Run against a real GPO, and make sure it passes"""
  43         (result, out, err) = self.runsubcmd("gpo", "fetch", self.gpo_guid, "-H", "ldap://%s" % os.environ["SERVER"], "--tmpdir", self.tempdir)
  44         self.assertCmdSuccess(result, "Ensuring gpo fetched successfully")
  45         shutil.rmtree(os.path.join(self.tempdir, "policy"))
  46
  47     def setUp(self):
  48         """set up a temporary GPO to work with"""
  49         super(GpoCmdTestCase, self).setUp()
  50         (result, out, err) = self.runsubcmd("gpo", "create", self.gpo_name,
  51                                             "-H", "ldap://%s" % os.environ["SERVER"],
  52                                             "-U%s%%%s" % (os.environ["USERNAME"], os.environ["PASSWORD"]),
  53                                             "--tmpdir", self.tempdir)
  54         shutil.rmtree(os.path.join(self.tempdir, "policy"))
  55         self.assertCmdSuccess(result, "Ensuring gpo created successfully")
  56         try:
  57             self.gpo_guid = "{%s}" % out.split("{")[1].split("}")[0]
  58         except IndexError:
  59             self.fail("Failed to find GUID in output: %s" % out)
  60
  61     def tearDown(self):
  62         """remove the temporary GPO to work with"""
  63         (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"]))
  64         self.assertCmdSuccess(result, "Ensuring gpo deleted successfully")
  65         super(GpoCmdTestCase, self).tearDown()