~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_selftest.py

- refactor handling of short option names

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 by Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License version 2 as published by
5
 
# the Free Software Foundation.
6
 
#
7
 
# This program is distributed in the hope that it will be useful,
8
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
9
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10
 
# GNU General Public License for more details.
11
 
#
12
 
# You should have received a copy of the GNU General Public License
13
 
# along with this program; if not, write to the Free Software
14
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
15
 
 
16
 
"""Tests for the test framework
17
 
"""
18
 
 
19
 
import os
20
 
import sys
21
 
import unittest
22
 
import warnings
23
 
 
24
 
import bzrlib
25
 
from bzrlib.tests import (
26
 
                          _load_module_by_name,
27
 
                          ChrootedTestCase,
28
 
                          TestCase,
29
 
                          TestCaseInTempDir,
30
 
                          TestCaseWithTransport,
31
 
                          TestSkipped,
32
 
                          TextTestRunner,
33
 
                          )
34
 
import bzrlib.errors as errors
35
 
 
36
 
 
37
 
class SelftestTests(TestCase):
38
 
 
39
 
    def test_import_tests(self):
40
 
        mod = _load_module_by_name('bzrlib.tests.test_selftest')
41
 
        self.assertEqual(mod.SelftestTests, SelftestTests)
42
 
 
43
 
    def test_import_test_failure(self):
44
 
        self.assertRaises(ImportError,
45
 
                          _load_module_by_name,
46
 
                          'bzrlib.no-name-yet')
47
 
 
48
 
 
49
 
class MetaTestLog(TestCase):
50
 
 
51
 
    def test_logging(self):
52
 
        """Test logs are captured when a test fails."""
53
 
        self.log('a test message')
54
 
        self._log_file.flush()
55
 
        self.assertContainsRe(self._get_log(), 'a test message\n')
56
 
 
57
 
 
58
 
class TestTreeShape(TestCaseInTempDir):
59
 
 
60
 
    def test_unicode_paths(self):
61
 
        filename = u'hell\u00d8'
62
 
        try:
63
 
            self.build_tree_contents([(filename, 'contents of hello')])
64
 
        except UnicodeEncodeError:
65
 
            raise TestSkipped("can't build unicode working tree in "
66
 
                "filesystem encoding %s" % sys.getfilesystemencoding())
67
 
        self.failUnlessExists(filename)
68
 
 
69
 
 
70
 
class TestSkippedTest(TestCase):
71
 
    """Try running a test which is skipped, make sure it's reported properly."""
72
 
 
73
 
    def test_skipped_test(self):
74
 
        # must be hidden in here so it's not run as a real test
75
 
        def skipping_test():
76
 
            raise TestSkipped('test intentionally skipped')
77
 
        runner = TextTestRunner(stream=self._log_file)
78
 
        test = unittest.FunctionTestCase(skipping_test)
79
 
        result = runner.run(test)
80
 
        self.assertTrue(result.wasSuccessful())
81
 
 
82
 
 
83
 
class TestTransportProviderAdapter(TestCase):
84
 
    """A group of tests that test the transport implementation adaption core.
85
 
 
86
 
    This is a meta test that the tests are applied to all available 
87
 
    transports.
88
 
 
89
 
    This will be generalised in the future which is why it is in this 
90
 
    test file even though it is specific to transport tests at the moment.
91
 
    """
92
 
 
93
 
    def test_get_transport_permutations(self):
94
 
        # this checks that we the module get_test_permutations call
95
 
        # is made by the adapter get_transport_test_permitations method.
96
 
        class MockModule(object):
97
 
            def get_test_permutations(self):
98
 
                return sample_permutation
99
 
        sample_permutation = [(1,2), (3,4)]
100
 
        from bzrlib.transport import TransportTestProviderAdapter
101
 
        adapter = TransportTestProviderAdapter()
102
 
        self.assertEqual(sample_permutation,
103
 
                         adapter.get_transport_test_permutations(MockModule()))
104
 
 
105
 
    def test_adapter_checks_all_modules(self):
106
 
        # this checks that the adapter returns as many permurtations as
107
 
        # there are in all the registered# transport modules for there
108
 
        # - we assume if this matches its probably doing the right thing
