/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_fifo_cache.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_fifo_cache.py
16
16
 
17
17
"""Tests for the fifo_cache module."""
18
18
 
19
 
from bzrlib import (
 
19
from .. import (
20
20
    fifo_cache,
21
21
    tests,
22
22
    )
 
23
from ..sixish import (
 
24
    viewitems,
 
25
    viewkeys,
 
26
    viewvalues,
 
27
    )
23
28
 
24
29
 
25
30
class TestFIFOCache(tests.TestCase):
33
38
        self.assertEqual(2, c[1])
34
39
        self.assertEqual(2, c.get(1))
35
40
        self.assertEqual(2, c.get(1, None))
36
 
        self.assertEqual([1], c.keys())
37
 
        self.assertEqual([1], list(c.iterkeys()))
38
 
        self.assertEqual([(1, 2)], c.items())
39
 
        self.assertEqual([(1, 2)], list(c.iteritems()))
40
 
        self.assertEqual([2], c.values())
41
 
        self.assertEqual([2], list(c.itervalues()))
 
41
        self.assertEqual([1], list(c))
 
42
        self.assertEqual({1}, viewkeys(c))
 
43
        self.assertEqual([(1, 2)], sorted(viewitems(c)))
 
44
        self.assertEqual([2], sorted(viewvalues(c)))
42
45
        self.assertEqual({1: 2}, c)
43
46
 
44
47
    def test_cache_size(self):
54
57
        self.assertEqual(0, len(c))
55
58
        self.assertEqual(None, c.get(1))
56
59
        self.assertEqual(None, c.get(1, None))
57
 
        self.assertEqual([], c.keys())
58
 
        self.assertEqual([], list(c.iterkeys()))
59
 
        self.assertEqual([], c.items())
60
 
        self.assertEqual([], list(c.iteritems()))
61
 
        self.assertEqual([], c.values())
62
 
        self.assertEqual([], list(c.itervalues()))
 
60
        self.assertEqual([], list(c))
 
61
        self.assertEqual(set(), viewkeys(c))
 
62
        self.assertEqual([], list(viewitems(c)))
 
63
        self.assertEqual([], list(viewvalues(c)))
63
64
        self.assertEqual({}, c)
64
65
 
65
66
    def test_add_maintains_fifo(self):
68
69
        c[2] = 3
69
70
        c[3] = 4
70
71
        c[4] = 5
71
 
        self.assertEqual([1, 2, 3, 4], sorted(c.keys()))
 
72
        self.assertEqual({1, 2, 3, 4}, viewkeys(c))
72
73
        c[5] = 6
73
74
        # This should pop out the oldest entry
74
 
        self.assertEqual([2, 3, 4, 5], sorted(c.keys()))
 
75
        self.assertEqual({2, 3, 4, 5}, viewkeys(c))
75
76
        # Replacing an item doesn't change the stored keys
76
77
        c[2] = 7
77
 
        self.assertEqual([2, 3, 4, 5], sorted(c.keys()))
 
78
        self.assertEqual({2, 3, 4, 5}, viewkeys(c))
78
79
        # But it does change the position in the FIFO
79
80
        c[6] = 7
80
 
        self.assertEqual([2, 4, 5, 6], sorted(c.keys()))
 
81
        self.assertEqual({2, 4, 5, 6}, viewkeys(c))
81
82
        self.assertEqual([4, 5, 2, 6], list(c._queue))
82
83
 
83
84
    def test_default_after_cleanup_count(self):
89
90
        c[4] = 5
90
91
        c[5] = 6
91
92
        # So far, everything fits
92
 
        self.assertEqual([1, 2, 3, 4, 5], sorted(c.keys()))
 
93
        self.assertEqual({1, 2, 3, 4, 5}, viewkeys(c))
93
94
        c[6] = 7
94
95
        # But adding one more should shrink down to after_cleanup_count
95
 
        self.assertEqual([3, 4, 5, 6], sorted(c.keys()))
 
96
        self.assertEqual({3, 4, 5, 6}, viewkeys(c))
96
97
 
97
98
    def test_clear(self):
98
99
        c = fifo_cache.FIFOCache(5)
102
103
        c[4] = 5
103
104
        c[5] = 6
104
105
        c.cleanup()
105
 
        self.assertEqual([2, 3, 4, 5], sorted(c.keys()))
 
106
        self.assertEqual({2, 3, 4, 5}, viewkeys(c))
106
107
        c.clear()
107
 
        self.assertEqual([], c.keys())
 
108
        self.assertEqual(set(), viewkeys(c))
108
109
        self.assertEqual([], list(c._queue))
109
110
        self.assertEqual({}, c)
110
111
 
193
194
 
194
195
    def test_cleanup_funcs(self):
195
196
        log = []
 
197
 
196
198
        def logging_cleanup(key, value):
197
199
            log.append((key, value))
198
200
        c = fifo_cache.FIFOCache(5, 4)
199
201
        c.add(1, 2, cleanup=logging_cleanup)
200
202
        c.add(2, 3, cleanup=logging_cleanup)
201
203
        c.add(3, 4, cleanup=logging_cleanup)
202
 
        c.add(4, 5, cleanup=None) # no cleanup for 4
203
 
        c[5] = 6 # no cleanup for 5
 
204
        c.add(4, 5, cleanup=None)  # no cleanup for 4
 
205
        c[5] = 6  # no cleanup for 5
204
206
        self.assertEqual([], log)
205
207
        # Adding another key should cleanup 1 & 2
206
208
        c.add(6, 7, cleanup=logging_cleanup)
224
226
 
225
227
    def test_cleanup_at_deconstruct(self):
226
228
        log = []
 
229
 
227
230
        def logging_cleanup(key, value):
228
231
            log.append((key, value))
229
232
        c = fifo_cache.FIFOCache()
230
233
        c.add(1, 2, cleanup=logging_cleanup)
231
234
        del c
232
 
        # TODO: We currently don't support calling the cleanup() funcs during
233
 
        #       __del__. We might want to consider implementing this.
234
 
        self.expectFailure("we don't call cleanups during __del__",
235
 
                           self.assertEqual, [(1, 2)], log)
236
 
        self.assertEqual([(1, 2)], log)
 
235
        # As a matter of design, bzr does not (can not) count on anything
 
236
        # being run from Python __del__ methods, because they may not run for
 
237
        # a long time, and because in cPython merely having them defined
 
238
        # interferes with garbage collection.
 
239
        self.assertEqual([], log)
237
240
 
238
241
 
239
242
class TestFIFOSizeCache(tests.TestCase):
246
249
        self.assertEqual('2', c[1])
247
250
        self.assertEqual('2', c.get(1))
248
251
        self.assertEqual('2', c.get(1, None))
249
 
        self.assertEqual([1], c.keys())
250
 
        self.assertEqual([1], list(c.iterkeys()))
251
 
        self.assertEqual([(1, '2')], c.items())
252
 
        self.assertEqual([(1, '2')], list(c.iteritems()))
253
 
        self.assertEqual(['2'], c.values())
254
 
        self.assertEqual(['2'], list(c.itervalues()))
 
252
        self.assertEqual([1], list(c))
 
253
        self.assertEqual({1}, viewkeys(c))
 
254
        self.assertEqual([(1, '2')], sorted(viewitems(c)))
 
255
        self.assertEqual(['2'], sorted(viewvalues(c)))
255
256
        self.assertEqual({1: '2'}, c)
256
 
        self.assertEqual(1024*1024, c.cache_size())
 
257
        self.assertEqual(1024 * 1024, c.cache_size())
257
258
 
258
259
    def test_missing(self):
259
260
        c = fifo_cache.FIFOSizeCache()
262
263
        self.assertEqual(0, len(c))
263
264
        self.assertEqual(None, c.get(1))
264
265
        self.assertEqual(None, c.get(1, None))
265
 
        self.assertEqual([], c.keys())
266
 
        self.assertEqual([], list(c.iterkeys()))
267
 
        self.assertEqual([], c.items())
268
 
        self.assertEqual([], list(c.iteritems()))
269
 
        self.assertEqual([], c.values())
270
 
        self.assertEqual([], list(c.itervalues()))
 
266
        self.assertEqual([], list(c))
 
267
        self.assertEqual(set(), viewkeys(c))
 
268
        self.assertEqual([], list(viewitems(c)))
 
269
        self.assertEqual([], list(viewvalues(c)))
271
270
        self.assertEqual({}, c)
272
271
 
273
272
    def test_add_maintains_fifo(self):
276
275
        c[2] = 'cde'
277
276
        c[3] = 'fghi'
278
277
        self.assertEqual({1: 'ab', 2: 'cde', 3: 'fghi'}, c)
279
 
        c[4] = 'jkl' # Collapse
 
278
        c[4] = 'jkl'  # Collapse
280
279
        self.assertEqual({3: 'fghi', 4: 'jkl'}, c)
281
280
        # Replacing an item will bump it to the end of the queue
282
281
        c[3] = 'mnop'
286
285
 
287
286
    def test_adding_large_key(self):
288
287
        c = fifo_cache.FIFOSizeCache(10, 8)
289
 
        c[1] = 'abcdefgh' # Adding a large key won't get cached at all
 
288
        c[1] = 'abcdefgh'  # Adding a large key won't get cached at all
290
289
        self.assertEqual({}, c)
291
290
        c[1] = 'abcdefg'
292
291
        self.assertEqual({1: 'abcdefg'}, c)