def test_delete_from_storage():
storage = Storage('/tmp/thebot.storage')
storage.clear()
storage['blah'] = 'minor'
del storage['blah']
eq_([], sorted(storage.keys()))
def test_delete_from_storage():
storage = Storage('/tmp/thebot.storage')
storage.clear()
storage['blah'] = 'minor'
del storage['blah']
eq_([], sorted(storage.keys()))
def test_simple_storage():
storage = Storage('/tmp/thebot.storage')
storage.clear()
eq_([], storage.keys())
storage['blah'] = 'minor'
storage['one'] = {'some': 'dict'}
eq_(['blah', 'one'], sorted(storage.keys()))
eq_('minor', storage['blah'])
def test_simple_storage():
storage = Storage('/tmp/thebot.storage')
storage.clear()
eq_([], storage.keys())
storage['blah'] = 'minor'
storage['one'] = {'some': 'dict'}
eq_(['blah', 'one'], sorted(storage.keys()))
eq_('minor', storage['blah'])
def test_storage_restores_bot_attribute():
bot = Bot(adapters=[TestAdapter], plugins=[TestPlugin])
storage = Storage('/tmp/thebot.storage', global_objects=dict(bot=bot))
storage.clear()
original = Request('blah')
original.bot = bot
storage['request'] = original
restored = storage['request']
eq_(restored.bot, original.bot)
def test_storage_restores_bot_attribute():
bot = Bot(adapters=[TestAdapter], plugins=[TestPlugin])
storage = Storage('/tmp/thebot.storage', global_objects=dict(bot=bot))
storage.clear()
original = Request('blah')
original.bot = bot
storage['request'] = original
restored = storage['request']
eq_(restored.bot, original.bot)
def test_storage_nesting():
storage = Storage('/tmp/thebot.storage')
storage.clear()
first = storage.with_prefix('first:')
second = storage.with_prefix('second:')
eq_([], storage.keys())
first['blah'] = 'minor'
second['one'] = {'some': 'dict'}
eq_(['first:blah', 'second:one'], sorted(storage.keys()))
eq_(['first:blah'], sorted(first.keys()))
eq_(['second:one'], sorted(second.keys()))
eq_('minor', first['blah'])
assert_raises(KeyError, lambda: second['blah'])
first.clear()
eq_(['second:one'], sorted(storage.keys()))
def test_storage_nesting():
storage = Storage('/tmp/thebot.storage')
storage.clear()
first = storage.with_prefix('first:')
second = storage.with_prefix('second:')
eq_([], storage.keys())
first['blah'] = 'minor'
second['one'] = {'some': 'dict'}
eq_(['first:blah', 'second:one'], sorted(storage.keys()))
eq_(['first:blah'], sorted(first.keys()))
eq_(['second:one'], sorted(second.keys()))
eq_('minor', first['blah'])
assert_raises(KeyError, lambda: second['blah'])
first.clear()
eq_(['second:one'], sorted(storage.keys()))