109
 
        # especially in combination with the tests for setting the right
110
 
        # classes below.
111
 
        from bzrlib.transport import (TransportTestProviderAdapter,
112
 
                                      _get_transport_modules
113
 
                                      )
114
 
        modules = _get_transport_modules()
115
 
        permutation_count = 0
116
 
        for module in modules:
117
 
            try:
118
 
                permutation_count += len(reduce(getattr, 
119
 
                    (module + ".get_test_permutations").split('.')[1:],
120
 
                     __import__(module))())
121
 
            except errors.DependencyNotPresent:
122
 
                pass
123
 
        input_test = TestTransportProviderAdapter(
124
 
            "test_adapter_sets_transport_class")
125
 
        adapter = TransportTestProviderAdapter()
126
 
        self.assertEqual(permutation_count,
127
 
                         len(list(iter(adapter.adapt(input_test)))))
128
 
 
129
 
    def test_adapter_sets_transport_class(self):
130
 
        # Check that the test adapter inserts a transport and server into the
131
 
        # generated test.
132
 
        #
133
 
        # This test used to know about all the possible transports and the
134
 
        # order they were returned but that seems overly brittle (mbp
135
 
        # 20060307)
136
 
        input_test = TestTransportProviderAdapter(
137
 
            "test_adapter_sets_transport_class")
138
 
        from bzrlib.transport import TransportTestProviderAdapter
139
 
        suite = TransportTestProviderAdapter().adapt(input_test)
140
 
        tests = list(iter(suite))
141
 
        self.assertTrue(len(tests) > 6)
142
 
        # there are at least that many builtin transports
143
 
        one_test = tests[0]
144
 
        self.assertTrue(issubclass(one_test.transport_class, 
145
 
                                   bzrlib.transport.Transport))
146
 
        self.assertTrue(issubclass(one_test.transport_server, 
147
 
                                   bzrlib.transport.Server))
148
 
 
149
 
 
150
 
class TestBranchProviderAdapter(TestCase):
151
 
    """A group of tests that test the branch implementation test adapter."""
152
 
 
153
 
    def test_adapted_tests(self):
154
 
        # check that constructor parameters are passed through to the adapted
155
 
        # test.
156
 
        from bzrlib.branch import BranchTestProviderAdapter
157
 
        input_test = TestBranchProviderAdapter(
158
 
            "test_adapted_tests")
159
 
        server1 = "a"
160
 
        server2 = "b"
161
 
        formats = [("c", "C"), ("d", "D")]
162
 
        adapter = BranchTestProviderAdapter(server1, server2, formats)
163
 
        suite = adapter.adapt(input_test)
164
 
        tests = list(iter(suite))
165
 
        self.assertEqual(2, len(tests))
166
 
        self.assertEqual(tests[0].branch_format, formats[0][0])
167
 
        self.assertEqual(tests[0].bzrdir_format, formats[0][1])
168
 
        self.assertEqual(tests[0].transport_server, server1)
169
 
        self.assertEqual(tests[0].transport_readonly_server, server2)
170
 
        self.assertEqual(tests[1].branch_format, formats[1][0])
171
 
        self.assertEqual(tests[1].bzrdir_format, formats[1][1])
172
 
        self.assertEqual(tests[1].transport_server, server1)
173
 
        self.assertEqual(tests[1].transport_readonly_server, server2)
174
 
 
175
 
 
176
 
class TestBzrDirProviderAdapter(TestCase):
177
 
    """A group of tests that test the bzr dir implementation test adapter."""
178
 
 
179
 
    def test_adapted_tests(self):
180
 
        # check that constructor parameters are passed through to the adapted
181
 
        # test.
182
 
        from bzrlib.bzrdir import BzrDirTestProviderAdapter
183
 
        input_test = TestBzrDirProviderAdapter(
184
 
            "test_adapted_tests")
185
 
        server1 = "a"
186
 
        server2 = "b"
187
 
        formats = ["c", "d"]
188
 
        adapter = BzrDirTestProviderAdapter(server1, server2, formats)
189
 
        suite = adapter.adapt(input_test)
190
 
        tests = list(iter(suite))
191
 
        self.assertEqual(2, len(tests))
