Creating workspaces test suit

Cargo.toml

@@ -12,7 +12,13 @@ near-sdk = "3.1.0"
 near-contract-standards = "3.2.0"
 
 [dev-dependencies]
+insta = { version = "1.31.0", features = ["json", "redactions"] }
 regex = "1"
+near-workspaces = { version = "0.7.0", features = ["unstable"]  }
+tokio = { version = "1.10.0", features = ["full"] }
+serde_json = { version = "1.0", features = ["arbitrary_precision"] }
+near-units = "0.2.0"
+anyhow = "1.0"
 
 [profile.release]
 codegen-units = 1

rust-toolchain.toml

@@ -0,0 +1,2 @@
+[toolchain]
+channel = "1.69.0"

test.sh

@@ -17,4 +17,6 @@ contract=i.zxcvn.testnet
 
 # near call $contract unsafe_self_upgrade --accountId $contract --args $(base64 < res/devgovgigs.wasm ) --base64 --gas 300000000000000
 
-near call $contract unsafe_migrate --accountId $contract --gas 300000000000000
+#near call contract.devhubopen.testnet unsafe_self_upgrade --accountId contract.devhubopen.testnet --args $(base64 < res/devgovgigs.wasm ) --base64 --gas 300000000000000
+
+# near call $contract unsafe_migrate --accountId $contract --gas 300000000000000

tests/README.md

@@ -0,0 +1,18 @@
+# Integration Test using near-workspaces-rs
+
+This codebase is designed to deploy and test the Devhub contract on a Sandbox node.
+
+## Dependencies
+
+- Rust: For writing the test suite.
+- `near-workspaces`: A custom library for handling contract calls.
+- `near_units`: For handling NEAR unit conversions.
+- `serde_json`: For JSON serialization and deserialization.
+
+### How to Run
+
+To run the test, use the following command:
+
+```bash
+cargo test
+```

tests/migration.rs

