← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test__chk_map.py

  • Committer: Andrew Bennetts
  • Date: 2009-11-25 07:27:43 UTC
  • mto: This revision was merged to the branch mainline in revision 4825.
  • Revision ID: andrew.bennetts@canonical.com-20091125072743-v6sv4m2mkt9iyslp
https://launchpad.net/bugs/426662
Terminate SSHSubprocesses when no refs to them are left, in case .close is never called.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test__chk_map.py
1
 
# Copyright (C) 2009, 2010, 2011 Canonical Ltd
 
1
# Copyright (C) 2009 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
25
25
 
26
26
 
27
27
def load_tests(standard_tests, module, loader):
28
 
    suite, _ = tests.permute_tests_for_extension(standard_tests, loader,
29
 
        'bzrlib._chk_map_py', 'bzrlib._chk_map_pyx')
 
28
    # parameterize all tests in this module
 
29
    suite = loader.suiteClass()
 
30
    import bzrlib._chk_map_py as py_module
 
31
    scenarios = [('python', {'module': py_module})]
 
32
    if CompiledChkMapFeature.available():
 
33
        import bzrlib._chk_map_pyx as c_module
 
34
        scenarios.append(('C', {'module': c_module}))
 
35
    else:
 
36
        # the compiled module isn't available, so we add a failing test
 
37
        class FailWithoutFeature(tests.TestCase):
 
38
            def test_fail(self):
 
39
                self.requireFeature(CompiledChkMapFeature)
 
40
        suite.addTest(loader.loadTestsFromTestCase(FailWithoutFeature))
 
41
    tests.multiply_tests(standard_tests, scenarios, suite)
30
42
    return suite
31
43
 
32
44
 
 
45
class _CompiledChkMapFeature(tests.Feature):
 
46
 
 
47
    def _probe(self):
 
48
        try:
 
49
            import bzrlib._chk_map_pyx
 
50
        except ImportError:
 
51
            return False
 
52
        return True
 
53
 
 
54
    def feature_name(self):
 
55
        return 'bzrlib._chk_map_pyx'
 
56
 
 
57
CompiledChkMapFeature = _CompiledChkMapFeature()
 
58
 
 
59
 
33
60
class TestSearchKeys(tests.TestCase):
34
61
 
35
62
    module = None # Filled in by test parameterization
236
263
        self.assertEqual(("sha1:1234",), node.key())
237
264
        self.assertEqual('pref\x00fo', node._search_prefix)
238
265
        self.assertEqual({'pref\x00fo\x00': ('sha1:abcd',)}, node._items)
239
 
 
240
 
 
241
 
class Test_BytesToTextKey(tests.TestCase):
242
 
 
243
 
    def assertBytesToTextKey(self, key, bytes):
244
 
        self.assertEqual(key,
245
 
                         self.module._bytes_to_text_key(bytes))
246
 
 
247
 
    def assertBytesToTextKeyRaises(self, bytes):
248
 
        # These are invalid bytes, and we want to make sure the code under test
249
 
        # raises an exception rather than segfaults, etc. We don't particularly
250
 
        # care what exception.
251
 
        self.assertRaises(Exception, self.module._bytes_to_text_key, bytes)
252
 
 
253
 
    def test_file(self):
254
 
        self.assertBytesToTextKey(('file-id', 'revision-id'),
255
 
                 'file: file-id\nparent-id\nname\nrevision-id\n'
256
 
                 'da39a3ee5e6b4b0d3255bfef95601890afd80709\n100\nN')
257
 
 
258
 
    def test_invalid_no_kind(self):
259
 
        self.assertBytesToTextKeyRaises(
260
 
                 'file  file-id\nparent-id\nname\nrevision-id\n'
261
 
                 'da39a3ee5e6b4b0d3255bfef95601890afd80709\n100\nN')
262
 
 
263
 
    def test_invalid_no_space(self):
264
 
        self.assertBytesToTextKeyRaises(
265
 
                 'file:file-id\nparent-id\nname\nrevision-id\n'
266
 
                 'da39a3ee5e6b4b0d3255bfef95601890afd80709\n100\nN')
267
 
 
268
 
    def test_invalid_too_short_file_id(self):
269
 
        self.assertBytesToTextKeyRaises('file:file-id')
270
 
 
271
 
    def test_invalid_too_short_parent_id(self):
272
 
        self.assertBytesToTextKeyRaises('file:file-id\nparent-id')
273
 
 
274
 
    def test_invalid_too_short_name(self):
275
 
        self.assertBytesToTextKeyRaises('file:file-id\nparent-id\nname')
276
 
 
277
 
    def test_dir(self):
278
 
        self.assertBytesToTextKey(('dir-id', 'revision-id'),
279
 
                 'dir: dir-id\nparent-id\nname\nrevision-id')