192
 
        self.assertEqual(tests[0].bzrdir_format, formats[0])
193
 
        self.assertEqual(tests[0].transport_server, server1)
194
 
        self.assertEqual(tests[0].transport_readonly_server, server2)
195
 
        self.assertEqual(tests[1].bzrdir_format, formats[1])
196
 
        self.assertEqual(tests[1].transport_server, server1)
197
 
        self.assertEqual(tests[1].transport_readonly_server, server2)
198
 
 
199
 
 
200
 
class TestRepositoryProviderAdapter(TestCase):
201
 
    """A group of tests that test the repository implementation test adapter."""
202
 
 
203
 
    def test_adapted_tests(self):
204
 
        # check that constructor parameters are passed through to the adapted
205
 
        # test.
206
 
        from bzrlib.repository import RepositoryTestProviderAdapter
207
 
        input_test = TestRepositoryProviderAdapter(
208
 
            "test_adapted_tests")
209
 
        server1 = "a"
210
 
        server2 = "b"
211
 
        formats = [("c", "C"), ("d", "D")]
212
 
        adapter = RepositoryTestProviderAdapter(server1, server2, formats)
213
 
        suite = adapter.adapt(input_test)
214
 
        tests = list(iter(suite))
215
 
        self.assertEqual(2, len(tests))
216
 
        self.assertEqual(tests[0].bzrdir_format, formats[0][1])
217
 
        self.assertEqual(tests[0].repository_format, formats[0][0])
218
 
        self.assertEqual(tests[0].transport_server, server1)
219
 
        self.assertEqual(tests[0].transport_readonly_server, server2)
220
 
        self.assertEqual(tests[1].bzrdir_format, formats[1][1])
221
 
        self.assertEqual(tests[1].repository_format, formats[1][0])
222
 
        self.assertEqual(tests[1].transport_server, server1)
223
 
        self.assertEqual(tests[1].transport_readonly_server, server2)
224
 
 
225
 
 
226
 
class TestInterRepositoryProviderAdapter(TestCase):
227
 
    """A group of tests that test the InterRepository test adapter."""
228
 
 
229
 
    def test_adapted_tests(self):
230
 
        # check that constructor parameters are passed through to the adapted
231
 
        # test.
232
 
        from bzrlib.repository import InterRepositoryTestProviderAdapter
233
 
        input_test = TestInterRepositoryProviderAdapter(
234
 
            "test_adapted_tests")
235
 
        server1 = "a"
236
 
        server2 = "b"
237
 
        formats = [(str, "C1", "C2"), (int, "D1", "D2")]
238
 
        adapter = InterRepositoryTestProviderAdapter(server1, server2, formats)
239
 
        suite = adapter.adapt(input_test)
240
 
        tests = list(iter(suite))
241
 
        self.assertEqual(2, len(tests))
242
 
        self.assertEqual(tests[0].interrepo_class, formats[0][0])
243
 
        self.assertEqual(tests[0].repository_format, formats[0][1])
244
 
        self.assertEqual(tests[0].repository_format_to, formats[0][2])
245
 
        self.assertEqual(tests[0].transport_server, server1)
246
 
        self.assertEqual(tests[0].transport_readonly_server, server2)
247
 
        self.assertEqual(tests[1].interrepo_class, formats[1][0])
248
 
        self.assertEqual(tests[1].repository_format, formats[1][1])
249
 
        self.assertEqual(tests[1].repository_format_to, formats[1][2])
250
 
        self.assertEqual(tests[1].transport_server, server1)
251
 
        self.assertEqual(tests[1].transport_readonly_server, server2)
252
 
 
253
 
 
254
 
class TestInterVersionedFileProviderAdapter(TestCase):
255
 
    """A group of tests that test the InterVersionedFile test adapter."""
256
 
 
257
 
    def test_adapted_tests(self):
258
 
        # check that constructor parameters are passed through to the adapted
259
 
        # test.
260
 
        from bzrlib.versionedfile import InterVersionedFileTestProviderAdapter
261
 
        input_test = TestInterRepositoryProviderAdapter(
262
 
            "test_adapted_tests")
263
 
        server1 = "a"
264
 
        server2 = "b"
