summaryrefslogtreecommitdiff
path: root/python/samba/tests/param.py
blob: 0a7f86adad75d83fe46a91956e23854ea17d93a2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# Unix SMB/CIFS implementation.
# Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
#
# 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/>.
#

"""Tests for samba.param."""

from samba import param
import samba.tests
import os


class LoadParmTestCase(samba.tests.TestCaseInTempDir):

    def setUp(self):
        super(LoadParmTestCase, self).setUp()
        self.tempf = os.path.join(self.tempdir, "test")
        open(self.tempf, 'w').write("empty")

    def tearDown(self):
        os.unlink(self.tempf)
        super(LoadParmTestCase, self).tearDown()

    def test_init(self):
        file = param.LoadParm()
        self.assertTrue(file is not None)

    def test_length(self):
        file = param.LoadParm()
        self.assertEqual(0, len(file))

    def test_set_workgroup(self):
        file = param.LoadParm()
        file.set("workgroup", "bla")
        self.assertEqual("BLA", file.get("workgroup"))

    def test_is_mydomain(self):
        file = param.LoadParm()
        file.set("workgroup", "bla")
        self.assertTrue(file.is_mydomain("BLA"))
        self.assertFalse(file.is_mydomain("FOOBAR"))

    def test_is_myname(self):
        file = param.LoadParm()
        file.set("netbios name", "bla")
        self.assertTrue(file.is_myname("BLA"))
        self.assertFalse(file.is_myname("FOOBAR"))

    def test_load_default(self):
        file = param.LoadParm()
        file.load_default()

    def test_section_nonexistent(self):
        samba_lp = param.LoadParm()
        samba_lp.load_default()
        self.assertRaises(KeyError, samba_lp.__getitem__, "nonexistent")

    def test_log_level(self):
        samba_lp = param.LoadParm()
        samba_lp.set("log level", "5 auth:4")
        self.assertEqual(5, samba_lp.log_level())

    def test_dump(self):
        samba_lp = param.LoadParm()
        # Just test successfull method execution (outputs to stdout)
        self.assertEqual(None, samba_lp.dump())

    def test_dump_to_file(self):
        samba_lp = param.LoadParm()
        self.assertEqual(None, samba_lp.dump(False, self.tempf))
        content = open(self.tempf, 'r').read()
        self.assertIn('[global]', content)
        self.assertIn('interfaces', content)

    def test_dump_a_parameter(self):
        samba_lp = param.LoadParm()
        samba_lp.load_default()
        # Just test successfull method execution
        self.assertEqual(None, samba_lp.dump_a_parameter('interfaces'))

    def test_dump_a_parameter_to_file(self):
        samba_lp = param.LoadParm()
        samba_lp.load_default()
        self.assertEqual(None,
                          samba_lp.dump_a_parameter('interfaces',
                                                    'global',
                                                    self.tempf))
        content = open(self.tempf, 'r').read()
        self.assertIn('127.0.0.', content)

    def test_samdb_url(self):
        samba_lp = param.LoadParm()
        samdb_url = samba_lp.samdb_url()
        self.assertTrue(samdb_url.startswith('tdb://'))
        self.assertTrue(samdb_url.endswith('/sam.ldb'))