//! project tests use std::{ collections::{HashMap, HashSet}, io, path::{Path, PathBuf}, str::FromStr, }; use ethers_solc::{ cache::{SolFilesCache, SOLIDITY_FILES_CACHE_FILENAME}, project_util::*, remappings::Remapping, ConfigurableArtifacts, ExtraOutputValues, Graph, Project, ProjectCompileOutput, ProjectPathsConfig, }; use pretty_assertions::assert_eq; #[allow(unused)] fn init_tracing() { tracing_subscriber::fmt() .with_env_filter(tracing_subscriber::EnvFilter::from_default_env()) .init(); } #[test] fn can_get_versioned_linkrefs() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/test-versioned-linkrefs"); let paths = ProjectPathsConfig::builder() .sources(root.join("src")) .lib(root.join("lib")) .build() .unwrap(); let project = Project::builder().paths(paths).ephemeral().no_artifacts().build().unwrap(); let compiled = project.compile().unwrap(); assert!(!compiled.has_compiler_errors()); } #[test] fn can_compile_hardhat_sample() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/hardhat-sample"); let paths = ProjectPathsConfig::builder() .sources(root.join("contracts")) .lib(root.join("node_modules")); let project = TempProject::::new(paths).unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.find("Greeter").is_some()); assert!(compiled.find("console").is_some()); assert!(!compiled.has_compiler_errors()); // nothing to compile let compiled = project.compile().unwrap(); assert!(compiled.find("Greeter").is_some()); assert!(compiled.find("console").is_some()); assert!(compiled.is_unchanged()); // delete artifacts std::fs::remove_dir_all(&project.paths().artifacts).unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.find("Greeter").is_some()); assert!(compiled.find("console").is_some()); assert!(!compiled.is_unchanged()); } #[test] fn can_compile_dapp_sample() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/dapp-sample"); let paths = ProjectPathsConfig::builder().sources(root.join("src")).lib(root.join("lib")); let project = TempProject::::new(paths).unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(!compiled.has_compiler_errors()); // nothing to compile let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(compiled.is_unchanged()); let cache = SolFilesCache::read(project.cache_path()).unwrap(); // delete artifacts std::fs::remove_dir_all(&project.paths().artifacts).unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(!compiled.is_unchanged()); let updated_cache = SolFilesCache::read(project.cache_path()).unwrap(); assert_eq!(cache, updated_cache); } #[test] fn can_compile_yul_sample() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/yul-sample"); let paths = ProjectPathsConfig::builder().sources(root); let project = TempProject::::new(paths).unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(compiled.find("SimpleStore").is_some()); assert!(!compiled.has_compiler_errors()); // nothing to compile let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(compiled.find("SimpleStore").is_some()); assert!(compiled.is_unchanged()); let cache = SolFilesCache::read(project.cache_path()).unwrap(); // delete artifacts std::fs::remove_dir_all(&project.paths().artifacts).unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(compiled.find("SimpleStore").is_some()); assert!(!compiled.is_unchanged()); let updated_cache = SolFilesCache::read(project.cache_path()).unwrap(); assert_eq!(cache, updated_cache); } #[test] fn can_compile_configured() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/dapp-sample"); let paths = ProjectPathsConfig::builder().sources(root.join("src")).lib(root.join("lib")); let handler = ConfigurableArtifacts { additional_values: ExtraOutputValues { metadata: true, ir: true, ir_optimized: true, ..Default::default() }, ..Default::default() }; let settings = handler.settings(); let project = TempProject::with_artifacts(paths, handler).unwrap().with_settings(settings); let compiled = project.compile().unwrap(); let artifact = compiled.find("Dapp").unwrap(); assert!(artifact.metadata.is_some()); assert!(artifact.ir.is_some()); assert!(artifact.ir_optimized.is_some()); } #[test] fn can_compile_dapp_detect_changes_in_libs() { let mut project = TempProject::::dapptools().unwrap(); let remapping = project.paths().libraries[0].join("remapping"); project .paths_mut() .remappings .push(Remapping::from_str(&format!("remapping={}/", remapping.display())).unwrap()); let src = project .add_source( "Foo", r#" pragma solidity ^0.8.10; import "remapping/Bar.sol"; contract Foo {} "#, ) .unwrap(); let lib = project .add_lib( "remapping/Bar", r#" pragma solidity ^0.8.10; contract Bar {} "#, ) .unwrap(); let graph = Graph::resolve(project.paths()).unwrap(); assert_eq!(graph.files().len(), 2); assert_eq!(graph.files().clone(), HashMap::from([(src, 0), (lib, 1),])); let compiled = project.compile().unwrap(); assert!(compiled.find("Foo").is_some()); assert!(compiled.find("Bar").is_some()); assert!(!compiled.has_compiler_errors()); // nothing to compile let compiled = project.compile().unwrap(); assert!(compiled.find("Foo").is_some()); assert!(compiled.is_unchanged()); let cache = SolFilesCache::read(&project.paths().cache).unwrap(); assert_eq!(cache.files.len(), 2); // overwrite lib project .add_lib( "remapping/Bar", r#" pragma solidity ^0.8.10; // changed lib contract Bar {} "#, ) .unwrap(); let graph = Graph::resolve(project.paths()).unwrap(); assert_eq!(graph.files().len(), 2); let compiled = project.compile().unwrap(); assert!(compiled.find("Foo").is_some()); assert!(compiled.find("Bar").is_some()); // ensure change is detected assert!(!compiled.is_unchanged()); } #[test] fn can_compile_dapp_detect_changes_in_sources() { let project = TempProject::::dapptools().unwrap(); let src = project .add_source( "DssSpell.t", r#" pragma solidity ^0.8.10; import "./DssSpell.t.base.sol"; contract DssSpellTest is DssSpellTestBase { } "#, ) .unwrap(); let base = project .add_source( "DssSpell.t.base", r#" pragma solidity ^0.8.10; contract DssSpellTestBase { address deployed_spell; function setUp() public { deployed_spell = address(0xA867399B43aF7790aC800f2fF3Fa7387dc52Ec5E); } } "#, ) .unwrap(); let graph = Graph::resolve(project.paths()).unwrap(); assert_eq!(graph.files().len(), 2); assert_eq!(graph.files().clone(), HashMap::from([(base, 0), (src, 1),])); assert_eq!(graph.imported_nodes(1).to_vec(), vec![0]); let compiled = project.compile().unwrap(); assert!(!compiled.has_compiler_errors()); assert!(compiled.find("DssSpellTest").is_some()); assert!(compiled.find("DssSpellTestBase").is_some()); // nothing to compile let compiled = project.compile().unwrap(); assert!(compiled.is_unchanged()); assert!(compiled.find("DssSpellTest").is_some()); assert!(compiled.find("DssSpellTestBase").is_some()); let cache = SolFilesCache::read(&project.paths().cache).unwrap(); assert_eq!(cache.files.len(), 2); let mut artifacts = compiled.into_artifacts().collect::>(); // overwrite import let _ = project .add_source( "DssSpell.t.base", r#" pragma solidity ^0.8.10; contract DssSpellTestBase { address deployed_spell; function setUp() public { deployed_spell = address(0); } } "#, ) .unwrap(); let graph = Graph::resolve(project.paths()).unwrap(); assert_eq!(graph.files().len(), 2); let compiled = project.compile().unwrap(); assert!(compiled.find("DssSpellTest").is_some()); assert!(compiled.find("DssSpellTestBase").is_some()); // ensure change is detected assert!(!compiled.is_unchanged()); // and all recompiled artifacts are different for (p, artifact) in compiled.into_artifacts() { let other = artifacts.remove(&p).unwrap(); assert_ne!(artifact, other); } } #[test] fn can_compile_dapp_sample_with_cache() { let tmp_dir = tempfile::tempdir().unwrap(); let root = tmp_dir.path(); let cache = root.join("cache").join(SOLIDITY_FILES_CACHE_FILENAME); let artifacts = root.join("out"); let manifest_dir = PathBuf::from(env!("CARGO_MANIFEST_DIR")); let orig_root = manifest_dir.join("test-data/dapp-sample"); let cache_testdata_dir = manifest_dir.join("test-data/cache-sample/"); copy_dir_all(orig_root, &tmp_dir).unwrap(); let paths = ProjectPathsConfig::builder() .cache(cache) .sources(root.join("src")) .artifacts(artifacts) .lib(root.join("lib")) .root(root) .build() .unwrap(); // first compile let project = Project::builder().paths(paths).build().unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(!compiled.has_compiler_errors()); // cache is used when nothing to compile let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(compiled.is_unchanged()); // deleted artifacts cause recompile even with cache std::fs::remove_dir_all(&project.artifacts_path()).unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(!compiled.is_unchanged()); // new file is compiled even with partial cache std::fs::copy(cache_testdata_dir.join("NewContract.sol"), root.join("src/NewContract.sol")) .unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.find("Dapp").is_some()); assert!(compiled.find("NewContract").is_some()); assert!(!compiled.is_unchanged()); assert_eq!( compiled.into_artifacts().map(|(artifact_id, _)| artifact_id.name).collect::>(), HashSet::from([ "Dapp".to_string(), "DappTest".to_string(), "DSTest".to_string(), "NewContract".to_string(), ]) ); // old cached artifact is not taken from the cache std::fs::copy(cache_testdata_dir.join("Dapp.sol"), root.join("src/Dapp.sol")).unwrap(); let compiled = project.compile().unwrap(); assert_eq!( compiled.into_artifacts().map(|(artifact_id, _)| artifact_id.name).collect::>(), HashSet::from([ "DappTest".to_string(), "NewContract".to_string(), "DSTest".to_string(), "Dapp".to_string(), ]) ); // deleted artifact is not taken from the cache std::fs::remove_file(&project.paths.sources.join("Dapp.sol")).unwrap(); let compiled: ProjectCompileOutput<_> = project.compile().unwrap(); assert!(compiled.find("Dapp").is_none()); } fn copy_dir_all(src: impl AsRef, dst: impl AsRef) -> io::Result<()> { std::fs::create_dir_all(&dst)?; for entry in std::fs::read_dir(src)? { let entry = entry?; let ty = entry.file_type()?; if ty.is_dir() { copy_dir_all(entry.path(), dst.as_ref().join(entry.file_name()))?; } else { std::fs::copy(entry.path(), dst.as_ref().join(entry.file_name()))?; } } Ok(()) } #[test] fn can_flatten_file() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/test-contract-libs"); let target = root.join("src").join("Foo.sol"); let paths = ProjectPathsConfig::builder() .sources(root.join("src")) .lib(root.join("lib1")) .lib(root.join("lib2")); let project = TempProject::::new(paths).unwrap(); let result = project.flatten(&target); assert!(result.is_ok()); let result = result.unwrap(); assert!(!result.contains("import")); assert!(result.contains("contract Foo")); assert!(result.contains("contract Bar")); } #[test] fn can_flatten_file_with_external_lib() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/hardhat-sample"); let paths = ProjectPathsConfig::builder() .sources(root.join("contracts")) .lib(root.join("node_modules")); let project = TempProject::::new(paths).unwrap(); let target = root.join("contracts").join("Greeter.sol"); let result = project.flatten(&target); assert!(result.is_ok()); let result = result.unwrap(); assert!(!result.contains("import")); assert!(result.contains("library console")); assert!(result.contains("contract Greeter")); } #[test] fn can_flatten_file_in_dapp_sample() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/dapp-sample"); let paths = ProjectPathsConfig::builder().sources(root.join("src")).lib(root.join("lib")); let project = TempProject::::new(paths).unwrap(); let target = root.join("src/Dapp.t.sol"); let result = project.flatten(&target); assert!(result.is_ok()); let result = result.unwrap(); assert!(!result.contains("import")); assert!(result.contains("contract DSTest")); assert!(result.contains("contract Dapp")); assert!(result.contains("contract DappTest")); } #[test] fn can_flatten_unique() { let project = TempProject::dapptools().unwrap(); let f = project .add_source( "A", r#" pragma solidity ^0.8.10; import "./C.sol"; import "./B.sol"; contract A { } "#, ) .unwrap(); project .add_source( "B", r#" pragma solidity ^0.8.10; import "./C.sol"; contract B { } "#, ) .unwrap(); project .add_source( "C", r#" pragma solidity ^0.8.10; import "./A.sol"; contract C { } "#, ) .unwrap(); let result = project.flatten(&f).unwrap(); assert_eq!( result, r#" pragma solidity ^0.8.10; contract C { } contract B { } contract A { } "# ); } #[test] fn can_flatten_file_with_duplicates() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/test-flatten-duplicates"); let paths = ProjectPathsConfig::builder().sources(root.join("contracts")); let project = TempProject::::new(paths).unwrap(); let target = root.join("contracts/FooBar.sol"); let result = project.flatten(&target); assert!(result.is_ok()); let result = result.unwrap(); assert_eq!( result, r#"//SPDX-License-Identifier: UNLICENSED pragma solidity >=0.6.0; contract Bar {} contract Foo {} contract FooBar {} "# ); } #[test] fn can_flatten_on_solang_failure() { let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("test-data/test-flatten-solang-failure"); let paths = ProjectPathsConfig::builder().sources(&root.join("contracts")); let project = TempProject::::new(paths).unwrap(); let target = root.join("contracts/Contract.sol"); let result = project.flatten(&target); assert!(result.is_ok()); let result = result.unwrap(); assert_eq!( result, r#"// SPDX-License-Identifier: UNLICENSED pragma solidity ^0.8.10; library Lib {} // Intentionally erroneous code contract Contract { failure(); } "# ); } #[test] fn can_flatten_multiline() { let project = TempProject::dapptools().unwrap(); let f = project .add_source( "A", r#" pragma solidity ^0.8.10; import "./C.sol"; import { IllegalArgument, IllegalState } from "./Errors.sol"; contract A { } "#, ) .unwrap(); project .add_source( "Errors", r#" pragma solidity ^0.8.10; error IllegalArgument(); error IllegalState(); "#, ) .unwrap(); project .add_source( "C", r#" pragma solidity ^0.8.10; contract C { } "#, ) .unwrap(); let result = project.flatten(&f).unwrap(); assert_eq!( result.trim(), r#"pragma solidity ^0.8.10; contract C { } error IllegalArgument(); error IllegalState(); contract A { }"# ); } #[test] fn can_detect_type_error() { let project = TempProject::::dapptools().unwrap(); project .add_source( "Contract", r#" pragma solidity ^0.8.10; contract Contract { function xyz() public { require(address(0), "Error"); } } "#, ) .unwrap(); let compiled = project.compile().unwrap(); assert!(compiled.has_compiler_errors()); } #[test] fn can_compile_single_files() { let tmp = TempProject::dapptools().unwrap(); let f = tmp .add_contract( "examples/Foo", r#" pragma solidity ^0.8.10; contract Foo {} "#, ) .unwrap(); let compiled = tmp.project().compile_file(f.clone()).unwrap(); assert!(!compiled.has_compiler_errors()); assert!(compiled.find("Foo").is_some()); let bar = tmp .add_contract( "examples/Bar", r#" pragma solidity ^0.8.10; contract Bar {} "#, ) .unwrap(); let compiled = tmp.project().compile_files(vec![f, bar]).unwrap(); assert!(!compiled.has_compiler_errors()); assert!(compiled.find("Foo").is_some()); assert!(compiled.find("Bar").is_some()); } #[test] fn consistent_bytecode() { let tmp = TempProject::dapptools().unwrap(); tmp.add_source( "LinkTest", r#" // SPDX-License-Identifier: MIT library LibTest { function foobar(uint256 a) public view returns (uint256) { return a * 100; } } contract LinkTest { function foo() public returns (uint256) { return LibTest.foobar(1); } } "#, ) .unwrap(); let compiled = tmp.compile().unwrap(); assert!(!compiled.has_compiler_errors()); let contract = compiled.find("LinkTest").unwrap(); let bytecode = &contract.bytecode.as_ref().unwrap().object; assert!(bytecode.is_unlinked()); let s = bytecode.as_str().unwrap(); assert!(!s.starts_with("0x")); let s = serde_json::to_string(&bytecode).unwrap(); assert_eq!(bytecode.clone(), serde_json::from_str(&s).unwrap()); } #[test] fn can_recompile_with_changes() { let mut tmp = TempProject::dapptools().unwrap(); tmp.project_mut().allowed_lib_paths = vec![tmp.root().join("modules")].into(); let content = r#" pragma solidity ^0.8.10; import "../modules/B.sol"; contract A {} "#; tmp.add_source("A", content).unwrap(); tmp.add_contract( "modules/B", r#" pragma solidity ^0.8.10; contract B {} "#, ) .unwrap(); let compiled = tmp.compile().unwrap(); assert!(!compiled.has_compiler_errors()); assert!(compiled.find("A").is_some()); assert!(compiled.find("B").is_some()); let compiled = tmp.compile().unwrap(); assert!(compiled.find("A").is_some()); assert!(compiled.find("B").is_some()); assert!(compiled.is_unchanged()); // modify A.sol tmp.add_source("A", format!("{}\n", content)).unwrap(); let compiled = tmp.compile().unwrap(); assert!(!compiled.has_compiler_errors()); assert!(!compiled.is_unchanged()); assert!(compiled.find("A").is_some()); assert!(compiled.find("B").is_some()); } #[test] fn can_recompile_with_lowercase_names() { init_tracing(); let tmp = TempProject::dapptools().unwrap(); tmp.add_source( "deployProxy.sol", r#" pragma solidity =0.8.12; contract DeployProxy {} "#, ) .unwrap(); let upgrade = r#" pragma solidity =0.8.12; import "./deployProxy.sol"; import "./ProxyAdmin.sol"; contract UpgradeProxy {} "#; tmp.add_source("upgradeProxy.sol", upgrade).unwrap(); tmp.add_source( "ProxyAdmin.sol", r#" pragma solidity =0.8.12; contract ProxyAdmin {} "#, ) .unwrap(); let compiled = tmp.compile().unwrap(); assert!(!compiled.has_compiler_errors()); assert!(compiled.find("DeployProxy").is_some()); assert!(compiled.find("UpgradeProxy").is_some()); assert!(compiled.find("ProxyAdmin").is_some()); let artifacts = tmp.artifacts_snapshot().unwrap(); assert_eq!(artifacts.artifacts.as_ref().len(), 3); artifacts.assert_artifacts_essentials_present(); let compiled = tmp.compile().unwrap(); assert!(compiled.find("DeployProxy").is_some()); assert!(compiled.find("UpgradeProxy").is_some()); assert!(compiled.find("ProxyAdmin").is_some()); assert!(compiled.is_unchanged()); // modify upgradeProxy.sol tmp.add_source("upgradeProxy.sol", format!("{}\n", upgrade)).unwrap(); let compiled = tmp.compile().unwrap(); assert!(!compiled.has_compiler_errors()); assert!(!compiled.is_unchanged()); assert!(compiled.find("DeployProxy").is_some()); assert!(compiled.find("UpgradeProxy").is_some()); assert!(compiled.find("ProxyAdmin").is_some()); let artifacts = tmp.artifacts_snapshot().unwrap(); assert_eq!(artifacts.artifacts.as_ref().len(), 3); artifacts.assert_artifacts_essentials_present(); } #[test] fn can_recompile_unchanged_with_empty_files() { let tmp = TempProject::dapptools().unwrap(); tmp.add_source( "A", r#" pragma solidity ^0.8.10; import "./B.sol"; contract A {} "#, ) .unwrap(); tmp.add_source( "B", r#" pragma solidity ^0.8.10; import "./C.sol"; "#, ) .unwrap(); let c = r#" pragma solidity ^0.8.10; contract C {} "#; tmp.add_source("C", c).unwrap(); let compiled = tmp.compile().unwrap(); assert!(!compiled.has_compiler_errors()); assert!(compiled.find("A").is_some()); assert!(compiled.find("C").is_some()); let compiled = tmp.compile().unwrap(); assert!(compiled.find("A").is_some()); assert!(compiled.find("C").is_some()); assert!(compiled.is_unchanged()); // modify C.sol tmp.add_source("C", format!("{}\n", c)).unwrap(); let compiled = tmp.compile().unwrap(); assert!(!compiled.has_compiler_errors()); assert!(!compiled.is_unchanged()); assert!(compiled.find("A").is_some()); assert!(compiled.find("C").is_some()); }