1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-24 21:34:56 +03:00
samba-mirror/python/samba/tests/samba_upgradedns_lmdb.py
Gary Lockyer f0cebbe4dd test upgradedns: ensure lmdb lock files linked
Add tests to check that the '-lock' files for the dns partitions as well as
the data files are linked when running
    samba_dnsupgrade --dns-backend=BIND9_DLZ
failure to create these links can cause corruption of the corresponding
data file.

BUG: https://bugzilla.samba.org/show_bug.cgi?id=14199

Signed-off-by: Gary Lockyer <gary@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
2019-12-20 07:35:41 +00:00

76 lines
2.6 KiB
Python

# Unix SMB/CIFS implementation.
# Copyright (C) Catalyst IT Ltd. 2019
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
from samba.tests.samba_tool.base import SambaToolCmdTest
import os
import shutil
class UpgradeDnsLmdbTestCase(SambaToolCmdTest):
"""
Tests for dns upgrade on a lmdb backend
"""
def setUp(self):
super(UpgradeDnsLmdbTestCase, self).setUp()
self.tempsambadir = os.path.join(self.tempdir, "samba")
os.mkdir(self.tempsambadir)
# provision a domain
#
# returns the tuple (ret, stdout, stderr)
def provision(self):
command = (
"samba-tool "
"domain provision "
"--realm=foo.example.com "
"--domain=FOO "
"--targetdir=%s "
"--backend-store=mdb "
"--use-ntvfs " % self.tempsambadir)
return self.run_command(command)
# upgrade a domains dns to BIND9
#
# returns the tuple (ret, stdout, stderr)
def upgrade_dns(self):
command = (
"samba_upgradedns "
"--dns-backend=BIND9_DLZ "
"--configfile %s/etc/smb.conf" % self.tempsambadir)
return self.run_command(command)
def tearDown(self):
super(UpgradeDnsLmdbTestCase, self).tearDown()
shutil.rmtree(self.tempsambadir)
def test_lmdb_lock_files_linked_on_upgrade_to_bind9_dlz(self):
"""
Ensure that links are created for the lock files as well as the
data files
"""
self.provision()
self.upgrade_dns()
directory = ("%s/bind-dns/dns/sam.ldb.d" % self.tempsambadir)
for filename in os.listdir(directory):
if filename.endswith(".ldb") and "DNSZONES" in filename:
lock_file = ("%s/%s-lock" % (directory, filename))
self.assertTrue(
os.path.isfile(lock_file),
msg=("Lock file %s/%s-lock for %s, does not exist" %
(directory, filename, filename)))