some bugfixes

test/test_files_bunkerfile.py

@@ -1,5 +1,6 @@
 import os
 import tarfile
+from io import BytesIO
 from bunker.files.bunkerfile import BunkeredFile
 
 def test_load_from_tar(tmpdir):
@@ -27,3 +28,12 @@ def test_load_from_tar(tmpdir):
 
 	assert a.read() == b"abcdefg"
 	assert b.read() == b"foobar"
+
+def test_from_BytesIO():
+	b = BytesIO(b"foobar")
+	f = BunkeredFile.from_BytesIO(b, "test")
+
+	assert len(f) == len(b"foobar")
+	assert f.read() == b"foobar"
+
+

test/test_files_tarfile.py

@@ -1,5 +1,6 @@
 import os
 import tarfile
+import pytest
 
 from bunker.files.tarfile import RewriteableTarFile
 from bunker.files.bunkerfile import BunkeredFile
@@ -21,3 +22,75 @@ def test_create(tmpdir):
 	f = tf.get_file("__bunker_main__")
 	
 	assert f.read() == b"foobar"
+
+def test_rewrite(tmpdir):
+	tmpdname = str(tmpdir)
+
+	tf = RewriteableTarFile.open(os.path.join(tmpdname, "test.bunker"))
+	f = BunkeredFile.empty("__bunker_main__")
+	tf.add_file(f)
+	tf.add_file(BunkeredFile.empty("test.tx"))
+
+	f = tf.get_file("__bunker_main__")
+	f.write(b"foobar")
+	
+	tf.close()
+	
+	assert f.closed == True
+
+	tf = RewriteableTarFile.open(os.path.join(tmpdname, "test.bunker"))
+	f = tf.get_file("__bunker_main__")
+	f.write(b"foobaz")
+	tf.close()
+	
+	tf = RewriteableTarFile.open(os.path.join(tmpdname, "test.bunker"))
+	f = tf.get_file("__bunker_main__")
+	assert f.read() == b"foobaz"
+
+def test_writeback(tmpdir):
+	tmpdname = str(tmpdir)
+
+	tf = RewriteableTarFile.open(os.path.join(tmpdname, "test.bunker"))
+	f = BunkeredFile.empty("__bunker_main__")
+	tf.add_file(f)
+	tf.add_file(BunkeredFile.empty("test.tx"))
+
+	f = tf.get_file("__bunker_main__")
+	f.write(b"foobar")
+	
+	tf.close()
+
+	tf = RewriteableTarFile.open(os.path.join(tmpdname, "test.bunker"))
+	f = tf.get_file("__bunker_main__")
+	f.write(b"foobaz")
+	f.close()
+	tf.close()
+	
+	tf = RewriteableTarFile.open(os.path.join(tmpdname, "test.bunker"))
+	f = tf.get_file("__bunker_main__")
+	assert f.read() == b"foobaz"
+
+
+
+def test_delete(tmpdir):
+	tmpdname = str(tmpdir)
+
+	tf = RewriteableTarFile.open(os.path.join(tmpdname, "test.bunker"))
+	f = BunkeredFile.empty("__bunker_main__")
+	tf.add_file(f)
+	tf.add_file(BunkeredFile.empty("test.tx"))
+
+	f = tf.get_file("__bunker_main__")
+	f.write(b"foobar")
+	
+	tf.close()
+
+	tf = RewriteableTarFile.open(os.path.join(tmpdname, "test.bunker"))
+	tf.delete_file("__bunker_main__")
+	tf.close()
+	
+	tf = RewriteableTarFile.open(os.path.join(tmpdname, "test.bunker"))
+
+	with pytest.raises(KeyError):
+		f = tf.get_file("__bunker_main__")
+