def test_default_is_zero(self):
flen = 1024
myfile = self.tmp_path("def_zero")
util.write_file(myfile, flen * b'\1', omode="wb")
block.wipe_file(myfile)
found = util.load_file(myfile, decode=False)
self.assertEqual(found, flen * b'\0')
def test_non_exist_dir_raises_file_not_found(self):
try:
p = self.tmp_path(os.path.sep.join(["enodir", "file"]))
block.wipe_file(p)
raise Exception("%s did not raise exception" % p)
except Exception as e:
if not util.is_file_not_found_exc(e):
raise Exception("exc was not file_not_found: %s" % e)
def test_non_exist_raises_file_not_found(self):
try:
p = self.tmp_path("enofile")
block.wipe_file(p)
raise Exception("%s did not raise exception" % p)
except Exception as e:
if not util.is_file_not_found_exc(e):
raise Exception("exc was not file_not_found: %s" % e)
def test_reader_fhandle(self):
srcfile = self.tmp_path("fhandle_src")
trgfile = self.tmp_path("fhandle_trg")
data = '\n'.join(["this is source file." for f in range(0, 10)] + [])
util.write_file(srcfile, data)
util.write_file(trgfile, 'a' * len(data))
with open(srcfile, "rb") as fp:
block.wipe_file(trgfile, reader=fp.read)
found = util.load_file(trgfile)
self.assertEqual(data, found)
def test_reader_used(self):
flen = 17
def reader(size):
return size * b'\1'
myfile = self.tmp_path("reader_used")
# populate with nulls
util.write_file(myfile, flen * b'\0', omode="wb")
block.wipe_file(myfile, reader=reader, buflen=flen)
found = util.load_file(myfile, decode=False)
self.assertEqual(found, flen * b'\1')
def test_reader_twice(self):
flen = 37
data = {'x': 20 * b'a' + 20 * b'b'}
expected = data['x'][0:flen]
def reader(size):
buf = data['x'][0:size]
data['x'] = data['x'][size:]
return buf
myfile = self.tmp_path("reader_twice")
util.write_file(myfile, flen * b'\xff', omode="wb")
block.wipe_file(myfile, reader=reader, buflen=20)
found = util.load_file(myfile, decode=False)
self.assertEqual(found, expected)
