def test_clean_seed_corpus(fs):
"""Test that seed corpus files are moved to root directory and deletes files
exceeding 1 MB limit."""
seed_corpus_dir = '/seeds'
fs.create_dir(seed_corpus_dir)
fs.create_file(os.path.join(seed_corpus_dir, 'a', 'abc'), contents='abc')
fs.create_file(os.path.join(seed_corpus_dir, 'def'), contents='def')
fs.create_file(os.path.join(seed_corpus_dir, 'efg'),
contents='a' * 1048577)
runner._clean_seed_corpus(seed_corpus_dir) # pylint: disable=protected-access
assert not os.path.exists(os.path.join(seed_corpus_dir, 'a', 'abc'))
assert not os.path.exists(os.path.join(seed_corpus_dir, 'def'))
assert not os.path.exists(os.path.join(seed_corpus_dir, 'efg'))
assert len(
os.listdir(seed_corpus_dir)) == 3 # Directory 'a' and two files.
with open(
os.path.join(
seed_corpus_dir,
'a9993e364706816aba3e25717850c26c9cd0d89d')) as file_handle:
assert file_handle.read() == 'abc'
with open(
os.path.join(
seed_corpus_dir,
'589c22335a381f122d129225f5c0ba3056ed5811')) as file_handle:
assert file_handle.read() == 'def'
def test_clean_seed_corpus_no_seeds(fs):
"""Test that seed corpus files are deleted if NO_SEEDS is set in the
environment to 'True'."""
seed_corpus_dir = '/seeds'
fs.create_dir(seed_corpus_dir)
seed_file = os.path.join(seed_corpus_dir, 'a')
fs.create_file(seed_file, contents='abc')
runner._clean_seed_corpus(seed_corpus_dir) # pylint: disable=protected-access
assert not os.path.exists(seed_file)
assert os.path.exists(seed_corpus_dir)
