Update/add osmosis test tube

README.md

@@ -205,6 +205,10 @@ impl<Chain: CwEnv> Example<Chain> {
 }
 ```
 
+### Testing with OsmosisTestTube
+
+OsmosisTestTube is available for testing in cw-orchestrator. In order to use it, you may need to install [clang] and [go] to compile the osmosis blockchain that serves as the backend for this env. 
+
 ## Contributing
 
 We'd really appreciate your help! Please read our [contributing guidelines](docs/src/contributing.md) to get started.

contracts/counter/Cargo.toml

@@ -31,4 +31,4 @@ counter-contract = { path = ".", features = ["interface"] }
 # Deps for deployment
 dotenv = { version = "0.15.0" }
 env_logger = { version = "0.10.0" }
-cw-orch = { path = "../../cw-orch", features = ["daemon"] }
+cw-orch = { path = "../../cw-orch", features = ["daemon", "test-tube"] }

contracts/counter/tests/test-tube.rs

@@ -0,0 +1,90 @@
+// ANCHOR: all
+use cosmwasm_std::coins;
+use counter_contract::{
+    contract::CONTRACT_NAME,
+    msg::{GetCountResponse, InstantiateMsg, QueryMsg},
+    CounterContract,
+};
+use cw_orch::test_tube::OsmosisTestTube;
+// Use prelude to get all the necessary imports
+use cw_orch::prelude::*;
+
+// ANCHOR: integration_test
+// ANCHOR: setup
+/// Instantiate the contract in any CosmWasm environment
+fn setup<Chain: CwEnv>(chain: Chain) -> CounterContract<Chain> {
+    // ANCHOR: constructor
+    // Construct the counter interface
+    let contract = CounterContract::new(CONTRACT_NAME, chain.clone());
+    // ANCHOR_END: constructor
+
+    // Upload the contract
+    let upload_resp = contract.upload().unwrap();
+
+    // Get the code-id from the response.
+    let code_id = upload_resp.uploaded_code_id().unwrap();
+    // or get it from the interface.
+    assert_eq!(code_id, contract.code_id().unwrap());
+
+    // Instantiate the contract
+    let msg = InstantiateMsg { count: 1i32 };
+    let init_resp = contract
+        .instantiate(&msg, Some(&chain.sender()), None)
+        .unwrap();
+
+    // Get the address from the response
+    let contract_addr = init_resp.instantiated_contract_address().unwrap();
+    // or get it from the interface.
+    assert_eq!(contract_addr, contract.address().unwrap());
+
+    // Return the interface
+    contract
+}
+// ANCHOR_END: setup
+
+// ANCHOR: count_test
+#[test]
+fn count() {
+    // Create the mock
+    let test_tube = OsmosisTestTube::new(coins(100_000_000_000, "uosmo"));
+
+    let account = test_tube
+        .init_account(coins(100_000_000_000, "uosmo"))
+        .unwrap();
+
+    // Set up the contract
+    let contract = setup(test_tube);
+
+    // Increment the count of the contract
+    contract
+        // Set the caller to user
+        .call_as(&account)
+        // Call the increment function (auto-generated function provided by CounterExecuteMsgFns)
+        .increment()
+        .unwrap();
+
+    // ANCHOR: query
+    // Get the count.
+    use counter_contract::CounterQueryMsgFns;
+    let count1 = contract.get_count().unwrap();
+    // ANCHOR_END: query
+
+    // or query it manually
+    let count2: GetCountResponse = contract.query(&QueryMsg::GetCount {}).unwrap();
+
+    assert_eq!(count1, count2);
+
+    // Check the count
+    assert_eq!(count1.count, 2);
+    // ANCHOR: reset
+    // Reset
+    use counter_contract::CounterExecuteMsgFns;
+    contract.reset(0).unwrap();
+
+    let count = contract.get_count().unwrap();
+    assert_eq!(count.count, 0);
+    // ANCHOR_END: reset
+}
+// ANCHOR_END: count_test
+// ANCHOR_END: integration_test
+// ANCHOR_END: all

cw-orch/Cargo.toml

@@ -52,6 +52,7 @@ daemon = [
   "dep:prost",
 ]
 eth = ["daemon", "dep:ethers-signers", "dep:ethers-core", "dep:snailquote"]
+test-tube = ["dep:osmosis-test-tube"]
 
 [dependencies]
 # Default deps
@@ -103,6 +104,9 @@ ethers-signers = { version = "2.0.7", optional = true }
 ethers-core = { version = "2.0.7", optional = true }
 snailquote = { version = "0.3.1", optional = true }
 
+# Test Tube env deps
+osmosis-test-tube = { version = "15.1.0", optional = true }
+
 [dev-dependencies]
 cw-orch = { features = ["daemon"], path = "." }
 uid = "0.1.7"

cw-orch/src/error.rs

@@ -11,6 +11,9 @@ pub enum CwOrchError {
     #[cfg(feature = "daemon")]
     #[error(transparent)]
     DaemonError(#[from] DaemonError),
+    #[cfg(feature = "test-tube")]
+    #[error(transparent)]
+    TestTubeError(#[from] osmosis_test_tube::RunnerError),
     #[error("JSON Conversion Error")]
     SerdeJson(#[from] ::serde_json::Error),
     #[error(transparent)]

cw-orch/src/keys/private.rs

@@ -277,6 +277,7 @@ mod tst {
         Ok(())
     }
 
+    #[cfg(feature = "eth")]
     #[test]
     pub fn inj() -> anyhow::Result<()> {
         let str_1: &str = "across left ignore gold echo argue track joy hire release captain enforce hotel wide flash hotel brisk joke midnight duck spare drop chronic stool";

cw-orch/src/lib.rs

@@ -29,6 +29,8 @@ mod interface_traits;
 mod keys;
 pub mod mock;
 mod paths;
+#[cfg(feature = "test-tube")]
+pub mod test_tube;
 
 pub mod state;