def test_copy_resources_to_bucket(tmp_path): """Tests that copy_resources_to_bucket copies the correct resources.""" # Do this so that Ctrl-C doesn't pollute the repo. cwd = os.getcwd() os.chdir(tmp_path) config_dir = 'config' config = { 'experiment_filestore': 'gs://gsutil-bucket', 'experiment': 'experiment' } try: with mock.patch('common.filestore_utils.rsync') as mocked_rsync: with mock.patch('common.filestore_utils.cp') as mocked_cp: run_experiment.copy_resources_to_bucket(config_dir, config) mocked_cp.assert_called_once_with( 'src.tar.gz', 'gs://gsutil-bucket/experiment/input/', parallel=True) mocked_rsync.assert_called_once_with( 'config', 'gs://gsutil-bucket/experiment/input/config', parallel=True) finally: os.chdir(cwd)
def test_copy_resources_to_bucket(): """Tests that copy_resources_to_bucket copies the correct resources.""" config_dir = 'config' config = { 'cloud_experiment_bucket': 'gs://gsutil-bucket', 'experiment': 'experiment' } with mock.patch('common.gsutil.rsync') as mocked_rsync: with mock.patch('common.gsutil.cp') as mocked_cp: run_experiment.copy_resources_to_bucket(config_dir, config) mocked_cp.assert_called_once_with( 'src.tar.gz', 'gs://gsutil-bucket/experiment/input/', parallel=True) mocked_rsync.assert_called_once_with( 'config', 'gs://gsutil-bucket/experiment/input/config', parallel=True)
def test_copy_resources_to_bucket(): """Tests that copy_resources_to_bucket copies the correct resources.""" config_dir = 'config' config = { 'cloud_experiment_bucket': 'gs://gsutil-bucket', 'experiment': 'experiment' } with mock.patch('common.gsutil.rsync') as mocked_rsync: run_experiment.copy_resources_to_bucket(config_dir, config) mocked_rsync.assert_any_call( BENCHMARKS_DIR, 'gs://gsutil-bucket/experiment/input/src', options=[ '-x', ('^\\.git/|^\\.pytype/|^\\.venv/|^.*\\.pyc$|^__pycache__/|' '.*~$|\\.pytest_cache/|.*/test_data/|' '^third_party/oss-fuzz/out/|^docs/') ]) mocked_rsync.assert_any_call( 'config', 'gs://gsutil-bucket/experiment/input/config')
def test_copy_resources_to_bucket(tmp_path): """Tests that copy_resources_to_bucket copies the correct resources.""" # Do this so that Ctrl-C doesn't pollute the repo. cwd = os.getcwd() os.chdir(tmp_path) config_dir = 'config' config = { 'experiment_filestore': 'gs://gsutil-bucket', 'experiment': 'experiment', 'benchmarks': ['libxslt_xpath'], 'oss_fuzz_corpus': True, } try: with mock.patch('common.filestore_utils.cp') as mocked_filestore_cp: with mock.patch( 'common.filestore_utils.rsync') as mocked_filestore_rsync: with mock.patch('common.gsutil.cp') as mocked_gsutil_cp: run_experiment.copy_resources_to_bucket(config_dir, config) mocked_filestore_cp.assert_called_once_with( 'src.tar.gz', 'gs://gsutil-bucket/experiment/input/', parallel=True) mocked_filestore_rsync.assert_called_once_with( 'config', 'gs://gsutil-bucket/experiment/input/config', parallel=True) mocked_gsutil_cp.assert_called_once_with( 'gs://libxslt-backup.clusterfuzz-external.appspot.com/' 'corpus/libFuzzer/libxslt_xpath/public.zip', 'gs://gsutil-bucket/experiment/oss_fuzz_corpora/' 'libxslt_xpath.zip', expect_zero=False, parallel=True) finally: os.chdir(cwd)