From 6f8dc93fc639abb17e2d478f052e48d78542d8be Mon Sep 17 00:00:00 2001
From: Matthias Seitz <matthias.seitz@outlook.de>
Date: Fri, 12 Aug 2022 01:42:52 +0200
Subject: [PATCH] feat(solc): add missing helper functions (#1592)

---
 ethers-solc/src/project_util/mod.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/ethers-solc/src/project_util/mod.rs b/ethers-solc/src/project_util/mod.rs
index 7d0c8b22..b5a35e5d 100644
--- a/ethers-solc/src/project_util/mod.rs
+++ b/ethers-solc/src/project_util/mod.rs
@@ -200,6 +200,20 @@ contract {} {{}}
         )
     }
 
+    /// Adds a new test file inside the project's test dir
+    pub fn add_test(&self, name: impl AsRef<str>, content: impl AsRef<str>) -> Result<PathBuf> {
+        let name = contract_file_name(name);
+        let tests = self.paths().tests.join(name);
+        create_contract_file(tests, content)
+    }
+
+    /// Adds a new script file inside the project's script dir
+    pub fn add_script(&self, name: impl AsRef<str>, content: impl AsRef<str>) -> Result<PathBuf> {
+        let name = contract_file_name(name);
+        let script = self.paths().scripts.join(name);
+        create_contract_file(script, content)
+    }
+
     /// Adds a new source file inside the project's source dir
     pub fn add_source(&self, name: impl AsRef<str>, content: impl AsRef<str>) -> Result<PathBuf> {
         let name = contract_file_name(name);