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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import optparse
import sys
sys.path.insert(0, 'bin/python')
import os
import samba
import samba.getopt as options
import random
import tempfile
import shutil
import subprocess
from samba.netcmd.main import samba_tool
# We try to use the test infrastructure of Samba 4.3+, but if it
# doesn't work, we are probably in a back-ported patch and trying to
# run on 4.1 or something.
#
# Don't copy this horror into ordinary tests -- it is special for
# performance tests that want to apply to old versions.
try:
from samba.tests.subunitrun import SubunitOptions, TestProgram
ANCIENT_SAMBA = False
except ImportError:
ANCIENT_SAMBA = True
samba.ensure_external_module("testtools", "testtools")
samba.ensure_external_module("subunit", "subunit/python")
from subunit.run import SubunitTestRunner
import unittest
parser = optparse.OptionParser("ad_dc_provision_performance.py [options] <host>")
sambaopts = options.SambaOptions(parser)
parser.add_option_group(sambaopts)
parser.add_option_group(options.VersionOptions(parser))
if not ANCIENT_SAMBA:
subunitopts = SubunitOptions(parser)
parser.add_option_group(subunitopts)
# use command line creds if available
credopts = options.CredentialsOptions(parser)
parser.add_option_group(credopts)
opts, args = parser.parse_args()
if len(args) < 1:
parser.print_usage()
sys.exit(1)
host = args[0]
lp = sambaopts.get_loadparm()
creds = credopts.get_credentials(lp)
random.seed(1)
class PerfTestException(Exception):
pass
class UserTests(samba.tests.TestCase):
def setUp(self):
super(UserTests, self).setUp()
self.tmpdir = tempfile.mkdtemp()
def tearDown(self):
shutil.rmtree(self.tmpdir)
def test_00_00_do_nothing(self):
# this gives us an idea of the overhead
pass
def _test_provision_subprocess(self, options=None, subdir=None):
if subdir is None:
d = self.tmpdir
else:
d = os.path.join(self.tmpdir, str(subdir))
os.mkdir(d)
cmd = ['bin/samba-tool', 'domain', 'provision', '--targetdir',
d, '--realm=realm.com', '--use-ntvfs', '--domain=dom']
if options:
options.extend(options)
subprocess.check_call(cmd)
test_01_00_provision_subprocess = _test_provision_subprocess
def test_01_00_provision_subprocess_overwrite(self):
for i in range(2):
self._test_provision_subprocess()
def test_02_00_provision_cmd_sambatool(self):
result = samba_tool('domain', 'provision',
'--targetdir=%s' % self.tmpdir,
'--use-ntvfs')
self.assertIsNone(result)
def test_03_00_provision_server_role(self):
for role in ('member', 'server', 'member', 'standalone'):
self._test_provision_subprocess(options=['--server-role', role],
subdir=role)
def test_04_00_provision_blank(self):
for i in range(2):
self._test_provision_subprocess(options=['--blank'],
subdir=i)
def test_05_00_provision_partitions_only(self):
self._test_provision_subprocess(options=['--partitions-only'])
if "://" not in host:
if os.path.isfile(host):
host = "tdb://%s" % host
else:
host = "ldap://%s" % host
if ANCIENT_SAMBA:
runner = SubunitTestRunner()
if not runner.run(unittest.TestLoader().loadTestsFromTestCase(
UserTests)).wasSuccessful():
sys.exit(1)
sys.exit(0)
else:
TestProgram(module=__name__, opts=subunitopts)
|
