machfs/machfs/main_test.py

50 lines
1.1 KiB
Python

from main import *
import os
import time
def test_roundtrip():
h = Volume()
f = File()
h[b'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]
def test_macos_mount():
h = Volume()
h.name = b'ElmoTest'
hf = File()
hf.data = b'12345' * 10
for i in reversed(range(100)):
last = b'testfile-%03d' % i
h[last] = hf
ser = h.write(10*1024*1024)
h2 = Volume()
h2.read(ser)
assert h2[b'testfile-000'].data == hf.data
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.decode('ascii'),'rb').read()
os.system('umount /Volumes/ElmoTest')
assert recovered == hf.data