@@ -0,0 +1,223 @@
+use near_units::parse_near;
+use near_workspaces::AccountId;
+use serde_json::json;
+
+const DEVHUB_CONTRACT: &str = "devgovgigs.near";
+const NEAR_SOCIAL: &str = "social.near";
+
+#[tokio::test]
+async fn test_deploy_contract_self_upgrade() -> anyhow::Result<()> {
+    // Test Flow:
+    // 1. Deploy devhub and near social contract on sandbox
+    // 2. Add all kinds of posts and add a community.
+    // 3. Upgrade the contract.
+    // 4. Get all the posts and community and check if migration was succesfull.
+
+    // Initialize the devhub and near social contract on chain,
+    // contract is devhub contract instance.
+    let contract = init_contracts().await?;
+
+    let deposit_amount = near_units::parse_near!("0.1");
+
+    // Add Posts
+    let add_idea_post = contract
+        .call("add_post")
+        .args_json(json!({
+            "parent_id": null,
+            "labels": [],
+            "body": {
+                "name": "This is a test idea.",
+                "description": "This is a test description.",
+                "post_type": "Idea",
+                "idea_version": "V1"
+            }
+        }))
+        .deposit(deposit_amount)
+        .transact()
+        .await?;
+    assert!(add_idea_post.is_success());
+
+    let add_submission_post = contract
+        .call("add_post")
+        .args_json(json!({
+            "parent_id": null,
+            "labels": [],
+            "body": {
+                "name": "Solution Test",
+                "description": "###### Requested amount: 100 NEAR\n###### Requested sponsor: @neardevgov.near\nThis is a test submission. ",
+                "post_type": "Submission",
+                "submission_version": "V1"
+            }
+        }))
+        .deposit(deposit_amount)
+        .max_gas()
+        .transact()
+        .await?;
+    assert!(add_submission_post.is_success());
+
+    let add_comment_post = contract
+        .call("add_post")
+        .args_json(json!({
+            "parent_id": 0,
+            "labels": [],
+            "body": {
+                "description": "This is test Comment.",
+                "comment_version": "V2",
+                "post_type": "Comment"
+            }
+        }))
+        .deposit(deposit_amount)
+        .max_gas()
+        .transact()
+        .await?;
+    assert!(add_comment_post.is_success());
+
+    let add_attestation_post = contract
+        .call("add_post")
+        .args_json(json!({
+            "parent_id": 1,
+            "labels": [],
+            "body": {
+                "name": "Attestation",
+                "description": "Description",
+                "attestation_version": "V1",
+                "post_type": "Attestation"
+            }
+        }))
+        .deposit(deposit_amount)
+        .max_gas()
+        .transact()
+        .await?;
+    assert!(add_attestation_post.is_success());
+
+    let add_sponsorship_post = contract
+        .call("add_post")
+        .args_json(json!({
+            "parent_id": 1,
+            "labels": [],
+            "body": {
+                "name": "Contributor fellowship",
+                "description": "Funding approved",
+                "amount": "1000",
+                "sponsorship_token": "Near",
+                "supervisor": "john.near",
+                "sponsorship_version": "V1",
+                "post_type": "Sponsorship"
+            }
+        }))
+        .deposit(deposit_amount)
+        .max_gas()
+        .transact()
+        .await?;
+    assert!(add_sponsorship_post.is_success());
+
+    // Add a community
+    let create_community = contract
+        .call("create_community")
+        .args_json(json!({
+            "inputs": {
+                "handle": "gotham",
+                "name": "Gotham",
+                "tag": "some",
+                "description": "This is a test community.",
+                "bio_markdown": "This is a sample text about your community.\nYou can change it on the community configuration page.",
+                "logo_url": "https://ipfs.near.social/ipfs/bafkreibysr2mkwhb4j36h2t7mqwhynqdy4vzjfygfkfg65kuspd2bawauu",
+                "banner_url": "https://ipfs.near.social/ipfs/bafkreic4xgorjt6ha5z4s5e3hscjqrowe5ahd7hlfc5p4hb6kdfp6prgy4"
+            }
+        }))
+        .max_gas()
+        .transact()
+        .await?;
+    assert!(create_community.is_success());
+
+    // Call self upgrade with current branch code
+    // compile the current code
+    let wasm = near_workspaces::compile_project("./").await?;
+
+    let mut contract_upgrade_result =
+        contract.call("unsafe_self_upgrade").args(wasm).max_gas().transact().await?;
+
+    while contract_upgrade_result.json::<String>()? == "needs-migration" {
+        contract_upgrade_result =
+            contract.call("unsafe_migrate").args_json(json!({})).max_gas().transact().await?;
+    }
+
+    let get_idea_post: serde_json::Value = contract
+        .call("get_post")
+        .args_json(json!({
+            "post_id" : 0
+        }))
+        .view()
+        .await?
+        .json()?;
+    insta::assert_json_snapshot!(get_idea_post, {".snapshot.timestamp" => "[timestamp]"});
+
+    let get_comment_posts: serde_json::Value = contract
+        .call("get_posts")
+        .args_json(json!({
+            "parent_id" : 0
+        }))
+        .view()
+        .await?
+        .json()?;
+    insta::assert_json_snapshot!(get_comment_posts, {"[].snapshot.timestamp" => "[timestamp]"});
+
+    let get_submission_post: serde_json::Value = contract
+        .call("get_post")
+        .args_json(json!({
+            "post_id" : 1
+        }))
+        .view()
+        .await?
+        .json()?;
+    insta::assert_json_snapshot!(get_submission_post, {".snapshot.timestamp" => "[timestamp]"});
+
+    let get_attestation_sponsorship_posts: serde_json::Value = contract
+        .call("get_posts")
+        .args_json(json!({
+            "parent_id" : 1
+        }))
+        .view()
+        .await?
+        .json()?;
+    insta::assert_json_snapshot!(get_attestation_sponsorship_posts, {"[].snapshot.timestamp" => "[timestamp]"});
+
+    let get_community: serde_json::Value = contract
+        .call("get_community")
+        .args_json(json!({
+            "handle" : "gotham"
+        }))
+        .view()
+        .await?
+        .json()?;
+    insta::assert_json_snapshot!(get_community);
+
+    Ok(())
+}
+
+async fn init_contracts() -> anyhow::Result<near_workspaces::Contract> {
+    let worker = near_workspaces::sandbox().await?;
+    let mainnet = near_workspaces::mainnet_archival().await?;
+
+    // NEAR social deployment
+    let near_social_id: AccountId = NEAR_SOCIAL.parse()?;
+    let near_social = worker
+        .import_contract(&near_social_id, &mainnet)
+        .initial_balance(parse_near!("10000 N"))
+        .transact()
+        .await?;
+    near_social.call("new").transact().await?.into_result()?;
+
+    // Devhub contract deployment
+    let contract_id: AccountId = DEVHUB_CONTRACT.parse()?;
+    let contract = worker
+        .import_contract(&contract_id, &mainnet)
+        .initial_balance(parse_near!("1000 N"))
+        .transact()
+        .await?;
+    let outcome = contract.call("new").args_json(json!({})).transact().await?;
+    assert!(outcome.is_success());
+    assert!(format!("{:?}", outcome).contains("Migrated to version:"));
+
+    Ok(contract)
+}

