/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/test_lazy_regex.py

  • Committer: Jelmer Vernooij
  • Date: 2020-04-05 19:11:34 UTC
  • mto: (7490.7.16 work)
  • mto: This revision was merged to the branch mainline in revision 7501.
  • Revision ID: jelmer@jelmer.uk-20200405191134-0aebh8ikiwygxma5
Populate the .gitignore file.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test_lazy_regex.py
1
 
# Copyright (C) 2006 Canonical Ltd
 
1
# Copyright (C) 2006, 2011 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
16
16
 
17
17
"""Test that lazy regexes are not compiled right away"""
18
18
 
 
19
import pickle
19
20
import re
20
21
 
21
 
from bzrlib import (
 
22
from .. import errors
 
23
from .. import (
22
24
    lazy_regex,
23
25
    tests,
24
26
    )
25
27
 
26
28
 
 
29
class TestErrors(tests.TestCase):
 
30
 
 
31
    def test_invalid_pattern(self):
 
32
        error = lazy_regex.InvalidPattern('Bad pattern msg.')
 
33
        self.assertEqualDiff("Invalid pattern(s) found. Bad pattern msg.",
 
34
                             str(error))
 
35
 
 
36
 
27
37
class InstrumentedLazyRegex(lazy_regex.LazyRegex):
28
38
    """Keep track of actions on the lazy regex"""
29
39
 
38
48
        return super(InstrumentedLazyRegex, self).__getattr__(attr)
39
49
 
40
50
    def _real_re_compile(self, *args, **kwargs):
41
 
        self._actions.append(('_real_re_compile',
42
 
                                               args, kwargs))
43
 
        return super(InstrumentedLazyRegex, self)._real_re_compile(*args, **kwargs)
 
51
        self._actions.append(('_real_re_compile', args, kwargs))
 
52
        return super(InstrumentedLazyRegex, self)._real_re_compile(
 
53
            *args, **kwargs)
44
54
 
45
55
 
46
56
class TestLazyRegex(tests.TestCase):
50
60
        actions = []
51
61
        InstrumentedLazyRegex.use_actions(actions)
52
62
 
53
 
        pattern = InstrumentedLazyRegex(args=('foo',))
 
63
        pattern = InstrumentedLazyRegex(args=('foo',), kwargs={})
54
64
        actions.append(('created regex', 'foo'))
55
65
        # This match call should compile the regex and go through __getattr__
56
66
        pattern.match('foo')
61
71
        self.assertEqual([('created regex', 'foo'),
62
72
                          ('__getattr__', 'match'),
63
73
                          ('_real_re_compile', ('foo',), {}),
64
 
                         ], actions)
 
74
                          ], actions)
 
75
 
 
76
    def test_bad_pattern(self):
 
77
        """Ensure lazy regex handles bad patterns cleanly."""
 
78
        p = lazy_regex.lazy_compile('RE:[')
 
79
        # As p.match is lazy, we make it into a lambda so its handled
 
80
        # by assertRaises correctly.
 
81
        e = self.assertRaises(lazy_regex.InvalidPattern,
 
82
                              lambda: p.match('foo'))
 
83
        # Expect either old or new form of error message
 
84
        self.assertContainsRe(e.msg, '^"RE:\\[" '
 
85
                              '(unexpected end of regular expression'
 
86
                              '|unterminated character set at position 3)$')
65
87
 
66
88
 
67
89
class TestLazyCompile(tests.TestCase):
102
124
        self.assertEqual('fooo', pattern.search('fooo').group())
103
125
 
104
126
    def test_split(self):
105
 
        pattern = lazy_regex.lazy_compile('[,;]*')
 
127
        pattern = lazy_regex.lazy_compile('[,;]+')
106
128
        self.assertEqual(['x', 'y', 'z'], pattern.split('x,y;z'))
107
129
 
108
 
 
109
 
class TestInstallLazyCompile(tests.TestCase):
110
 
 
111
 
    def setUp(self):
112
 
        super(TestInstallLazyCompile, self).setUp()
113
 
        self.addCleanup(lazy_regex.reset_compile)
114
 
 
115
 
    def test_install(self):
116
 
        lazy_regex.install_lazy_compile()
117
 
        pattern = re.compile('foo')
118
 
        self.assertIsInstance(pattern, lazy_regex.LazyRegex)
119
 
 
120
 
    def test_reset(self):
121
 
        lazy_regex.install_lazy_compile()
122
 
        lazy_regex.reset_compile()
123
 
        pattern = re.compile('foo')
124
 
        self.failIf(isinstance(pattern, lazy_regex.LazyRegex),
125
 
                    'lazy_regex.reset_compile() did not restore the original'
126
 
                    ' compile() function %s' % (type(pattern),))
127
 
        # but the returned object should still support regex operations
128
 
        m = pattern.match('foo')
129
 
        self.assertEqual('foo', m.group())
 
130
    def test_pickle(self):
 
131
        # When pickling, just compile the regex.
 
132
        # Sphinx, which we use for documentation, pickles
 
133
        # some compiled regexes.
 
134
        lazy_pattern = lazy_regex.lazy_compile('[,;]+')
 
135
        pickled = pickle.dumps(lazy_pattern)
 
136
        unpickled_lazy_pattern = pickle.loads(pickled)
 
137
        self.assertEqual(
 
138
            ['x', 'y', 'z'], unpickled_lazy_pattern.split('x,y;z'))