summaryrefslogtreecommitdiff
path: root/selftest/tests/test_samba.py
blob: b49463eabb96e1ad693250d620cdad5abb72663e (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
108
109
110
111
112
113
114
115
116
# test_run.py -- Tests for selftest.target.samba
# Copyright (C) 2012 Jelmer Vernooij <jelmer@samba.org>
#
# 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; version 3
# of the License or (at your option) any later version of
# the License.
#
# 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, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA  02110-1301, USA.

"""Tests for selftest.target.samba."""

from cStringIO import StringIO

from selftest.tests import TestCase

from selftest.target.samba import (
    bindir_path,
    get_interface,
    mk_realms_stanza,
    write_krb5_conf,
    )


class BinDirPathTests(TestCase):

    def test_mapping(self):
        self.assertEquals("exe4",
            bindir_path({"exe": "exe4"}, "/some/path", "exe"))
        self.assertEquals("/bin/ls",
            bindir_path({"exe": "ls"}, "/bin", "exe"))

    def test_no_mapping(self):
        self.assertEqual("exe", bindir_path({}, "/some/path", "exe"))
        self.assertEqual("/bin/ls",
            bindir_path({}, "/bin", "ls"))


class MkRealmsStanzaTests(TestCase):

    def test_basic(self):
        self.assertEqual(
           mk_realms_stanza("rijk", "dnsnaam", "domein", "ipv4_kdc"),
          '''\
 rijk = {
  kdc = ipv4_kdc:88
  admin_server = ipv4_kdc:88
  default_domain = dnsnaam
 }
 dnsnaam = {
  kdc = ipv4_kdc:88
  admin_server = ipv4_kdc:88
  default_domain = dnsnaam
 }
 domein = {
  kdc = ipv4_kdc:88
  admin_server = ipv4_kdc:88
  default_domain = dnsnaam
 }

''')


class WriteKrb5ConfTests(TestCase):

    def test_simple(self):
        f = StringIO()
        write_krb5_conf(f, "rijk", "dnsnaam", "domein", "kdc_ipv4")
        self.assertEquals('''\
#Generated krb5.conf for rijk

[libdefaults]
\tdefault_realm = rijk
\tdns_lookup_realm = false
\tdns_lookup_kdc = false
\tticket_lifetime = 24h
\tforwardable = yes
\tallow_weak_crypto = yes

[realms]
 rijk = {
  kdc = kdc_ipv4:88
  admin_server = kdc_ipv4:88
  default_domain = dnsnaam
 }
 dnsnaam = {
  kdc = kdc_ipv4:88
  admin_server = kdc_ipv4:88
  default_domain = dnsnaam
 }
 domein = {
  kdc = kdc_ipv4:88
  admin_server = kdc_ipv4:88
  default_domain = dnsnaam
 }

''', f.getvalue())


class GetInterfaceTests(TestCase):

    def test_get_interface(self):
        self.assertEquals(21, get_interface("localdc"))
        self.assertEquals(4, get_interface("localshare4"))

    def test_unknown(self):
        self.assertRaises(KeyError, get_interface, "unknown")