mirror of
https://github.com/samba-team/samba.git
synced 2025-01-12 09:18:10 +03:00
c247afbda0
TestCase.assertEquals() is an alias for TestCase.assertEqual() and has been deprecated since Python 2.7. When we run our tests with in python developer mode (`PYTHONDEVMODE=1 make test`) we get 580 DeprecationWarnings about this. Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz> Reviewed-by: Noel Power <npower@samba.org>
137 lines
5.7 KiB
Python
137 lines
5.7 KiB
Python
# Unix SMB/CIFS implementation.
|
|
# Copyright (C) Catalyst.Net LTD 2015
|
|
# Copyright (C) Sean Dague <sdague@linux.vnet.ibm.com> 2011
|
|
#
|
|
# Catalyst.Net's contribution was written by Douglas Bagnall
|
|
# <douglas.bagnall@catalyst.net.nz>.
|
|
#
|
|
# 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/>.
|
|
#
|
|
|
|
import os
|
|
import ldb
|
|
from samba.tests.samba_tool.base import SambaToolCmdTest
|
|
from samba import sites
|
|
|
|
|
|
class BaseSitesCmdTestCase(SambaToolCmdTest):
|
|
"""Tests for samba-tool sites subnets"""
|
|
def setUp(self):
|
|
super(BaseSitesCmdTestCase, self).setUp()
|
|
self.dburl = "ldap://%s" % os.environ["DC_SERVER"]
|
|
self.creds_string = "-U%s%%%s" % (os.environ["DC_USERNAME"],
|
|
os.environ["DC_PASSWORD"])
|
|
|
|
self.samdb = self.getSamDB("-H", self.dburl, self.creds_string)
|
|
self.config_dn = str(self.samdb.get_config_basedn())
|
|
|
|
|
|
class SitesCmdTestCase(BaseSitesCmdTestCase):
|
|
|
|
def test_site_create(self):
|
|
sitename = 'new_site'
|
|
|
|
result, out, err = self.runsubcmd("sites", "create", sitename,
|
|
"-H", self.dburl, self.creds_string)
|
|
self.assertCmdSuccess(result, out, err)
|
|
|
|
dnsites = ldb.Dn(self.samdb, "CN=Sites,%s" % self.config_dn)
|
|
dnsite = ldb.Dn(self.samdb, "CN=%s,%s" % (sitename, dnsites))
|
|
|
|
ret = self.samdb.search(base=dnsites, scope=ldb.SCOPE_ONELEVEL,
|
|
expression='(cn=%s)' % sitename)
|
|
self.assertEqual(len(ret), 1)
|
|
|
|
# now delete it
|
|
self.samdb.delete(dnsite, ["tree_delete:0"])
|
|
|
|
|
|
class SitesSubnetCmdTestCase(BaseSitesCmdTestCase):
|
|
def setUp(self):
|
|
super(SitesSubnetCmdTestCase, self).setUp()
|
|
self.sitename = "testsite"
|
|
self.sitename2 = "testsite2"
|
|
self.samdb.transaction_start()
|
|
sites.create_site(self.samdb, self.config_dn, self.sitename)
|
|
sites.create_site(self.samdb, self.config_dn, self.sitename2)
|
|
self.samdb.transaction_commit()
|
|
|
|
def tearDown(self):
|
|
self.samdb.transaction_start()
|
|
sites.delete_site(self.samdb, self.config_dn, self.sitename)
|
|
sites.delete_site(self.samdb, self.config_dn, self.sitename2)
|
|
self.samdb.transaction_commit()
|
|
super(SitesSubnetCmdTestCase, self).tearDown()
|
|
|
|
def test_site_subnet_create(self):
|
|
cidrs = (("10.9.8.0/24", self.sitename),
|
|
("50.60.0.0/16", self.sitename2),
|
|
("50.61.0.0/16", self.sitename2), # second subnet on the site
|
|
("50.0.0.0/8", self.sitename), # overlapping subnet, other site
|
|
("50.62.1.2/32", self.sitename), # single IP
|
|
("aaaa:bbbb:cccc:dddd:eeee:ffff:2222:1100/120",
|
|
self.sitename2),
|
|
)
|
|
|
|
for cidr, sitename in cidrs:
|
|
result, out, err = self.runsubcmd("sites", "subnet", "create",
|
|
cidr, sitename,
|
|
"-H", self.dburl,
|
|
self.creds_string)
|
|
self.assertCmdSuccess(result, out, err)
|
|
|
|
ret = self.samdb.search(base=self.config_dn,
|
|
scope=ldb.SCOPE_SUBTREE,
|
|
expression=('(&(objectclass=subnet)(cn=%s))'
|
|
% cidr))
|
|
self.assertIsNotNone(ret)
|
|
self.assertEqual(len(ret), 1)
|
|
|
|
dnsubnets = ldb.Dn(self.samdb,
|
|
"CN=Subnets,CN=Sites,%s" % self.config_dn)
|
|
|
|
for cidr, sitename in cidrs:
|
|
dnsubnet = ldb.Dn(self.samdb, ("Cn=%s,CN=Subnets,CN=Sites,%s" %
|
|
(cidr, self.config_dn)))
|
|
|
|
ret = self.samdb.search(base=dnsubnets, scope=ldb.SCOPE_ONELEVEL,
|
|
expression='(CN=%s)' % cidr)
|
|
self.assertIsNotNone(ret)
|
|
self.assertEqual(len(ret), 1)
|
|
self.samdb.delete(dnsubnet, ["tree_delete:0"])
|
|
|
|
def test_site_subnet_create_should_fail(self):
|
|
cidrs = (("10.9.8.0/33", self.sitename), # mask too big
|
|
("50.60.0.0/8", self.sitename2), # insufficient zeros
|
|
("50.261.0.0/16", self.sitename2), # bad octet
|
|
("7.0.0.0.0/0", self.sitename), # insufficient zeros
|
|
("aaaa:bbbb:cccc:dddd:eeee:ffff:2222:1100/119",
|
|
self.sitename), # insufficient zeros
|
|
)
|
|
|
|
for cidr, sitename in cidrs:
|
|
result, out, err = self.runsubcmd("sites", "subnet", "create",
|
|
cidr, sitename,
|
|
"-H", self.dburl,
|
|
self.creds_string)
|
|
self.assertCmdFail(result)
|
|
|
|
ret = self.samdb.search(base=self.config_dn,
|
|
scope=ldb.SCOPE_SUBTREE,
|
|
expression=('(&(objectclass=subnet)(cn=%s))'
|
|
% cidr))
|
|
|
|
self.assertIsNotNone(ret)
|
|
self.assertEqual(len(ret), 0)
|