add a way to statically insert mock addresses

CosmWasm · Mar 11, 2024 · ae2e069 · ae2e069
1 parent df67017
commit ae2e069
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/packages/easy-addr/Cargo.toml b/packages/easy-addr/Cargo.toml
@@ -0,0 +1,14 @@
+[package]
+name = "easy-addr"
+version.workspace = true
+edition = "2021"
+publish = false
+
+[lib]
+proc-macro = true
+
+[dependencies]
+cosmwasm-std = { workspace = true }
+proc-macro2 = "1"
+quote = "1"
+syn = { version = "1.0.6", features = ["full", "printing", "extra-traits"] }
diff --git a/packages/easy-addr/src/lib.rs b/packages/easy-addr/src/lib.rs
@@ -0,0 +1,15 @@
+use cosmwasm_std::testing::mock_dependencies;
+
+use proc_macro::TokenStream;
+use quote::quote;
+use syn::parse_macro_input;
+
+#[proc_macro]
+pub fn addr(input: TokenStream) -> TokenStream {
+    let input = parse_macro_input!(input as syn::LitStr).value();
+    let addr = mock_dependencies()
+        .api
+        .addr_make(input.as_str())
+        .to_string();
+    TokenStream::from(quote! {#addr})
+}