2020-09-06 16:04:27 +03:00
|
|
|
import szip
|
|
|
|
import os
|
|
|
|
|
2021-04-02 08:26:14 +03:00
|
|
|
const (
|
|
|
|
test_out_zip = 'v_test_zip.zip'
|
2022-05-11 16:48:41 +03:00
|
|
|
test_dir_zip = 'v_test_dir_zip.zip'
|
2021-04-02 08:26:14 +03:00
|
|
|
test_path = 'zip files'
|
2022-05-09 07:45:36 +03:00
|
|
|
test_path2 = '.zip folder'
|
2022-05-11 16:48:41 +03:00
|
|
|
test_path3 = 'test zip folder'
|
|
|
|
test_path3_1 = os.join_path(test_path3, '1', '1')
|
|
|
|
test_path3_2 = os.join_path(test_path3, '2', '1')
|
|
|
|
test_path3_3 = os.join_path(test_path3, '3', '1')
|
|
|
|
test_path3_4 = os.join_path(test_path3, '4', '1')
|
2021-07-25 21:29:51 +03:00
|
|
|
fname1 = 'file_1.txt'
|
|
|
|
fpath1 = os.join_path(test_path, fname1)
|
|
|
|
fname2 = 'file_2.txt'
|
|
|
|
fpath2 = os.join_path(test_path, fname2)
|
2022-05-09 07:45:36 +03:00
|
|
|
fname3 = '.New Text Document.txt'
|
|
|
|
fpath3 = os.join_path(test_path2, fname3)
|
2022-05-11 16:48:41 +03:00
|
|
|
fname4 = 'file.txt'
|
|
|
|
fpath4 = os.join_path(test_path3_1, fname4)
|
|
|
|
fpath5 = os.join_path(test_path3_2, fname4)
|
|
|
|
fpath6 = os.join_path(test_path3_4, fname4)
|
2021-04-02 08:26:14 +03:00
|
|
|
)
|
2021-04-02 02:59:01 +03:00
|
|
|
|
2021-11-28 16:30:26 +03:00
|
|
|
fn cleanup() {
|
2021-08-28 09:35:39 +03:00
|
|
|
os.chdir(os.temp_dir()) or {}
|
2021-04-02 08:26:14 +03:00
|
|
|
os.rmdir_all(test_path) or {}
|
2022-05-09 07:45:36 +03:00
|
|
|
os.rmdir_all(test_path2) or {}
|
2022-05-11 16:48:41 +03:00
|
|
|
os.rmdir_all(test_path3) or {}
|
2021-11-28 16:30:26 +03:00
|
|
|
os.rm(test_out_zip) or {}
|
2022-05-11 16:48:41 +03:00
|
|
|
os.rm(test_dir_zip) or {}
|
2021-11-28 16:30:26 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
fn testsuite_begin() ? {
|
|
|
|
cleanup()
|
|
|
|
}
|
|
|
|
|
|
|
|
fn testsuite_end() ? {
|
|
|
|
cleanup()
|
|
|
|
}
|
|
|
|
|
|
|
|
fn test_szip_create_temp_files() ? {
|
2021-04-02 08:26:14 +03:00
|
|
|
os.mkdir(test_path) ?
|
2022-05-09 07:45:36 +03:00
|
|
|
os.mkdir(test_path2) ?
|
2021-04-02 08:26:14 +03:00
|
|
|
os.write_file(fpath1, 'file one') ?
|
|
|
|
os.write_file(fpath2, 'file two') ?
|
2022-05-09 07:45:36 +03:00
|
|
|
os.write_file(fpath3, 'file three') ?
|
2021-04-02 08:26:14 +03:00
|
|
|
assert os.exists(fpath1)
|
|
|
|
assert os.exists(fpath2)
|
2022-05-09 07:45:36 +03:00
|
|
|
assert os.exists(fpath3)
|
2021-04-02 08:26:14 +03:00
|
|
|
}
|
2021-04-02 02:59:01 +03:00
|
|
|
|
2021-04-02 08:26:14 +03:00
|
|
|
fn test_zipping_files() ? {
|
2022-05-09 07:45:36 +03:00
|
|
|
mut files := (os.ls(test_path) ?).map(os.join_path(test_path, it))
|
|
|
|
files << (os.ls(test_path2) ?).map(os.join_path(test_path2, it))
|
2021-04-02 08:26:14 +03:00
|
|
|
szip.zip_files(files, test_out_zip) ?
|
2021-04-02 02:59:01 +03:00
|
|
|
assert os.exists(test_out_zip)
|
2022-05-09 07:45:36 +03:00
|
|
|
os.rm(fpath1) ?
|
|
|
|
os.rm(fpath2) ?
|
|
|
|
os.rm(fpath3) ?
|
2021-04-02 08:26:14 +03:00
|
|
|
}
|
2021-04-02 02:59:01 +03:00
|
|
|
|
2021-04-02 08:26:14 +03:00
|
|
|
fn test_extract_zipped_files() ? {
|
|
|
|
szip.extract_zip_to_dir(test_out_zip, test_path) ?
|
2022-05-09 07:45:36 +03:00
|
|
|
szip.extract_zip_to_dir(test_out_zip, test_path2) ?
|
2021-04-02 08:26:14 +03:00
|
|
|
assert os.exists(fpath1)
|
|
|
|
assert os.exists(fpath2)
|
2022-05-09 07:45:36 +03:00
|
|
|
assert os.exists(fpath3)
|
2021-04-02 08:26:14 +03:00
|
|
|
assert (os.read_file(fpath1) ?) == 'file one'
|
|
|
|
assert (os.read_file(fpath2) ?) == 'file two'
|
2022-05-09 07:45:36 +03:00
|
|
|
assert (os.read_file(fpath3) ?) == 'file three'
|
|
|
|
cleanup()
|
2021-04-02 08:26:14 +03:00
|
|
|
}
|
2021-07-25 21:29:51 +03:00
|
|
|
|
|
|
|
fn test_reading_zipping_files() ? {
|
|
|
|
n_files := 2
|
|
|
|
mut file_name_list := []string{}
|
|
|
|
for i in 0 .. n_files {
|
|
|
|
file_name_list << 'file_${i:02}.txt'
|
|
|
|
}
|
|
|
|
|
2022-05-09 07:45:36 +03:00
|
|
|
cleanup()
|
2021-07-25 21:29:51 +03:00
|
|
|
os.mkdir(test_path) ?
|
2022-05-09 07:45:36 +03:00
|
|
|
os.mkdir(test_path2) ?
|
|
|
|
os.write_file(fpath3, 'file three') ?
|
2021-07-25 21:29:51 +03:00
|
|
|
for c, f_name in file_name_list {
|
|
|
|
tmp_path := os.join_path(test_path, f_name)
|
|
|
|
os.write_file(tmp_path, 'file ${c:02}') ?
|
|
|
|
assert os.exists(tmp_path)
|
|
|
|
}
|
|
|
|
files := (os.ls(test_path) ?).map(os.join_path(test_path, it))
|
|
|
|
|
|
|
|
szip.zip_files(files, test_out_zip) ?
|
|
|
|
assert os.exists(test_out_zip)
|
|
|
|
|
|
|
|
mut zp := szip.open(test_out_zip, szip.CompressionLevel.no_compression, szip.OpenMode.read_only) ?
|
|
|
|
n_entries := zp.total() ?
|
|
|
|
assert n_entries == n_files
|
|
|
|
|
|
|
|
unsafe {
|
|
|
|
data_len := 'file XX'.len
|
|
|
|
buf_size := 32
|
|
|
|
buf := malloc(data_len * 2)
|
|
|
|
|
|
|
|
for _ in 0 .. n_files {
|
|
|
|
zp.open_entry_by_index(0) ?
|
|
|
|
name := zp.name()
|
|
|
|
assert name in file_name_list
|
|
|
|
|
|
|
|
zp.read_entry_buf(buf, buf_size) ?
|
|
|
|
buf[data_len] = 0
|
|
|
|
tmp_str := tos(buf, data_len)
|
|
|
|
|
|
|
|
assert tmp_str[0..4] == 'file'
|
|
|
|
assert tmp_str[5..7] == name[5..7]
|
|
|
|
|
|
|
|
zp.close_entry()
|
|
|
|
}
|
|
|
|
|
|
|
|
free(buf)
|
|
|
|
}
|
|
|
|
zp.close()
|
|
|
|
}
|
2022-05-11 16:48:41 +03:00
|
|
|
|
|
|
|
fn test_zip_folder() ? {
|
|
|
|
cleanup()
|
|
|
|
os.mkdir_all(test_path3_1) ?
|
|
|
|
os.mkdir_all(test_path3_2) ?
|
|
|
|
os.mkdir_all(test_path3_3) ?
|
|
|
|
os.mkdir_all(test_path3_4) ?
|
|
|
|
os.write_file(fpath4, '4') ?
|
|
|
|
os.write_file(fpath5, '5') ?
|
|
|
|
os.write_file(fpath6, '6') ?
|
|
|
|
|
|
|
|
szip.zip_folder(test_path3, test_dir_zip) ?
|
|
|
|
assert os.exists(test_dir_zip)
|
|
|
|
|
|
|
|
os.rmdir_all(test_path3) ?
|
|
|
|
os.mkdir_all(test_path3) ?
|
|
|
|
szip.extract_zip_to_dir(test_dir_zip, test_path3) ?
|
|
|
|
assert os.exists(test_path3_1)
|
|
|
|
assert os.exists(test_path3_2)
|
|
|
|
assert os.exists(test_path3_3) // This is the empty dir
|
|
|
|
assert os.exists(test_path3_4)
|
|
|
|
assert (os.read_file(fpath4) ?) == '4'
|
|
|
|
assert (os.read_file(fpath5) ?) == '5'
|
|
|
|
assert (os.read_file(fpath6) ?) == '6'
|
|
|
|
}
|
|
|
|
|
|
|
|
fn test_zip_folder_omit_empty_directories() ? {
|
|
|
|
cleanup()
|
|
|
|
os.mkdir_all(test_path3_1) ?
|
|
|
|
os.mkdir_all(test_path3_2) ?
|
|
|
|
os.mkdir_all(test_path3_3) ?
|
|
|
|
os.mkdir_all(test_path3_4) ?
|
|
|
|
os.write_file(fpath4, '4') ?
|
|
|
|
os.write_file(fpath5, '5') ?
|
|
|
|
os.write_file(fpath6, '6') ?
|
|
|
|
|
|
|
|
szip.zip_folder(test_path3, test_dir_zip, omit_empty_folders: true) ?
|
|
|
|
assert os.exists(test_dir_zip)
|
|
|
|
|
|
|
|
os.rmdir_all(test_path3) ?
|
|
|
|
os.mkdir_all(test_path3) ?
|
|
|
|
szip.extract_zip_to_dir(test_dir_zip, test_path3) ?
|
|
|
|
assert os.exists(test_path3_1)
|
|
|
|
assert os.exists(test_path3_2)
|
|
|
|
assert !os.exists(test_path3_3) // This is the empty dir, should be omitted with `omit_empty_folders`
|
|
|
|
assert os.exists(test_path3_4)
|
|
|
|
assert (os.read_file(fpath4) ?) == '4'
|
|
|
|
assert (os.read_file(fpath5) ?) == '5'
|
|
|
|
assert (os.read_file(fpath6) ?) == '6'
|
|
|
|
}
|