import szip import os const ( test_out_zip = 'v_test_zip.zip' test_dir_zip = 'v_test_dir_zip.zip' test_path = 'zip files' test_path2 = '.zip folder' 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') fname1 = 'file_1.txt' fpath1 = os.join_path(test_path, fname1) fname2 = 'file_2.txt' fpath2 = os.join_path(test_path, fname2) fname3 = '.New Text Document.txt' fpath3 = os.join_path(test_path2, fname3) 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) ) fn cleanup() { os.chdir(os.temp_dir()) or {} os.rmdir_all(test_path) or {} os.rmdir_all(test_path2) or {} os.rmdir_all(test_path3) or {} os.rm(test_out_zip) or {} os.rm(test_dir_zip) or {} } fn testsuite_begin() { cleanup() } fn testsuite_end() { cleanup() } fn test_szip_create_temp_files() { os.mkdir(test_path)? os.mkdir(test_path2)? os.write_file(fpath1, 'file one')? os.write_file(fpath2, 'file two')? os.write_file(fpath3, 'file three')? assert os.exists(fpath1) assert os.exists(fpath2) assert os.exists(fpath3) } fn test_zipping_files() { 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)) szip.zip_files(files, test_out_zip)? assert os.exists(test_out_zip) os.rm(fpath1)? os.rm(fpath2)? os.rm(fpath3)? } fn test_extract_zipped_files() { szip.extract_zip_to_dir(test_out_zip, test_path)? szip.extract_zip_to_dir(test_out_zip, test_path2)? assert os.exists(fpath1) assert os.exists(fpath2) assert os.exists(fpath3) assert (os.read_file(fpath1)?) == 'file one' assert (os.read_file(fpath2)?) == 'file two' assert (os.read_file(fpath3)?) == 'file three' cleanup() } 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' } cleanup() os.mkdir(test_path)? os.mkdir(test_path2)? os.write_file(fpath3, 'file three')? 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() } 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' }