Use created loadparm context, rather than the one specified on the command-line.
[amitay/samba.git] / source4 / scripting / python / samba / tests / samdb.py
1 #!/usr/bin/python
2
3 # Unix SMB/CIFS implementation. Tests for SamDB
4 # Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2008
5 #   
6 # This program is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 3 of the License, or
9 # (at your option) any later version.
10 #   
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 # GNU General Public License for more details.
15 #   
16 # You should have received a copy of the GNU General Public License
17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 #
19 from samba.auth import system_session
20 from samba.credentials import Credentials
21 import os
22 from samba.provision import setup_samdb, guess_names, setup_templatesdb, make_smbconf, find_setup_dir
23 from samba.samdb import SamDB
24 from samba.tests import TestCaseInTempDir
25 from samba.dcerpc import security
26 from unittest import TestCase
27 import uuid
28 from samba import param
29
30 class SamDBTestCase(TestCaseInTempDir):
31
32     def setup_path(self, relpath):
33         return os.path.join(find_setup_dir(), relpath)
34
35     def setUp(self):
36         super(SamDBTestCase, self).setUp()
37         invocationid = str(uuid.uuid4())
38         domaindn = "DC=COM,DC=EXAMPLE"
39         self.domaindn = domaindn
40         configdn = "CN=Configuration," + domaindn
41         schemadn = "CN=Schema," + configdn
42         domainguid = str(uuid.uuid4())
43         policyguid = str(uuid.uuid4())
44         creds = Credentials()
45         creds.set_anonymous()
46         domainsid = security.random_sid()
47         hostguid = str(uuid.uuid4())
48         path = os.path.join(self.tempdir, "samdb.ldb")
49         session_info = system_session()
50         
51         hostname="foo"
52         domain="EXAMPLE"
53         dnsdomain="example.com" 
54         serverrole="domain controller"
55
56         smbconf = os.path.join(self.tempdir, "smb.conf")
57         make_smbconf(smbconf, self.setup_path, hostname, domain, dnsdomain, 
58                      serverrole, self.tempdir)
59
60         self.lp = param.LoadParm()
61         self.lp.load(smbconf)
62
63         names = guess_names(lp=self.lp, hostname=hostname, 
64                             domain=domain, dnsdomain=dnsdomain, 
65                             serverrole=serverrole, 
66                             domaindn=self.domaindn, configdn=configdn, 
67                             schemadn=schemadn)
68         setup_templatesdb(os.path.join(self.tempdir, "templates.ldb"), 
69                           self.setup_path, session_info=session_info, 
70                           credentials=creds, lp=self.lp)
71         self.samdb = setup_samdb(path, self.setup_path, session_info, creds, 
72                                  self.lp, names, 
73                                  lambda x: None, domainsid, 
74                                  "# no aci", domainguid, 
75                                  policyguid, False, "secret", 
76                                  "secret", "secret", invocationid, 
77                                  "secret", "domain controller")
78
79     def tearDown(self):
80         for f in ['templates.ldb', 'schema.ldb', 'configuration.ldb', 
81                   'users.ldb', 'samdb.ldb', 'smb.conf']:
82             os.remove(os.path.join(self.tempdir, f))
83         super(SamDBTestCase, self).tearDown()
84
85     def test_add_foreign(self):
86         self.samdb.add_foreign(self.domaindn, "S-1-5-7", "Somedescription")
87