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)