selftest.py: Add get_interface.
[ab/samba.git/.git] / selftest / tests / test_samba.py
1 # test_run.py -- Tests for selftest.target.samba
2 # Copyright (C) 2012 Jelmer Vernooij <jelmer@samba.org>
3 #
4 # This program is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU General Public License
6 # as published by the Free Software Foundation; version 3
7 # of the License or (at your option) any later version of
8 # the License.
9 #
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with this program; if not, write to the Free Software
17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
18 # MA  02110-1301, USA.
19
20 """Tests for selftest.target.samba."""
21
22 from cStringIO import StringIO
23
24 from selftest.tests import TestCase
25
26 from selftest.target.samba import (
27     bindir_path,
28     get_interface,
29     mk_realms_stanza,
30     write_krb5_conf,
31     )
32
33
34 class BinDirPathTests(TestCase):
35
36     def test_mapping(self):
37         self.assertEquals("exe4",
38             bindir_path({"exe": "exe4"}, "/some/path", "exe"))
39         self.assertEquals("/bin/ls",
40             bindir_path({"exe": "ls"}, "/bin", "exe"))
41
42     def test_no_mapping(self):
43         self.assertEqual("exe", bindir_path({}, "/some/path", "exe"))
44         self.assertEqual("/bin/ls",
45             bindir_path({}, "/bin", "ls"))
46
47
48 class MkRealmsStanzaTests(TestCase):
49
50     def test_basic(self):
51         self.assertEqual(
52            mk_realms_stanza("rijk", "dnsnaam", "domein", "ipv4_kdc"),
53           '''\
54  rijk = {
55   kdc = ipv4_kdc:88
56   admin_server = ipv4_kdc:88
57   default_domain = dnsnaam
58  }
59  dnsnaam = {
60   kdc = ipv4_kdc:88
61   admin_server = ipv4_kdc:88
62   default_domain = dnsnaam
63  }
64  domein = {
65   kdc = ipv4_kdc:88
66   admin_server = ipv4_kdc:88
67   default_domain = dnsnaam
68  }
69
70 ''')
71
72
73 class WriteKrb5ConfTests(TestCase):
74
75     def test_simple(self):
76         f = StringIO()
77         write_krb5_conf(f, "rijk", "dnsnaam", "domein", "kdc_ipv4")
78         self.assertEquals('''\
79 #Generated krb5.conf for rijk
80
81 [libdefaults]
82 \tdefault_realm = rijk
83 \tdns_lookup_realm = false
84 \tdns_lookup_kdc = false
85 \tticket_lifetime = 24h
86 \tforwardable = yes
87 \tallow_weak_crypto = yes
88
89 [realms]
90  rijk = {
91   kdc = kdc_ipv4:88
92   admin_server = kdc_ipv4:88
93   default_domain = dnsnaam
94  }
95  dnsnaam = {
96   kdc = kdc_ipv4:88
97   admin_server = kdc_ipv4:88
98   default_domain = dnsnaam
99  }
100  domein = {
101   kdc = kdc_ipv4:88
102   admin_server = kdc_ipv4:88
103   default_domain = dnsnaam
104  }
105
106 ''', f.getvalue())
107
108
109 class GetInterfaceTests(TestCase):
110
111     def test_get_interface(self):
112         self.assertEquals(21, get_interface("localdc"))
113         self.assertEquals(4, get_interface("localshare4"))
114
115     def test_unknown(self):
116         self.assertRaises(KeyError, get_interface, "unknown")