mirror of
https://github.com/robmcmullen/atrcopy.git
synced 2024-11-25 16:32:07 +00:00
171 lines
6.4 KiB
Python
171 lines
6.4 KiB
Python
from __future__ import print_function
|
|
from builtins import object
|
|
import numpy as np
|
|
|
|
from mock import *
|
|
|
|
from atrcopy import SegmentData, AtariDosDiskImage, Dos33DiskImage,InvalidBinaryFile
|
|
from atrcopy.errors import *
|
|
|
|
|
|
class BaseFilesystemModifyTest(object):
|
|
diskimage_type = None
|
|
sample_data = None
|
|
num_files_in_sample = 0
|
|
|
|
def setup(self):
|
|
rawdata = SegmentData(self.sample_data.copy())
|
|
self.image = self.diskimage_type(rawdata)
|
|
|
|
def check_entries(self, entries, prefix=b"TEST", save=None):
|
|
if type(prefix) is not bytes: prefix = prefix.encode("utf-8")
|
|
orig_num_files = len(self.image.files)
|
|
filenames = []
|
|
count = 1
|
|
for data in entries:
|
|
filename = b"%s%d.BIN" % (prefix, count)
|
|
self.image.write_file(filename, None, data)
|
|
assert len(self.image.files) == orig_num_files + count
|
|
data2 = np.fromstring(self.image.find_file(filename), dtype=np.uint8)
|
|
assert np.array_equal(data, data2[0:len(data)])
|
|
count += 1
|
|
|
|
# loop over them again to make sure data wasn't overwritten
|
|
count = 1
|
|
for data in entries:
|
|
filename = b"%s%d.BIN" % (prefix, count)
|
|
data2 = np.fromstring(self.image.find_file(filename), dtype=np.uint8)
|
|
assert np.array_equal(data, data2[0:len(data)])
|
|
count += 1
|
|
filenames.append(filename)
|
|
|
|
if save is not None:
|
|
self.image.save(save)
|
|
|
|
return filenames
|
|
|
|
def test_small(self):
|
|
assert len(self.image.files) == self.num_files_in_sample
|
|
|
|
data = np.asarray([0xff, 0xff, 0x00, 0x60, 0x01, 0x60, 1, 2], dtype=np.uint8)
|
|
self.image.write_file(b"TEST.XEX", None, data)
|
|
assert len(self.image.files) == self.num_files_in_sample + 1
|
|
|
|
data2 = np.fromstring(self.image.find_file(b"TEST.XEX"), dtype=np.uint8)
|
|
assert np.array_equal(data, data2[0:len(data)])
|
|
|
|
def test_50k(self):
|
|
assert len(self.image.files) == self.num_files_in_sample
|
|
|
|
data = np.arange(50*1024, dtype=np.uint8)
|
|
self.image.write_file(b"RAMP50K.BIN", None, data)
|
|
assert len(self.image.files) == self.num_files_in_sample + 1
|
|
|
|
data2 = self.image.find_file(b"RAMP50K.BIN")
|
|
assert data.tostring() == data2
|
|
|
|
def test_many_small(self):
|
|
entries = [
|
|
np.asarray([0xff, 0xff, 0x00, 0x60, 0x01, 0x60, 1, 2], dtype=np.uint8),
|
|
np.arange(1*1024, dtype=np.uint8),
|
|
np.arange(2*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(4*1024, dtype=np.uint8),
|
|
np.arange(5*1024, dtype=np.uint8),
|
|
np.arange(6*1024, dtype=np.uint8),
|
|
np.arange(7*1024, dtype=np.uint8),
|
|
np.arange(8*1024, dtype=np.uint8),
|
|
np.arange(9*1024, dtype=np.uint8),
|
|
np.arange(10*1024, dtype=np.uint8),
|
|
]
|
|
self.check_entries(entries, save=b"many_small.atr")
|
|
|
|
def test_big_failure(self):
|
|
assert len(self.image.files) == self.num_files_in_sample
|
|
|
|
data = np.arange(50*1024, dtype=np.uint8)
|
|
self.image.write_file(b"RAMP50K.BIN", None, data)
|
|
assert len(self.image.files) == self.num_files_in_sample + 1
|
|
with pytest.raises(NotEnoughSpaceOnDisk):
|
|
huge = np.arange(500*1024, dtype=np.uint8)
|
|
self.image.write_file(b"RAMP500K.BIN", None, huge)
|
|
assert len(self.image.files) == self.num_files_in_sample + 1
|
|
|
|
def test_delete(self):
|
|
entries1 = [
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(10*1024, dtype=np.uint8),
|
|
np.arange(10*1024, dtype=np.uint8),
|
|
]
|
|
entries2 = [
|
|
np.arange(10*1024, dtype=np.uint8),
|
|
np.arange(5*1024, dtype=np.uint8),
|
|
]
|
|
|
|
filenames = self.check_entries(entries1, "FIRST")
|
|
assert len(self.image.files) == self.num_files_in_sample + 11
|
|
self.image.delete_file(filenames[2])
|
|
self.image.delete_file(filenames[5])
|
|
self.image.delete_file(filenames[0])
|
|
self.image.delete_file(filenames[8])
|
|
assert len(self.image.files) == self.num_files_in_sample + 7
|
|
|
|
filename = self.check_entries(entries2, b"SECOND", save="test_delete.atr")
|
|
assert len(self.image.files) == self.num_files_in_sample + 9
|
|
|
|
def test_delete_all(self):
|
|
entries1 = [
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(3*1024, dtype=np.uint8),
|
|
np.arange(10*1024, dtype=np.uint8),
|
|
np.arange(11*1024, dtype=np.uint8),
|
|
np.arange(12*1024, dtype=np.uint8),
|
|
]
|
|
for dirent in self.image.files:
|
|
self.image.delete_file(dirent.filename)
|
|
assert len(self.image.files) == 0
|
|
|
|
class TestAtariDosSDImage(BaseFilesystemModifyTest):
|
|
diskimage_type = AtariDosDiskImage
|
|
sample_data = np.fromfile("../test_data/dos_sd_test1.atr", dtype=np.uint8)
|
|
num_files_in_sample = 5
|
|
|
|
class TestAtariDosEDImage(BaseFilesystemModifyTest):
|
|
diskimage_type = AtariDosDiskImage
|
|
sample_data = np.fromfile("../test_data/dos_ed_test1.atr", dtype=np.uint8)
|
|
num_files_in_sample = 5
|
|
|
|
class TestAtariDosDDImage(BaseFilesystemModifyTest):
|
|
diskimage_type = AtariDosDiskImage
|
|
sample_data = np.fromfile("../test_data/dos_dd_test1.atr", dtype=np.uint8)
|
|
num_files_in_sample = 5
|
|
|
|
class TestDos33Image(BaseFilesystemModifyTest):
|
|
diskimage_type = Dos33DiskImage
|
|
sample_data = np.fromfile("../test_data/dos33_master.dsk", dtype=np.uint8)
|
|
num_files_in_sample = 19
|
|
|
|
|
|
if __name__ == "__main__":
|
|
t = TestAtariDosSDImage()
|
|
for name in dir(t):
|
|
print(name)
|
|
if name.startswith("test_"):
|
|
t.setup()
|
|
getattr(t, name)()
|