From 4abd60ccb0e1fe0ca0737c6f21d3ed922d5eece5 Mon Sep 17 00:00:00 2001
From: Duc Huong <huong@codelight.co>
Date: Tue, 26 Dec 2023 15:31:50 +0700
Subject: [PATCH] feat: add sui cluster test scripts

---
 scalar/docker/runner.Dockerfile |  2 ++
 scalar/scripts/build.sh         | 49 +++++++++++++++++++++++++++++++++
 scalar/scripts/entry.sh         | 13 +++++++++
 scalar/scripts/run.sh           |  8 ++++++
 4 files changed, 72 insertions(+)

diff --git a/scalar/docker/runner.Dockerfile b/scalar/docker/runner.Dockerfile
index d736f37fa9b01..78a939918d2e0 100644
--- a/scalar/docker/runner.Dockerfile
+++ b/scalar/docker/runner.Dockerfile
@@ -4,6 +4,8 @@ FROM debian:bookworm-slim AS runtime
 RUN apt-get update && apt-get install -y libjemalloc-dev ca-certificates
 # For debuging
 RUN apt-get update && apt-get install -y nano curl
+# For libpq
+RUN apt-get update && apt-get install -y libpq-dev
 ENV LD_PRELOAD /usr/lib/x86_64-linux-gnu/libjemalloc.so
 # ARG PROFILE=release
 # ARG BUILD_DATE
diff --git a/scalar/scripts/build.sh b/scalar/scripts/build.sh
index 0b30ab9e5a61a..423f17c36d8c6 100755
--- a/scalar/scripts/build.sh
+++ b/scalar/scripts/build.sh
@@ -71,6 +71,55 @@ scalar() {
     rm ${SCRIPT_DIR}/${BIN_NAME}
 }
 
