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)
