def test_put_with_lease(self):
blob = LocalFileBlob(os.path.join(TEST_FOLDER, "foobar.blob"))
test_input = (1, 2, 3)
blob.delete(silent=True)
blob.put(test_input, lease_period=0.01)
blob.lease(0.01)
self.assertEqual(blob.get(), test_input)
def test_put_error(self):
blob = LocalFileBlob(os.path.join(TEST_FOLDER, "foobar"))
with mock.patch("os.rename", side_effect=throw(Exception)):
self.assertRaises(Exception, lambda: blob.put([1, 2, 3]))
def test_put_without_lease(self):
blob = LocalFileBlob(os.path.join(TEST_FOLDER, "foobar.blob"))
input = (1, 2, 3)
blob.delete(silent=True)
blob.put(input)
self.assertEqual(blob.get(), input)