machfs/test_all.py
2018-11-06 07:29:10 +08:00

74 lines
1.7 KiB
Python

from machfs import *
import os
import time
def test_upperlower():
h = Volume()
h['alpha'] = File()
assert h['alpha'] is h['ALPHA']
assert list(h.keys()) == ['alpha']
def test_roundtrip():
h = Volume()
f = File()
h['single file'] = f
f.data = f.rsrc = b'1234' * 4096
copies = [h.write(800*1024)]
for i in range(2):
h2 = Volume()
h2.read(copies[-1])
copies.append(h2.write(800*1024))
assert copies[0] == copies[1]
assert copies[1] == copies[2]
assert f.data in copies[-1]
def test_macos_mount():
h = Volume()
h.name = 'ElmoTest'
hf = File()
hf.data = b'12345' * 10
for i in reversed(range(100)):
last = 'testfile-%03d' % i
h[last] = hf
ser = h.write(10*1024*1024)
open('/tmp/SMALL.dmg','wb').write(ser)
os.system('hdiutil attach /tmp/SMALL.dmg')
n = 10
while 1:
n += 1
assert n < 200
time.sleep(0.1)
try:
os.stat('/Volumes/ElmoTest/testfile-000')
except:
pass
else:
break
recovered = open('/Volumes/ElmoTest/' + last,'rb').read()
os.system('umount /Volumes/ElmoTest')
assert recovered == hf.data
h2 = Volume()
h2.read(ser)
assert h2['testfile-000'].data == hf.data
# def test_extents_overflow():
# h = Volume()
# h.read(open('SourceForEmulator.dmg','rb').read())
# assert h['aa'].data == b'a' * 278528
def test_many_sizes():
sizes = [800*1024, 1024*1024]
while sizes[-1] < 4*1024*1024*1024:
sizes.append(sizes[-1] * 2)
# okay, we have heaps of sizes
v = Volume()
v.name = 'ImportantTestVol'
for s in sizes:
ser = v.write(s-512)
open('/tmp/SMALL-%X.dmg'%s, 'wb').write(ser)