class Test(unittest.TestCase):
def setUp(self):
campaign_id = 'testcampaign'
self.origindir = tempfile.mkdtemp()
self.localdir = tempfile.mkdtemp()
self.outputdir = tempfile.mkdtemp()
# create some files
self.file_count = 5
self.files = []
for i in range(self.file_count):
(fd, f) = tempfile.mkstemp(dir=self.origindir)
os.write(fd, 'abacab%d' % i)
os.close(fd)
self.files.append(f)
# create a set
self.sfs = SeedfileSet(campaign_id, self.origindir, self.localdir,
self.outputdir)
def tearDown(self):
for f in self.files:
os.remove(f)
self.assertFalse(os.path.exists(f))
for d in (self.origindir, self.localdir, self.outputdir):
shutil.rmtree(d)
self.assertFalse(os.path.exists(d))
def test_pickle(self):
import pickle
self.assertTrue(hasattr(self.sfs, 'things'))
# no files added yet
self.assertEqual(0, len(self.sfs.things))
# add the files
self.sfs._setup()
# confirm that the files are there
self.assertEqual(self.file_count, len(self.sfs.things))
unpickled = pickle.loads(pickle.dumps(self.sfs))
pprint(unpickled.__dict__)
def test_set_directories(self):
self.assertEqual(self.sfs.originpath, self.origindir)
self.assertEqual(self.sfs.localpath, self.localdir)
self.assertEqual(self.sfs.outputpath, self.outputdir)
self.assertEqual(None, self.sfs.origindir)
self.assertEqual(None, self.sfs.localdir)
self.assertEqual(None, self.sfs.outputdir)
self.sfs._set_directories()
self.assertEqual(Directory, self.sfs.origindir.__class__)
self.assertEqual(Directory, self.sfs.localdir.__class__)
self.assertEqual(Directory, self.sfs.outputdir.__class__)
# make sure the file(s) we created in setUp are in origindir
self.assertEqual(self.file_count, len(self.sfs.origindir.files))
def test_copy_files_to_localdir(self):
# mock the things
self.sfs.origindir = [1, 2, 3, 4, 5]
copied = []
self.sfs.copy_file_from_origin = lambda x: copied.append(x)
# do the test
self.sfs._copy_files_to_localdir()
self.assertEqual(self.sfs.origindir, copied)
def test_copy_file_from_origin(self):
pass
def test_add_local_files_to_set(self):
pass
def test_add_file(self):
self.assertNotEqual(0, len(self.files))
self.assertEqual(0, len(self.sfs.things))
self.sfs.add_file(*self.files)
self.assertEqual(5, len(self.sfs.things))
for thing in self.sfs.things.itervalues():
self.assertEqual(SeedFile, thing.__class__)
def test_init(self):
self.assertEqual(self.outputdir, self.sfs.seedfile_output_base_dir)
self.assertEqual(0, len(self.sfs.things))
def test_getstate_is_pickle_friendly(self):
# getstate should return a pickleable object
import pickle
state = self.sfs.__getstate__()
try:
pickle.dumps(state)
except Exception, e:
self.fail('Failed to pickle state: %s' % e)
class Test(unittest.TestCase):
def setUp(self):
campaign_id = 'testcampaign'
self.origindir = tempfile.mkdtemp()
self.localdir = tempfile.mkdtemp()
self.outputdir = tempfile.mkdtemp()
# create some files
self.file_count = 5
self.files = []
for i in range(self.file_count):
(fd, f) = tempfile.mkstemp(dir=self.origindir)
os.write(fd, 'abacab%d' % i)
os.close(fd)
self.files.append(f)
# create a set
self.sfs = SeedfileSet(campaign_id, self.origindir, self.localdir, self.outputdir)
def tearDown(self):
for f in self.files:
os.remove(f)
self.assertFalse(os.path.exists(f))
for d in (self.origindir, self.localdir, self.outputdir):
shutil.rmtree(d)
self.assertFalse(os.path.exists(d))
def test_pickle(self):
import pickle
self.assertTrue(hasattr(self.sfs, 'things'))
# no files added yet
self.assertEqual(0, len(self.sfs.things))
# add the files
self.sfs._setup()
# confirm that the files are there
self.assertEqual(self.file_count, len(self.sfs.things))
unpickled = pickle.loads(pickle.dumps(self.sfs))
pprint(unpickled.__dict__)
def test_set_directories(self):
self.assertEqual(self.sfs.originpath, self.origindir)
self.assertEqual(self.sfs.localpath, self.localdir)
self.assertEqual(self.sfs.outputpath, self.outputdir)
self.assertEqual(None, self.sfs.origindir)
self.assertEqual(None, self.sfs.localdir)
self.assertEqual(None, self.sfs.outputdir)
self.sfs._set_directories()
self.assertEqual(Directory, self.sfs.origindir.__class__)
self.assertEqual(Directory, self.sfs.localdir.__class__)
self.assertEqual(Directory, self.sfs.outputdir.__class__)
# make sure the file(s) we created in setUp are in origindir
self.assertEqual(self.file_count, len(self.sfs.origindir.files))
def test_copy_files_to_localdir(self):
# mock the things
self.sfs.origindir = [1, 2, 3, 4, 5]
copied = []
self.sfs.copy_file_from_origin = lambda x: copied.append(x)
# do the test
self.sfs._copy_files_to_localdir()
self.assertEqual(self.sfs.origindir, copied)
def test_copy_file_from_origin(self):
pass
def test_add_local_files_to_set(self):
pass
def test_add_file(self):
self.assertNotEqual(0, len(self.files))
self.assertEqual(0, len(self.sfs.things))
self.sfs.add_file(*self.files)
self.assertEqual(5, len(self.sfs.things))
for thing in self.sfs.things.itervalues():
self.assertEqual(SeedFile, thing.__class__)
def test_init(self):
self.assertEqual(self.outputdir, self.sfs.seedfile_output_base_dir)
self.assertEqual(0, len(self.sfs.things))
def test_getstate_is_pickle_friendly(self):
# getstate should return a pickleable object
import pickle
state = self.sfs.__getstate__()
try:
pickle.dumps(state)
except Exception, e:
self.fail('Failed to pickle state: %s' % e)