265
 
        formats = [(str, "C1", "C2"), (int, "D1", "D2")]
266
 
        adapter = InterVersionedFileTestProviderAdapter(server1, server2, formats)
267
 
        suite = adapter.adapt(input_test)
268
 
        tests = list(iter(suite))
269
 
        self.assertEqual(2, len(tests))
270
 
        self.assertEqual(tests[0].interversionedfile_class, formats[0][0])
271
 
        self.assertEqual(tests[0].versionedfile_factory, formats[0][1])
272
 
        self.assertEqual(tests[0].versionedfile_factory_to, formats[0][2])
273
 
        self.assertEqual(tests[0].transport_server, server1)
274
 
        self.assertEqual(tests[0].transport_readonly_server, server2)
275
 
        self.assertEqual(tests[1].interversionedfile_class, formats[1][0])
276
 
        self.assertEqual(tests[1].versionedfile_factory, formats[1][1])
277
 
        self.assertEqual(tests[1].versionedfile_factory_to, formats[1][2])
278
 
        self.assertEqual(tests[1].transport_server, server1)
279
 
        self.assertEqual(tests[1].transport_readonly_server, server2)
280
 
 
281
 
 
282
 
class TestRevisionStoreProviderAdapter(TestCase):
283
 
    """A group of tests that test the RevisionStore test adapter."""
284
 
 
285
 
    def test_adapted_tests(self):
286
 
        # check that constructor parameters are passed through to the adapted
287
 
        # test.
288
 
        from bzrlib.store.revision import RevisionStoreTestProviderAdapter
289
 
        input_test = TestRevisionStoreProviderAdapter(
290
 
            "test_adapted_tests")
291
 
        # revision stores need a store factory - i.e. RevisionKnit
292
 
        #, a readonly and rw transport 
293
 
        # transport servers:
294
 
        server1 = "a"
295
 
        server2 = "b"
296
 
        store_factories = ["c", "d"]
297
 
        adapter = RevisionStoreTestProviderAdapter(server1, server2, store_factories)
298
 
        suite = adapter.adapt(input_test)
299
 
        tests = list(iter(suite))
300
 
        self.assertEqual(2, len(tests))
301
 
        self.assertEqual(tests[0].store_factory, store_factories[0][0])
302
 
        self.assertEqual(tests[0].transport_server, server1)
303
 
        self.assertEqual(tests[0].transport_readonly_server, server2)
304
 
        self.assertEqual(tests[1].store_factory, store_factories[1][0])
305
 
        self.assertEqual(tests[1].transport_server, server1)
306
 
        self.assertEqual(tests[1].transport_readonly_server, server2)
307
 
 
308
 
 
309
 
class TestWorkingTreeProviderAdapter(TestCase):
310
 
    """A group of tests that test the workingtree implementation test adapter."""
311
 
 
312
 
    def test_adapted_tests(self):
313
 
        # check that constructor parameters are passed through to the adapted
314
 
        # test.
315
 
        from bzrlib.workingtree import WorkingTreeTestProviderAdapter
316
 
        input_test = TestWorkingTreeProviderAdapter(
317
 
            "test_adapted_tests")
318
 
        server1 = "a"
319
 
        server2 = "b"
320
 
        formats = [("c", "C"), ("d", "D")]
321
 
        adapter = WorkingTreeTestProviderAdapter(server1, server2, formats)
322
 
        suite = adapter.adapt(input_test)
323
 
        tests = list(iter(suite))
324
 
        self.assertEqual(2, len(tests))
325
 
        self.assertEqual(tests[0].workingtree_format, formats[0][0])
326
 
        self.assertEqual(tests[0].bzrdir_format, formats[0][1])
327
 
        self.assertEqual(tests[0].transport_server, server1)
328
 
        self.assertEqual(tests[0].transport_readonly_server, server2)
329
 
        self.assertEqual(tests[1].workingtree_format, formats[1][0])
330
 
        self.assertEqual(tests[1].bzrdir_format, formats[1][1])
331
 
        self.assertEqual(tests[1].transport_server, server1)
332
 
        self.assertEqual(tests[1].transport_readonly_server, server2)
333
 
 
334
 
 
335
 
