diff options
| -rw-r--r-- | tests/config_test.py | 58 | ||||
| -rw-r--r-- | tests/crypto_test.py | 79 | ||||
| -rw-r--r-- | tests/file_test.py | 72 |
3 files changed, 209 insertions, 0 deletions
diff --git a/tests/config_test.py b/tests/config_test.py new file mode 100644 index 0000000..7bc50c2 --- /dev/null +++ b/tests/config_test.py @@ -0,0 +1,58 @@ +# 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/>. + +import yaml + +from nose.tools import * +from os import path, remove, rmdir + +from combox.config import (config_cb, get_secret, get_stdin) + + +CONFIG_DIR = path.join('tests', 'files', 'config') + +# sample config info. +CONFIG_INFO = [CONFIG_DIR, '2', + 'gdrive', 'gdrive/path', '1024', + 'dbox', 'dbox/path', '1024'] + +CONFIG_INFO_ITER = iter(CONFIG_INFO) + +def config_iter(dummy): + "Iterates through CONFIG_INFO" + return next(CONFIG_INFO_ITER) + +def test_config(): + "Tests the combox's config function." + + config_dir = CONFIG_DIR + pass_func = lambda: 'topsecret' + input_func = config_iter + + config_cb(config_dir, pass_func, input_func) + + # check if the config yaml file is valid + config_file = path.join(CONFIG_DIR, 'config.yaml') + try: + config = yaml.load(file(config_file, 'r')) + except yaml.YAMLError, exc: + raise AssertionError("Error in configuration file:", exc) + + # remove config dir + remove(config_file) + rmdir(CONFIG_DIR) diff --git a/tests/crypto_test.py b/tests/crypto_test.py new file mode 100644 index 0000000..aaa3bd2 --- /dev/null +++ b/tests/crypto_test.py @@ -0,0 +1,79 @@ +# 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 glob import glob +from nose.tools import * +from os import path, remove + +from combox.file import (split_data, glue_data, write_file, + read_file, write_shards, read_shards) +from combox.crypto import encrypt, decrypt, encrypt_shards, decrypt_shards + + +FILES_DIR = path.join('tests','files') +TEST_FILE = path.join(FILES_DIR,'the-red-star.jpg') +PASS = 'topsecret' + +def test_encryption(): + """ Read file, encrypt it to a cipher, write cipher to file, read + encrypted file, decrypt it, write decrypted data to file. +""" + + f = path.abspath(TEST_FILE) + f_content = read_file(f) + + # encrypt + f_cipher = encrypt(f_content, PASS) + # decrypt + f_content_decrypted = decrypt(f_cipher, PASS) + + assert f_content == f_content_decrypted + + +def test_split_encryption(): + """Read file, split it, encrypt shards, write encrypted shards to + file, read encrypted shards from file, decrypt the encrypted shards, + glue the shards together, write glued data to file. + """ + + f = path.abspath(TEST_FILE) + f_content = read_file(f) + f_shards = split_data(f_content, 5) + + # encrypt shards + ciphered_shards = encrypt_shards(f_shards, PASS) + + # write ciphered shards to disk + f_path = FILES_DIR + f_basename = "%s.ciphered" % path.basename(f) + write_shards(ciphered_shards, f_path, f_basename) + + # read ciphered shards from disk + ciphered_shards = read_shards(f_path, f_basename) + + # decrypt shards + f_parts = decrypt_shards(ciphered_shards, PASS) + # glue them shards together + f_content_glued = glue_data(f_parts) + + assert f_content == f_content_glued + + # remove ciphered shards from disk + c_shards = glob("%s.ciphered*" % TEST_FILE) + for shard in c_shards: + remove(shard) diff --git a/tests/file_test.py b/tests/file_test.py new file mode 100644 index 0000000..e418dfc --- /dev/null +++ b/tests/file_test.py @@ -0,0 +1,72 @@ +# 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 glob import glob +from nose.tools import * +from os import path, remove + +from combox.file import (split_data, glue_data, write_file, + read_file, write_shards, read_shards) + +FILES_DIR = path.join('tests','files') +TEST_FILE = path.join(FILES_DIR,'the-red-star.jpg') + +def test_split(): + """Test to split, glue and create a copy of the image file from the + glued image file. + + """ + + f = path.abspath(TEST_FILE) + f_content = read_file(f) + f_parts = split_data(f_content, 5) + f_content_glued = glue_data(f_parts) + + assert f_content == f_content_glued + #f_copy = path.abspath('tests/files/the-red-star-copy.jpg') + #write_file(f_copy, f_content) + + +def test_shards(): + """Split file into N shards, write them to disk, glue them together +and check if they're the same as the orginal file. + """ + + SHARDS = 5 + f = path.abspath(TEST_FILE) + f_content = read_file(f) + f_shards = split_data(f_content, SHARDS) + f_path = FILES_DIR + f_basename = path.basename(f) + write_shards(f_shards, f_path, f_basename) + + # check if the shards have been created. + for i in range(0,SHARDS): + shard = "%s.shard%s" % (TEST_FILE, i) + assert path.isfile(shard) + + + f_shards = read_shards(f_path, f_basename) + f_content_glued = glue_data(f_shards) + + assert f_content == f_content_glued + + # remove shards from disk. + shards = glob("%s.shard*" % TEST_FILE) + for shard in shards: + remove(shard) |