# -*- coding: utf-8 -*- # # Copyright (C) 2016 Dr. Robert C. Green II. # # 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 # . import os import time import yaml from filecmp import cmp from glob import glob from os import path, remove from shutil import copyfile from threading import Lock from nose.tools import * from watchdog.observers import Observer from combox.config import get_nodedirs from combox.crypto import (decrypt_and_glue, split_and_encrypt, encrypt_shards) from combox.events import ComboxDirMonitor, NodeDirMonitor from combox.file import (relative_path, purge_dir, hash_file, read_file, write_file, move_shards, rm_shards, mk_nodedir, rm_nodedir, move_nodedir, node_paths, rm_path, split_data, write_shards) from combox.silo import ComboxSilo from tests.utils import (get_config, shardedp, dirp, renamedp, path_deletedp, rm_nodedirs, rm_configdir, purge_nodedirs, purge, not_shardedp) class TestEvents(object): """ Class that tests the events.py module. """ @classmethod def setup_class(self): """Set things up.""" self.silo_lock = Lock() self.monitor_lock = Lock() self.config = get_config() self.silo = ComboxSilo(self.config, self.silo_lock) self.FILES_DIR = self.config['combox_dir'] self.NODE_DIR = get_nodedirs(self.config)[0] self.TEST_FILE = path.join(self.FILES_DIR, 'thgttg-21st.png') self.lorem = path.join(self.FILES_DIR, 'lorem.txt') self.ipsum = path.join(self.FILES_DIR, 'ipsum.txt') self.lorem_moved = path.join(self.FILES_DIR, 'lorem.moved.txt') self.lorem_ipsum = path.join(self.FILES_DIR, 'lorem.ipsum.txt') # files that have to be removed at the end of this test class # should be appended to this list. self.purge_list = [] def test_CDM(self): """ Tests the ComboxDirMonitor class. """ event_handler = ComboxDirMonitor(self.config, self.silo_lock, self.monitor_lock) observer = Observer() observer.schedule(event_handler, self.FILES_DIR, recursive=True) observer.start() # Test - new file addition self.TEST_FILE_COPY_0 = "%s.mutant" % self.TEST_FILE copyfile(self.TEST_FILE, self.TEST_FILE_COPY_0) ## wait for ComboxDirMonitor to split and scatter the file in the ## node directories. time.sleep(2) ## check if the shards were created. shardedp(self.TEST_FILE_COPY_0) ## check if the new file's info is in silo silo = ComboxSilo(self.config, self.silo_lock) assert silo.exists(self.TEST_FILE_COPY_0) # Test - tmp file creation. # combox must ignore it. self.TEST_TMP_FILE = "%s~" % self.TEST_FILE copyfile(self.TEST_FILE, self.TEST_TMP_FILE) # wait for a second. time.sleep(2) ## confirm that shards were not created. not_shardedp(self.TEST_TMP_FILE) ## confirm that it did not get registered in the silo. assert not silo.exists(self.TEST_TMP_FILE) # purge it later. self.purge_list.append(self.TEST_TMP_FILE) # Test - File deletion. remove(self.TEST_FILE_COPY_0) time.sleep(1) path_deletedp(self.TEST_FILE_COPY_0) ## check if the new file's info is removed from silo silo = ComboxSilo(self.config, self.silo_lock) assert not silo.exists(self.TEST_FILE_COPY_0) # Test - directory creation self.TEST_DIR_0 = path.join(self.FILES_DIR, 'foo') os.mkdir(self.TEST_DIR_0) time.sleep(2) ## check if TEST_DIR_0 is created under node directories. dirp(self.TEST_DIR_0) self.TEST_DIR_1 = path.join(self.TEST_DIR_0, 'bar') os.mkdir(self.TEST_DIR_1) time.sleep(2) ## check if TEST_DIR_1 is created under node directories. dirp(self.TEST_DIR_1) # Test - new file in a nested directory self.TEST_FILE_COPY_1 = path.join(self.TEST_DIR_1, path.basename(self.TEST_FILE)) copyfile(self.TEST_FILE, self.TEST_FILE_COPY_1) time.sleep(2) shardedp(self.TEST_FILE_COPY_1) # Test - dir rename self.TEST_DIR_1_NEW = path.join(path.dirname(self.TEST_DIR_1), 'snafu') self.TEST_FILE_COPY_1_NEW = path.join(self.TEST_DIR_1_NEW, path.basename(self.TEST_FILE)) os.rename(self.TEST_DIR_1, self.TEST_DIR_1_NEW) time.sleep(3) renamedp(self.TEST_DIR_1, self.TEST_DIR_1_NEW) renamedp(self.TEST_FILE_COPY_1, self.TEST_FILE_COPY_1_NEW) ## check if the new file's info is updated in silo silo = ComboxSilo(self.config, self.silo_lock) assert not silo.exists(self.TEST_FILE_COPY_1) assert silo.exists(self.TEST_FILE_COPY_1_NEW) # Test directory & file deletion purge_dir(self.TEST_DIR_0) # remove the directory itself. os.rmdir(self.TEST_DIR_0) time.sleep(2) path_deletedp(self.TEST_FILE_COPY_1_NEW) path_deletedp(self.TEST_DIR_1, True) path_deletedp(self.TEST_DIR_0, True) # Test - file modification self.lorem_file = path.join(self.FILES_DIR, 'lorem.txt') self.lorem_file_copy = "%s.copy" % self.lorem_file # this will shard lorem.txt.copy in the node directories. copyfile(self.lorem_file, self.lorem_file_copy) time.sleep(2) shardedp(self.lorem_file_copy) ## check if the lorem_file_copy's info is stored in silo silo = ComboxSilo(self.config, self.silo_lock) lorem_file_copy_hash = silo.db.get(self.lorem_file_copy) self.ipsum_file = path.join(self.FILES_DIR, 'ipsum.txt') ipsum_content = read_file(self.ipsum_file) lorem_copy_content = read_file(self.lorem_file_copy) lorem_copy_content = "%s\n%s" % (lorem_copy_content, ipsum_content) # write lorem's new content to lorem_file_copy write_file(self.lorem_file_copy, lorem_copy_content) time.sleep(2) ## check if the lorem_file_copy's info is updated in silo silo = ComboxSilo(self.config, self.silo_lock) assert lorem_file_copy_hash != silo.db.get(self.lorem_file_copy) # decrypt_and_glue will decrypt the file shards, glues them and # writes it to the respective file decrypt_and_glue(self.lorem_file_copy, self.config) time.sleep(2) lorem_content_from_disk = read_file(self.lorem_file_copy) assert lorem_copy_content == lorem_content_from_disk # remove lorem_file_copy and confirm that its shards are deleted # in the node directories. remove(self.lorem_file_copy) time.sleep(2) path_deletedp(self.lorem_file_copy) ## check if the lorem_file_copy's info is deleted from silo silo = ComboxSilo(self.config, self.silo_lock) assert not silo.exists(self.lorem_file_copy) observer.stop() observer.join() def test_CDM_housekeep(self): """ComboxDirMonitor's housekeep method test.""" # test file deletion and addition os.rename(self.lorem, self.lorem_moved) cdm = ComboxDirMonitor(self.config, self.silo_lock, self.monitor_lock) cdm.housekeep() silo = ComboxSilo(self.config, self.silo_lock) assert not silo.exists(self.lorem) assert silo.exists(self.lorem_moved) shardedp(self.lorem_moved) ##1 os.rename(self.lorem_moved, self.lorem) rm_shards(self.lorem_moved, self.config) silo.remove(self.lorem_moved) # test file modification copyfile(self.lorem, self.lorem_ipsum) assert path.exists(self.lorem_ipsum) cdm = ComboxDirMonitor(self.config, self.silo_lock, self.monitor_lock) cdm.housekeep() silo = ComboxSilo(self.config, self.silo_lock) assert silo.exists(self.lorem_ipsum) ipsum_content = read_file(self.ipsum) lorem_ipsum_content = read_file(self.lorem_ipsum) lorem_ipsum_content = "%s\n%s" % (lorem_ipsum_content, ipsum_content) write_file(self.lorem_ipsum, lorem_ipsum_content) cdm.housekeep() silo = ComboxSilo(self.config, self.silo_lock) assert not silo.stale(self.lorem_ipsum) def test_NDM_numnodes(self): """Tests whether the NodeDirMonitor's num_nodes variable has the right value. """ nmonitor = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) assert_equal(2, nmonitor.num_nodes) def test_NDM_oncreated(self): """Testing on_created method in NodeDirMonitor""" nodes = get_nodedirs(self.config) num_nodes = len(get_nodedirs(self.config)) nmonitors = [] observers = [] # create an observer for each node directory and make it # monitor them. for node in nodes: nmonitor = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) observer = Observer() observer.schedule(nmonitor, node, recursive=True) observer.start() nmonitors.append(nmonitor) observers.append(observer) # Test - new file addition, when shard is created in node_dirs self.TEST_FILE_MUTANT = "%s.mutant" % self.TEST_FILE fmutant_content = read_file(self.TEST_FILE) split_and_encrypt(self.TEST_FILE_MUTANT, self.config, fmutant_content) ## wait for NodeDirMonitor to reconstruct the shards and put ## it in combox directory time.sleep(2) assert fmutant_content == read_file(self.TEST_FILE_MUTANT) ## check if the new file's info is in silo assert self.silo.exists(self.TEST_FILE_MUTANT) assert_equal(None, self.silo.node_get('file_created', self.TEST_FILE_MUTANT)) self.purge_list.append(self.TEST_FILE_MUTANT) # Test - directory creation self.FOO_DIR = path.join(self.FILES_DIR, 'foo') mk_nodedir(self.FOO_DIR, self.config) time.sleep(2) ## check if FOO_DIR is created under the combox directory assert path.isdir(self.FOO_DIR) assert_equal(None, self.silo.node_get('file_created', self.FOO_DIR)) self.BAR_DIR = path.join(self.FOO_DIR, 'bar') mk_nodedir(self.BAR_DIR, self.config) time.sleep(2) ## check if BAR_DIR is created under the combox directory. assert path.isdir(self.BAR_DIR) assert_equal(None, self.silo.node_get('file_created', self.BAR_DIR)) self.purge_list.append(self.FOO_DIR) # stop the zarking observers. for i in range(num_nodes): observers[i].stop() observers[i].join() def test_NDM_onmodified(self): """Testing on_modified method in NodeDirMonitor""" nodes = get_nodedirs(self.config) num_nodes = len(get_nodedirs(self.config)) nmonitors = [] observers = [] # create an observer for each node directory and make it # monitor them. for node in nodes: nmonitor = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) observer = Observer() observer.schedule(nmonitor, node, recursive=True) observer.start() nmonitors.append(nmonitor) observers.append(observer) # Test - shard modification self.lorem_file = path.join(self.FILES_DIR, 'lorem.txt') lorem_content = read_file(self.lorem_file) self.lorem_file_copy = "%s.copy" % self.lorem_file copyfile(self.lorem_file, self.lorem_file_copy) split_and_encrypt(self.lorem_file_copy, self.config, lorem_content) time.sleep(2) self.silo.update(self.lorem_file_copy) shardedp(self.lorem_file_copy) self.silo.reload() lorem_file_copy_hash = self.silo.db.get(self.lorem_file_copy) self.ipsum_file = path.join(self.FILES_DIR, 'ipsum.txt') ipsum_content = read_file(self.ipsum_file) lorem_copy_content = "%s\n%s" % (lorem_content, ipsum_content) split_and_encrypt(self.lorem_file_copy, self.config, lorem_copy_content) time.sleep(2) assert lorem_copy_content == read_file(self.lorem_file_copy) ## check if the lorem_file_copy's info is updated in silo self.silo.reload() assert lorem_file_copy_hash != self.silo.db.get(self.lorem_file_copy) assert_equal(None, self.silo.node_get('file_modified', self.lorem_file_copy)) self.purge_list.append(self.lorem_file_copy) # stop the zarking observers. for i in range(num_nodes): observers[i].stop() observers[i].join() def test_NDM_ondeleted(self): """Testing on_delete method in NodeDirMonitor""" nodes = get_nodedirs(self.config) num_nodes = len(get_nodedirs(self.config)) nmonitors = [] observers = [] # create an observer for each node directory and make it # monitor them. for node in nodes: nmonitor = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) observer = Observer() observer.schedule(nmonitor, node, recursive=True) observer.start() nmonitors.append(nmonitor) observers.append(observer) BAR_DIR = path.join(self.FILES_DIR, 'bar') mk_nodedir(BAR_DIR, self.config) # wait for the `bar' directory to be created inside combox # directory. time.sleep(2) # Test - directory deletion inside node directory. rm_nodedir(BAR_DIR, self.config) time.sleep(2) assert not path.exists(BAR_DIR) the_guide = path.join(self.FILES_DIR, 'the.guide') split_and_encrypt(the_guide, self.config, read_file(self.TEST_FILE)) time.sleep(2) assert path.exists(the_guide) # Test - Shard deletion. rm_shards(the_guide, self.config) time.sleep(5) assert not path.exists(the_guide) ## check if the new file's info is removed from silo silo = ComboxSilo(self.config, self.silo_lock) assert not silo.exists(the_guide) self.purge_list.append(BAR_DIR) self.purge_list.append(the_guide) # stop the zarking observers. for i in range(num_nodes): observers[i].stop() observers[i].join() def test_GoogleDrive_file_modify(self): """Simulates Google Drive client's file modification behavior and checks if combox is interpreting it properly. """ nodes = get_nodedirs(self.config) num_nodes = len(get_nodedirs(self.config)) nmonitors = [] observers = [] # create an observer for each node directory and make it # monitor them. for node in nodes: nmonitor = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) observer = Observer() observer.schedule(nmonitor, node, recursive=True) observer.start() nmonitors.append(nmonitor) observers.append(observer) # Test - shard modification lorem_content = read_file(self.lorem) self.lorem_copy = "%s.copy" % self.lorem copyfile(self.lorem, self.lorem_copy) split_and_encrypt(self.lorem_copy, self.config, lorem_content) self.silo.update(self.lorem_copy) shardedp(self.lorem_copy) lorem_copy_hash = self.silo.db.get(self.lorem_copy) ipsum_content = read_file(self.ipsum) lorem_copy_content = "%s\n%s" % (lorem_content, ipsum_content) time.sleep(2) # Modify shards in the first n-1 node directories in the usual # way. For the nth node directory simulate Google Drive # official client's way of modifiying the shard. rel_path = relative_path(self.lorem_copy, self.config) # no. of shards = no. of nodes. SHARDS = len(self.config['nodes_info'].keys()) f_shards = split_data(lorem_copy_content, SHARDS) # encrypt shards ciphered_shards = encrypt_shards(f_shards, self.config['topsecret']) # write ciphered shards to disk f_basename = rel_path # gets the list of node' directories. nodes = get_nodedirs(self.config) last_node_index = len(nodes) - 1 nodes_subset = nodes[:last_node_index] last_node = nodes[last_node_index] # write n-1 shards to the first n-1 node directories write_shards(ciphered_shards, nodes_subset, f_basename) # now for the nth shard, simulate Google Drive's client # behavior. last_shard_path = "%s.shard%d" % (path.join(last_node, f_basename), last_node_index) # remove the shard first rm_path(last_shard_path) # write the latest version of the shard write_file(last_shard_path, ciphered_shards[last_node_index]) time.sleep(3) self.silo.reload() assert lorem_copy_content == read_file(self.lorem_copy) ## check if the lorem_copy's info is updated in silo assert lorem_copy_hash != self.silo.db.get(self.lorem_copy) assert_equal(None, self.silo.node_get('file_modified', self.lorem_copy)) self.purge_list.append(self.lorem_copy) # stop the zarking observers. for i in range(num_nodes): observers[i].stop() observers[i].join() def untest_NDM(self): """ Tests the NodeDirMonitor class. """ event_handler = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) observer = Observer() observer.schedule(event_handler, self.NODE_DIR, recursive=True) observer.start() #### # Test - new file addition, when shard is created in node_dirs self.TEST_FILE_MUTANT = "%s.mutant" % self.TEST_FILE fmutant_content = read_file(self.TEST_FILE) split_and_encrypt(self.TEST_FILE_MUTANT, self.config, fmutant_content) ## wait for NodeDirMonitor to reconstruct the shards and put ## it in combox directory time.sleep(1) assert fmutant_content == read_file(self.TEST_FILE_MUTANT) ## check if the new file's info is in silo silo = ComboxSilo(self.config, self.silo_lock) assert silo.exists(self.TEST_FILE_MUTANT) #### # Test - Shard deletion. rm_shards(self.TEST_FILE_MUTANT, self.config) time.sleep(1) assert not path.exists(self.TEST_FILE_MUTANT) ## check if the new file's info is removed from silo silo = ComboxSilo(self.config, self.silo_lock) assert not silo.exists(self.TEST_FILE_MUTANT) # Test - directory deletion inside node directory rm_nodedir(self.BAR_DIR, self.config) time.sleep(1) assert not path.exists(self.BAR_DIR) # Test - shard modification self.lorem_file = path.join(self.FILES_DIR, 'lorem.txt') lorem_content = read_file(self.lorem_file) self.lorem_file_copy = "%s.copy" % self.lorem_file copyfile(self.lorem_file, self.lorem_file_copy) split_and_encrypt(self.lorem_file_copy, self.config, lorem_content) silo = ComboxSilo(self.config, self.silo_lock) silo.update(self.lorem_file_copy) shardedp(self.lorem_file_copy) silo = ComboxSilo(self.config, self.silo_lock) lorem_file_copy_hash = silo.db.get(self.lorem_file_copy) self.ipsum_file = path.join(self.FILES_DIR, 'ipsum.txt') ipsum_content = read_file(self.ipsum_file) lorem_copy_content = "%s\n%s" % (lorem_content, ipsum_content) split_and_encrypt(self.lorem_file_copy, self.config, lorem_copy_content) time.sleep(1) ## check if the lorem_file_copy's info is updated in silo silo = ComboxSilo(self.config, self.silo_lock) assert lorem_copy_content == read_file(self.lorem_file_copy) assert lorem_file_copy_hash != silo.db.get(self.lorem_file_copy) self.purge_list.append(self.lorem_file_copy) observer.stop() observer.join() def test_NDM_onmoved(self): """Testing on_moved method in NodeDirMonitor""" nodes = get_nodedirs(self.config) num_nodes = len(get_nodedirs(self.config)) nmonitors = [] observers = [] # create an observer for each node directory and make it # monitor them. for node in nodes: nmonitor = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) observer = Observer() observer.schedule(nmonitor, node, recursive=True) observer.start() nmonitors.append(nmonitor) observers.append(observer) # event_handler = NodeDirMonitor(self.config, self.silo_lock, # self.monitor_lock) # observer = Observer() # observer.schedule(event_handler, self.NODE_DIR, recursive=True) # observer.start() self.testf = "%s.onm" % self.TEST_FILE copyfile(self.TEST_FILE, self.testf) silo = ComboxSilo(self.config, self.silo_lock) silo.update(self.testf) split_and_encrypt(self.testf, self.config) time.sleep(1) self.testf_moved = "%s.onm.moved" % self.TEST_FILE # test file move/rename move_shards(self.testf, self.testf_moved, self.config) time.sleep(1) assert path.isfile(self.testf_moved) assert silo.exists(self.testf_moved) assert not silo.exists(self.testf) # test directory move/rename dirt = path.join(self.FILES_DIR, "fom") os.mkdir(dirt) mk_nodedir(dirt, self.config) dirt_lorem = path.join(dirt, "lorem.txt") copyfile(self.lorem, dirt_lorem) split_and_encrypt(dirt_lorem, self.config) time.sleep(1) silo.update(dirt_lorem) dirt_m = path.join(self.FILES_DIR, "mof") dirt_m_lorem = path.join(dirt_m, "lorem.txt") move_nodedir(dirt, dirt_m, self.config) time.sleep(1) assert path.isdir(dirt_m) assert path.isfile(dirt_m_lorem) assert not path.isdir(dirt) assert not path.isdir(dirt_lorem) assert silo.exists(dirt_m_lorem) assert not silo.exists(dirt_lorem) self.purge_list.append(self.testf_moved) self.purge_list.append(dirt_m) # stop the zarking observers. for i in range(num_nodes): observers[i].stop() observers[i].join() def test_NDM_housekeep_delete(self): """Testing NodeDirMonitor's housekeep method's delete functionality.""" # files for testing deletion. testf1 = path.join(self.FILES_DIR, 'hitchhikers.png') testf2 = path.join(self.FILES_DIR, 'lorem.housekeep') copyfile(self.TEST_FILE, testf1) copyfile(self.lorem, testf2) split_and_encrypt(testf2, self.config) testf2_shard0 = node_paths(testf2, self.config, True)[0] remove(testf2_shard0) silo = ComboxSilo(self.config, self.silo_lock) silo.update(testf1) silo.update(testf2) ndm = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) ndm.housekeep() assert not path.exists(testf1) assert path.exists(testf2) self.purge_list.append(testf1) self.purge_list.append(testf2) def test_NDM_housekeep_create(self): """Testing NodeDirMonitor's housekeep method's create functionality.""" # test shard creation hmutant = "%s.mutant" % self.TEST_FILE hmutant_content = read_file(self.TEST_FILE) split_and_encrypt(hmutant, self.config, hmutant_content) lorem_c = "%s.c" % self.lorem split_and_encrypt(lorem_c, self.config, read_file(self.lorem)) # delete lorem_c's shard0 remove(node_paths(lorem_c, self.config, True)[0]) ndm = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) ndm.housekeep() assert path.exists(hmutant) assert hmutant_content == read_file(hmutant) assert not path.exists(lorem_c) assert_equal(len(get_nodedirs(self.config))-1, self.silo.node_get('file_created', lorem_c)) self.purge_list.append(hmutant) self.purge_list.append(lorem_c) def untest_NDM_housekeep(self): """Testing NodeDirMonitor's housekeep method.""" # files for testing deletion. testf1 = path.join(self.FILES_DIR, 'hitchhikers.png') testf2 = path.join(self.FILES_DIR, 'lorem.housekeep') copyfile(self.TEST_FILE, testf1) copyfile(self.lorem, testf2) silo = ComboxSilo(self.config, self.silo_lock) silo.update(testf1) silo.update(testf2) ndm = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) ndm.housekeep() assert not path.exists(testf1) assert not path.exists(testf2) self.purge_list.append(testf1) self.purge_list.append(testf2) # test shard creation hmutant = "%s.mutant" % self.TEST_FILE hmutant_content = read_file(self.TEST_FILE) split_and_encrypt(hmutant, self.config, hmutant_content) ndm = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) ndm.housekeep() assert path.exists(hmutant) assert hmutant_content == read_file(hmutant) self.purge_list.append(hmutant) # test shard modification lcopy = "%s.copy" % self.lorem copyfile(self.lorem, lcopy) lcopy_content = read_file(lcopy) split_and_encrypt(lcopy, self.config, lcopy_content) silo = ComboxSilo(self.config, self.silo_lock) silo.update(lcopy) shardedp(lcopy) silo = ComboxSilo(self.config, self.silo_lock) lcopy_hash = silo.db.get(lcopy) ipsum_content = read_file(self.ipsum) lcopy_content = "%s\n%s" % (lcopy_content, ipsum_content) split_and_encrypt(lcopy, self.config, lcopy_content) ndm = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) ndm.housekeep() ## check if the lorem_file_copy's info is updated in silo silo = ComboxSilo(self.config, self.silo_lock) assert lcopy_content == read_file(lcopy) assert hash_file(lcopy, lcopy_content) == silo.db.get(lcopy) self.purge_list.append(lcopy) def test_NDM_shardp(self): """Testing shardp method in NodeDirMonitor class""" shard = 'some.shard0' not_shard = 'some.extension' ndm = NodeDirMonitor(self.config, self.silo_lock, self.monitor_lock) assert_equal(True, ndm.shardp(shard)) assert_equal(False, ndm.shardp(not_shard)) def teardown(self): """Cleans up things after each test in this class""" purge_nodedirs(self.config) self.silo.db.deldb() purge(self.purge_list) @classmethod def teardown_class(self): """Purge the mess created by this test""" rm_shards(self.TEST_FILE, self.config) rm_shards(self.lorem, self.config) rm_shards(self.ipsum, self.config) rm_path(self.lorem_ipsum) rm_nodedirs(self.config) rm_configdir()