class TestLimitSizeFS(TestWrapFS):
_dont_retest = TestWrapFS._dont_retest + ("test_big_file",)
def setUp(self):
super(TestLimitSizeFS,self).setUp()
self.fs = LimitSizeFS(self.fs,1024*1024*2) # 2MB limit
def tearDown(self):
self.fs.removedir("/",force=True)
self.assertEquals(self.fs.cur_size,0)
super(TestLimitSizeFS,self).tearDown()
self.fs.close()
def test_storage_error(self):
total_written = 0
for i in xrange(1024*2):
try:
total_written += 1030
self.fs.setcontents("file %i" % i, b("C")*1030)
except StorageSpaceError:
self.assertTrue(total_written > 1024*1024*2)
self.assertTrue(total_written < 1024*1024*2 + 1030)
break
else:
self.assertTrue(False,"StorageSpaceError not raised")
