diff options
Diffstat (limited to 'tests/test_file.py')
| -rw-r--r-- | tests/test_file.py | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/tests/test_file.py b/tests/test_file.py new file mode 100644 index 0000000..590e36f --- /dev/null +++ b/tests/test_file.py @@ -0,0 +1,47 @@ +# Copyright (C) 2014 Combox author(s). See AUTHORS. +# +# This file is part of Combox. +# +# Combox is free software: you can redistribute it and/or modify it +# under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Combox is distributed in the hope that it will be useful, but +# WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with Combox (see COPYING). If not, see +# <http://www.gnu.org/licenses/>. + +from os import path + +from file import (split_data, glue_data, write_file, + read_file, write_shards, read_shards) + +### Test to split, glue and create a copy of the image file from the +### glued image file. +f = path.abspath('tests/files/the-red-star.jpg') +f_content = read_file(f) +f_parts = split_data(f_content, 5) +f_content = glue_data(f_parts) +f_copy = path.abspath('tests/files/the-red-star-copy.jpg') +write_file(f_copy, f_content) + + +## read file and split it into N shards and write the shards to disk. +SHARDS = 5 +f = path.abspath('tests/files/the-red-star.jpg') +f_content = read_file(f) +f_shards = split_data(f_content, SHARDS) +f_path = path.dirname(f) +f_basename = path.basename(f) +write_shards(f_shards, f_path, f_basename) + +## read file shards, glue them together and write to disk. +f_shards = read_shards(f_path, f_basename) +f_content = glue_data(f_shards) +f_copy = path.abspath('tests/files/the-red-star-glued.jpg') +write_file(f_copy, f_content) |