From 2db221c3845e2b1c44f8237c550704dca41df91a Mon Sep 17 00:00:00 2001 From: Matthieu Patou Date: Wed, 2 Nov 2011 20:24:20 +0100 Subject: s4-sites: Document, fix under optimal coding, use exceptions Autobuild-User: Matthieu Patou Autobuild-Date: Wed Nov 2 22:35:00 CET 2011 on sn-devel-104 --- source4/dsdb/tests/python/sites.py | 36 ++++++++++++++++++------------------ 1 file changed, 18 insertions(+), 18 deletions(-) (limited to 'source4/dsdb/tests/python') diff --git a/source4/dsdb/tests/python/sites.py b/source4/dsdb/tests/python/sites.py index d3f5c5769f..8b984b2fe8 100644 --- a/source4/dsdb/tests/python/sites.py +++ b/source4/dsdb/tests/python/sites.py @@ -80,7 +80,6 @@ class SitesBaseTests(samba.tests.TestCase): #tests on sites class SimpleSitesTests(SitesBaseTests): - def test_create(self): """test creation of 1 site""" @@ -88,30 +87,31 @@ class SimpleSitesTests(SitesBaseTests): ok = sites.create_site(self.ldb_admin, self.ldb_admin.get_config_basedn(), "testsamba") self.ldb_admin.transaction_commit() - self.assertTrue(ok) - ok = False - try: - ok = sites.create_site(self.ldb_admin, self.ldb_admin.get_config_basedn(), - "testsamba") - self.assertFalse(ok) - except: - self.assertFalse(ok) + + self.assertRaises(sites.SiteAlreadyExistsException, + sites.create_site, self.ldb_admin, self.ldb_admin.get_config_basedn(), + "testsamba") def test_delete(self): - """test creation of 1 site""" + """test removal of 1 site""" self.ldb_admin.transaction_start() ok = sites.delete_site(self.ldb_admin, self.ldb_admin.get_config_basedn(), "testsamba") + self.ldb_admin.transaction_commit() - self.assertTrue(ok) - ok = False - try: - ok = sites.delete_site(self.ldb_admin, self.ldb_admin.get_config_basedn(), - "testsamba") - self.assertFalse(ok) - except: - self.assertFalse(ok) + + self.assertRaises(sites.SiteNotFoundException, + sites.delete_site, self.ldb_admin, self.ldb_admin.get_config_basedn(), + "testsamba") + + + def test_delete_not_empty(self): + """test removal of 1 site with servers""" + + self.assertRaises(sites.SiteServerNotEmptyException, + sites.delete_site, self.ldb_admin, self.ldb_admin.get_config_basedn(), + "Default-First-Site-Name") ldb = SamDB(ldapshost, credentials=creds, session_info=system_session(lp), lp=lp) -- cgit