+sui_test_validator() {
+    BIN_NAME=sui-test-validator
+    WORKING_DIR=/scalar
+    docker exec -it ${BUILDER} cargo build --manifest-path ${WORKING_DIR}/crates/${BIN_NAME}/Cargo.toml --profile dev --bin ${BIN_NAME}
+    docker cp ${BUILDER}:${WORKING_DIR}/target/${PROFILE}/${BIN_NAME} ${SCRIPT_DIR}/${BIN_NAME}
+    docker cp ${SCRIPT_DIR}/${BIN_NAME} ${RUNNER}:/usr/local/bin
+    rm ${SCRIPT_DIR}/${BIN_NAME}
+}
+
+sui_cluster_test() {
+    BIN_NAME=sui-cluster-test
+    WORKING_DIR=/scalar
+    BASIC_DIR_NAME=basics
+    BASIC_OUTER_DIR=/sui_programmability/examples
+    BASIC_PACKAGES_DIR=${BASIC_OUTER_DIR}/${BASIC_DIR_NAME}
+    MANAGED_COIN_DIR_NAME=managed_coin
+    MANAGED_OUTER_DIR=/crates/sui-core/src/unit_tests/data
+    MANAGED_COIN_PACKAGES_DIR=${MANAGED_OUTER_DIR}/${MANAGED_COIN_DIR_NAME}
+    PACKAGES_DIR_NAME=packages
+    PACKAGES_OUTER_DIR=/crates/sui-framework
+    PACKAGES_DIR=${PACKAGES_OUTER_DIR}/${PACKAGES_DIR_NAME}
+
+    docker exec -it ${BUILDER} cargo build --manifest-path ${WORKING_DIR}/crates/${BIN_NAME}/Cargo.toml --profile dev --bin ${BIN_NAME}
+    docker cp ${BUILDER}:${WORKING_DIR}/target/${PROFILE}/${BIN_NAME} ${SCRIPT_DIR}/${BIN_NAME}
+    docker cp ${SCRIPT_DIR}/${BIN_NAME} ${RUNNER}:/usr/local/bin
+
+    docker exec -it ${RUNNER} mkdir -p ${WORKING_DIR}/crates/sui-test-transaction-builder
+
+    mkdir -p ${SCRIPT_DIR}/${BASIC_DIR_NAME}
+    docker cp ${BUILDER}:${WORKING_DIR}${BASIC_PACKAGES_DIR} ${SCRIPT_DIR}
+    docker exec -it ${RUNNER} mkdir -p ${WORKING_DIR}${BASIC_PACKAGES_DIR}
+    docker cp ${SCRIPT_DIR}/${BASIC_DIR_NAME}/ ${RUNNER}:${WORKING_DIR}${BASIC_OUTER_DIR}
+
+    mkdir -p ${SCRIPT_DIR}/${MANAGED_COIN_DIR_NAME}
+    docker cp ${BUILDER}:${WORKING_DIR}${MANAGED_COIN_PACKAGES_DIR} ${SCRIPT_DIR}
+    docker exec -it ${RUNNER} mkdir -p ${WORKING_DIR}${MANAGED_COIN_PACKAGES_DIR}
+    docker cp ${SCRIPT_DIR}/${MANAGED_COIN_DIR_NAME}/ ${RUNNER}:${WORKING_DIR}${MANAGED_OUTER_DIR}
+
+    mkdir -p ${SCRIPT_DIR}/${PACKAGES_DIR_NAME}
+    docker cp ${BUILDER}:${WORKING_DIR}${PACKAGES_DIR} ${SCRIPT_DIR}
+    docker exec -it ${RUNNER} mkdir -p ${WORKING_DIR}${PACKAGES_DIR}
+    docker cp ${SCRIPT_DIR}/${PACKAGES_DIR_NAME}/ ${RUNNER}:${WORKING_DIR}${PACKAGES_OUTER_DIR}
+
+    rm ${SCRIPT_DIR}/${BIN_NAME}
+    rm -rf ${SCRIPT_DIR}/${BASIC_DIR_NAME}
+    rm -rf ${SCRIPT_DIR}/${MANAGED_COIN_DIR_NAME}
+    rm -rf ${SCRIPT_DIR}/${PACKAGES_DIR_NAME}
+}
+
 consensus() {
     BIN_NAME=consensus-node
     WORKING_DIR=/scalar
diff --git a/scalar/scripts/entry.sh b/scalar/scripts/entry.sh
index 65e8c74a16c80..b7713c55cd24f 100755
--- a/scalar/scripts/entry.sh
+++ b/scalar/scripts/entry.sh
@@ -16,6 +16,19 @@ reth_test_client() {
     RUST_LOG=info /usr/local/bin/reth-test-client send_raw_tx ${TX_COUNT}
 }  
 
+sui_cluster_test() {
+    RUST_LOG=info /usr/local/bin/sui-cluster-test \
+        custom-remote \
+        --fullnode-address http://127.0.0.1:9000 \
+        --faucet-address http://127.0.0.1:9123
+}
+
+sui_test_validator() {
+    RUST_LOG=info /usr/local/bin/sui-test-validator \
+        --fullnode-rpc-port 9000 \
+        --faucet-port 9123
+}
+
 scalar_reth() {
     RUST_LOG=info /usr/local/bin/scalar-reth node \
         --chain dev \
diff --git a/scalar/scripts/run.sh b/scalar/scripts/run.sh
index ad1d70209eb4a..8ef6d013601b7 100755
--- a/scalar/scripts/run.sh
+++ b/scalar/scripts/run.sh
@@ -6,6 +6,14 @@ scalar_cluster() {
   docker exec -it ${RUNNER} /entry.sh scalar_cluster
 }
 
+sui_test_validator() {
+  docker exec -it ${RUNNER} /entry.sh sui_test_validator
+}
+
+sui_cluster_test() {
+  docker exec -it ${RUNNER} /entry.sh sui_cluster_test
+}
+
 # HuongND 2023-12-14
 reth_test_cluster() {
   docker exec -it scalar-runner rm -rf /root/.local/share/reth