tests/snapshots/migration__deploy_contract_self_upgrade-2.snap

@@ -0,0 +1,23 @@
+---
+source: tests/migration.rs
+expression: get_comment_posts
+---
+[
+  {
+    "post_version": "V0",
+    "id": {
+      "$serde_json::private::Number": "2"
+    },
+    "author_id": "devgovgigs.near",
+    "likes": [],
+    "snapshot": {
+      "editor_id": "devgovgigs.near",
+      "timestamp": "[timestamp]",
+      "labels": [],
+      "post_type": "Comment",
+      "comment_version": "V2",
+      "description": "This is test Comment."
+    },
+    "snapshot_history": []
+  }
+]

tests/snapshots/migration__deploy_contract_self_upgrade-3.snap

@@ -0,0 +1,22 @@
+---
+source: tests/migration.rs
+expression: get_submission_post
+---
+{
+  "post_version": "V0",
+  "id": {
+    "$serde_json::private::Number": "1"
+  },
+  "author_id": "devgovgigs.near",
+  "likes": [],
+  "snapshot": {
+    "editor_id": "devgovgigs.near",
+    "timestamp": "[timestamp]",
+    "labels": [],
+    "post_type": "Submission",
+    "submission_version": "V1",
+    "name": "Solution Test",
+    "description": "###### Requested amount: 100 NEAR\n###### Requested sponsor: @neardevgov.near\nThis is a test submission. "
+  },
+  "snapshot_history": []
+}

tests/snapshots/migration__deploy_contract_self_upgrade-4.snap

@@ -0,0 +1,45 @@
+---
+source: tests/migration.rs
+expression: get_attestation_sponsorship_posts
+---
+[
+  {
+    "post_version": "V0",
+    "id": {
+      "$serde_json::private::Number": "3"
+    },
+    "author_id": "devgovgigs.near",
+    "likes": [],
+    "snapshot": {
+      "editor_id": "devgovgigs.near",
+      "timestamp": "[timestamp]",
+      "labels": [],
+      "post_type": "Attestation",
+      "attestation_version": "V1",
+      "name": "Attestation",
+      "description": "Description"
+    },
+    "snapshot_history": []
+  },
+  {
+    "post_version": "V0",
+    "id": {
+      "$serde_json::private::Number": "4"
+    },
+    "author_id": "devgovgigs.near",
+    "likes": [],
+    "snapshot": {
+      "editor_id": "devgovgigs.near",
+      "timestamp": "[timestamp]",
+      "labels": [],
+      "post_type": "Sponsorship",
+      "sponsorship_version": "V1",
+      "name": "Contributor fellowship",
+      "description": "Funding approved",
+      "sponsorship_token": "Near",
+      "amount": "1000",
+      "supervisor": "john.near"
+    },
+    "snapshot_history": []
+  }
+]

tests/snapshots/migration__deploy_contract_self_upgrade-5.snap

@@ -0,0 +1,30 @@
+---
+source: tests/migration.rs
+expression: get_community
+---
+{
+  "admins": [
+    "devgovgigs.near"
+  ],
+  "handle": "gotham",
+  "name": "Gotham",
+  "tag": "some",
+  "description": "This is a test community.",
+  "logo_url": "https://ipfs.near.social/ipfs/bafkreibysr2mkwhb4j36h2t7mqwhynqdy4vzjfygfkfg65kuspd2bawauu",
+  "banner_url": "https://ipfs.near.social/ipfs/bafkreic4xgorjt6ha5z4s5e3hscjqrowe5ahd7hlfc5p4hb6kdfp6prgy4",
+  "bio_markdown": "This is a sample text about your community.\nYou can change it on the community configuration page.",
+  "github_handle": null,
+  "telegram_handle": [],
+  "twitter_handle": null,
+  "website_url": null,
+  "github": null,
+  "board": null,
+  "wiki1": null,
+  "wiki2": null,
+  "features": {
+    "telegram": true,
+    "github": true,
+    "board": true,
+    "wiki": true
+  }
+}