class StateSaveBufferTestCase(TestCase):
@setup
def setup_buffer(self):
self.buffer_size = 5
self.buffer = StateSaveBuffer(self.buffer_size)
def test_save(self):
assert self.buffer.save(1, 2)
assert not self.buffer.save(1, 3)
assert not self.buffer.save(1, 4)
assert not self.buffer.save(1, 5)
assert not self.buffer.save(1, 6)
assert self.buffer.save(1, 7)
assert_equal(self.buffer.buffer[1], 7)
def test__iter__(self):
self.buffer.save(1, 2)
self.buffer.save(2, 3)
items = list(self.buffer)
assert not self.buffer.buffer
assert_equal(items, [(1, 2), (2, 3)])
class TestStateSaveBuffer(TestCase):
@setup
def setup_buffer(self):
self.buffer_size = 5
self.buffer = StateSaveBuffer(self.buffer_size)
def test_save(self):
assert self.buffer.save(1, 2)
assert not self.buffer.save(1, 3)
assert not self.buffer.save(1, 4)
assert not self.buffer.save(1, 5)
assert not self.buffer.save(1, 6)
assert self.buffer.save(1, 7)
assert_equal(self.buffer.buffer[1], 7)
def test__iter__(self):
self.buffer.save(1, 2)
self.buffer.save(2, 3)
items = list(self.buffer)
assert not self.buffer.buffer
assert_equal(items, [(1, 2), (2, 3)])
def setup_manager(self):
self.store = mock.Mock()
self.store.build_key.side_effect = lambda t, i: '%s%s' % (t, i)
self.buffer = StateSaveBuffer(1)
self.manager = PersistentStateManager(self.store, self.buffer)
def setup_buffer(self):
self.buffer_size = 5
self.buffer = StateSaveBuffer(self.buffer_size)
def setup_buffer(self):
self.buffer_size = 5
self.buffer = StateSaveBuffer(self.buffer_size)