def __init__(self, filename=None, reactor=None):
if not filename:
filename = settings.FC_CHUTESTORAGE_SAVE_PATH
PDStorage.__init__(self, filename, reactor, settings.FC_CHUTESTORAGE_SAVE_TIMER)
# Has it been loaded?
if len(ChuteStorage.chuteList) == 0:
out.verbose("Loading chutes from disk: %s\n" % (filename))
self.loadFromDisk()
def test_storage():
"""
Test PDStorage class
"""
temp = tempfile.mkdtemp()
filename = os.path.join(temp, "storage")
storage = PDStorage(filename, MagicMock(), None)
# Constructor should have called this start function.
assert storage.repeater.start.called
# PDStorage needs to be subclassed; the base class always returns not
# saveable.
assert storage.attrSaveable() is False
storage = TestStorage(filename)
data = {"key": "value"}
with open(filename, "w") as output:
output.write("BAD CONTENTS")
# The first attempt to read it will fail and try to delete the file. We
# will cause the unlink to fail on the first try and let it succeed on the
# second try.
with patch("paradrop.lib.utils.pdos.unlink", side_effect=Exception("Boom!")):
storage.loadFromDisk()
assert os.path.exists(filename)
storage.loadFromDisk()
assert not os.path.exists(filename)
# The first write will fail because we have not provided data yet.
storage.saveToDisk()
assert not os.path.exists(filename)
# Now we will save some data and verify that we can reload it.
storage.setAttr(data)
# Cause the save to fail on the first try, then let it succeed.
with patch("paradrop.lib.utils.pdos.open", side_effect=Exception("Boom!")):
storage.saveToDisk()
assert not os.path.exists(filename)
storage.saveToDisk()
assert os.path.exists(filename)
storage.setAttr(None)
storage.loadFromDisk()
assert storage.getAttr() == data
# Clean up
pdos.remove(temp)
