1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
import io
import os
import pathlib
from typing import Optional, Union
import pytest
import py7zr
import py7zr.io
from py7zr import SevenZipFile
from . import p7zip_test
testdata_path = pathlib.Path(os.path.dirname(__file__)).joinpath("data")
os.umask(0o022)
class TestArchiveWriter(py7zr.io.Py7zIO):
def __init__(self, fname, target: SevenZipFile):
self.fname = fname
self.target = target
self.buffer = io.BytesIO()
def write(self, s: Union[bytes, bytearray]) -> int:
return self.buffer.write(s)
def read(self, size: Optional[int] = None) -> bytes:
return self.buffer.read(size)
def seek(self, offset: int, whence: int = 0) -> int:
return self.buffer.seek(offset, whence)
def flush(self) -> None:
pass
def close(self) -> None:
self.target.writef(self.buffer, self.fname)
def size(self) -> int:
return self.buffer.getbuffer().nbytes
class TestWriterFactory(py7zr.io.WriterFactory):
def __init__(self, target: SevenZipFile):
self.target = target
def create(self, filename: str) -> py7zr.io.Py7zIO:
return TestArchiveWriter(filename, self.target)
@pytest.mark.files
def test_read_write_new(tmp_path):
with py7zr.SevenZipFile(tmp_path.joinpath("target.7z"), "w") as target:
with py7zr.SevenZipFile(testdata_path.joinpath("mblock_1.7z").open(mode="rb")) as source:
source.extractall(factory=TestWriterFactory(target))
p7zip_test(tmp_path / "target.7z")
|
