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 import os
  23 import sys
  24
  25 from cStringIO import StringIO
  26
  27 from selftest.tests import TestCase
  28
  29 from selftest.target.samba import (
  30     bindir_path,
  31     cleanup_child,
  32     mk_realms_stanza,
  33     write_krb5_conf,
  34     )
  35
  36
  37 class BinDirPathTests(TestCase):
  38
  39     def test_mapping(self):
  40         self.assertEquals("exe4",
  41             bindir_path({"exe": "exe4"}, "/some/path", "exe"))
  42         self.assertEquals("/bin/ls",
  43             bindir_path({"exe": "ls"}, "/bin", "exe"))
  44
  45     def test_no_mapping(self):
  46         self.assertEqual("exe", bindir_path({}, "/some/path", "exe"))
  47         self.assertEqual("/bin/ls",
  48             bindir_path({}, "/bin", "ls"))
  49
  50
  51 class MkRealmsStanzaTests(TestCase):
  52
  53     def test_basic(self):
  54         self.assertEqual(
  55            mk_realms_stanza("rijk", "dnsnaam", "domein", "ipv4_kdc"),
  56           '''\
  57  rijk = {
  58   kdc = ipv4_kdc:88
  59   admin_server = ipv4_kdc:88
  60   default_domain = dnsnaam
  61  }
  62  dnsnaam = {
  63   kdc = ipv4_kdc:88
  64   admin_server = ipv4_kdc:88
  65   default_domain = dnsnaam
  66  }
  67  domein = {
  68   kdc = ipv4_kdc:88
  69   admin_server = ipv4_kdc:88
  70   default_domain = dnsnaam
  71  }
  72
  73 ''')
  74
  75
  76 class WriteKrb5ConfTests(TestCase):
  77
  78     def test_simple(self):
  79         f = StringIO()
  80         write_krb5_conf(f, "rijk", "dnsnaam", "domein", "kdc_ipv4")
  81         self.assertEquals('''\
  82 #Generated krb5.conf for rijk
  83
  84 [libdefaults]
  85 \tdefault_realm = rijk
  86 \tdns_lookup_realm = false
  87 \tdns_lookup_kdc = false
  88 \tticket_lifetime = 24h
  89 \tforwardable = yes
  90 \tallow_weak_crypto = yes
  91
  92 [realms]
  93  rijk = {
  94   kdc = kdc_ipv4:88
  95   admin_server = kdc_ipv4:88
  96   default_domain = dnsnaam
  97  }
  98  dnsnaam = {
  99   kdc = kdc_ipv4:88
 100   admin_server = kdc_ipv4:88
 101   default_domain = dnsnaam
 102  }
 103  domein = {
 104   kdc = kdc_ipv4:88
 105   admin_server = kdc_ipv4:88
 106   default_domain = dnsnaam
 107  }
 108
 109 ''', f.getvalue())