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)