class TestTestCaseWithTransport(TestCaseWithTransport):
336
 
    """Tests for the convenience functions TestCaseWithTransport introduces."""
337
 
 
338
 
    def test_get_readonly_url_none(self):
339
 
        from bzrlib.transport import get_transport
340
 
        from bzrlib.transport.memory import MemoryServer
341
 
        from bzrlib.transport.readonly import ReadonlyTransportDecorator
342
 
        self.transport_server = MemoryServer
343
 
        self.transport_readonly_server = None
344
 
        # calling get_readonly_transport() constructs a decorator on the url
345
 
        # for the server
346
 
        url = self.get_readonly_url()
347
 
        url2 = self.get_readonly_url('foo/bar')
348
 
        t = get_transport(url)
349
 
        t2 = get_transport(url2)
350
 
        self.failUnless(isinstance(t, ReadonlyTransportDecorator))
351
 
        self.failUnless(isinstance(t2, ReadonlyTransportDecorator))
352
 
        self.assertEqual(t2.base[:-1], t.abspath('foo/bar'))
353
 
 
354
 
    def test_get_readonly_url_http(self):
355
 
        from bzrlib.transport import get_transport
356
 
        from bzrlib.transport.local import LocalRelpathServer
357
 
        from bzrlib.transport.http import HttpServer, HttpTransportBase
358
 
        self.transport_server = LocalRelpathServer
359
 
        self.transport_readonly_server = HttpServer
360
 
        # calling get_readonly_transport() gives us a HTTP server instance.
361
 
        url = self.get_readonly_url()
362
 
        url2 = self.get_readonly_url('foo/bar')
363
 
        # the transport returned may be any HttpTransportBase subclass
364
 
        t = get_transport(url)
365
 
        t2 = get_transport(url2)
366
 
        self.failUnless(isinstance(t, HttpTransportBase))
367
 
        self.failUnless(isinstance(t2, HttpTransportBase))
368
 
        self.assertEqual(t2.base[:-1], t.abspath('foo/bar'))
369
 
 
370
 
    def test_is_directory(self):
371
 
        """Test assertIsDirectory assertion"""
372
 
        t = self.get_transport()
373
 
        self.build_tree(['a_dir/', 'a_file'], transport=t)
374
 
        self.assertIsDirectory('a_dir', t)
375
 
        self.assertRaises(AssertionError, self.assertIsDirectory, 'a_file', t)
376
 
        self.assertRaises(AssertionError, self.assertIsDirectory, 'not_here', t)
377
 
 
378
 
 
379
 
class TestChrootedTest(ChrootedTestCase):
380
 
 
381
 
    def test_root_is_root(self):
382
 
        from bzrlib.transport import get_transport
383
 
        t = get_transport(self.get_readonly_url())
384
 
        url = t.base
385
 
        self.assertEqual(url, t.clone('..').base)
386
 
 
387
 
 
388
 
class TestExtraAssertions(TestCase):
389
 
    """Tests for new test assertions in bzrlib test suite"""
390
 
 
391
 
    def test_assert_isinstance(self):
392
 
        self.assertIsInstance(2, int)
393
 
        self.assertIsInstance(u'', basestring)
394
 
        self.assertRaises(AssertionError, self.assertIsInstance, None, int)
395
 
        self.assertRaises(AssertionError, self.assertIsInstance, 23.3, int)
396
 
 
397
 
 
398
 
class TestConvenienceMakers(TestCaseWithTransport):
399
 
    """Test for the make_* convenience functions."""
400
 
 
401
 
    def test_make_branch_and_tree_with_format(self):
402
 
        # we should be able to supply a format to make_branch_and_tree
403
 
        self.make_branch_and_tree('a', format=bzrlib.bzrdir.BzrDirMetaFormat1())
404
 
        self.make_branch_and_tree('b', format=bzrlib.bzrdir.BzrDirFormat6())
405
 
        self.assertIsInstance(bzrlib.bzrdir.BzrDir.open('a')._format,
406
 
                              bzrlib.bzrdir.BzrDirMetaFormat1)
407
 
        self.assertIsInstance(bzrlib.bzrdir.BzrDir.open('b')._format,
408
 
                              bzrlib.bzrdir.BzrDirFormat6)
409