Stake Table Integration test

.env

@@ -33,7 +33,7 @@ ESPRESSO_SEQUENCER4_API_PORT=24004
 ESPRESSO_SEQUENCER_URL=http://sequencer0:${ESPRESSO_SEQUENCER_API_PORT}
 ESPRESSO_SEQUENCER_MAX_CONNECTIONS=25
 ESPRESSO_SEQUENCER_STORAGE_PATH=/store/sequencer
-ESPRESSO_SEQUENCER_GENESIS_FILE=/genesis/demo.toml
+ESPRESSO_SEQUENCER_GENESIS_FILE=data/genesis/demo.toml
 ESPRESSO_SEQUENCER_L1_PORT=8545
 ESPRESSO_SEQUENCER_L1_POLLING_INTERVAL=100ms
 ESPRESSO_SEQUENCER_L1_WS_PORT=8546

.github/workflows/test.yml

@@ -190,11 +190,14 @@ jobs:
     needs: [build-test-bins, build-test-artifacts-postgres]
     strategy:
       matrix:
-        version: [02,99]
+        version: [03]
         include:
           - version: 02
             compose: "-f process-compose.yaml -D"
-
+          - version: 03
+            env:
+              ESPRESSO_SEQUENCER_GENESIS_FILE=data/genesis/demo-pos-base.toml
+            compose: "-f process-compose.yaml -D"
           - version: 99
             compose: "-f process-compose.yaml -f process-compose-mp.yml -D"
     runs-on: ubuntu-latest
@@ -231,7 +234,7 @@ jobs:
           NEXTEST_PROFILE: integration
           INTEGRATION_TEST_SEQUENCER_VERSION: ${{ matrix.version }}
         run: |
-          cargo nextest run --archive-file nextest-archive-postgres.tar.zst --verbose --no-fail-fast --nocapture \
+          cargo nextest run --archive-file nextest-archive-postgres.tar.zst --verbose --no-fail-fast \
           --workspace-remap $PWD $(if [ "${{ matrix.version }}" == "2" ]; then echo " smoke"; fi)
         timeout-minutes: 10
 

Cargo.toml

@@ -224,14 +224,6 @@ paste = "1.0"
 rand = "0.8.5"
 time = "0.3"
 trait-set = "0.3.0"
-
-[profile.dev]
-# No optimizations
-opt-level = 0
-# Skip compiling the debug information.
-debug = false
-# Skip linking symbols.
-strip = true
 [profile.test]
 opt-level = 1
 [profile.test.package.tests]

builder/src/lib.rs

@@ -139,7 +139,7 @@ pub mod testing {
                 start_voting_time: 0,
                 stop_proposing_time: 0,
                 stop_voting_time: 0,
-                epoch_height: 0,
+                epoch_height: 150,
             };
 
             Self {

client/Cargo.toml

@@ -15,3 +15,4 @@ surf-disco = { workspace = true }
 tokio = { workspace = true }
 tracing = { workspace = true }
 vbs = { workspace = true }
+hotshot-types = { workspace = true }

client/src/lib.rs

@@ -1,7 +1,8 @@
 use anyhow::Context;
-use espresso_types::{FeeAccount, FeeAmount, FeeMerkleTree, Header};
+use espresso_types::{FeeAccount, FeeAmount, FeeMerkleTree, Header, PubKey};
 use ethers::types::Address;
 use futures::{stream::BoxStream, StreamExt};
+use hotshot_types::stake_table::StakeTableEntry;
 use jf_merkle_tree::{
     prelude::{MerkleProof, Sha3Node},
     MerkleTreeScheme,
@@ -51,7 +52,7 @@ impl SequencerClient {
         height: u64,
     ) -> anyhow::Result<BoxStream<'static, Result<Header, ClientError>>> {
         self.0
-            .socket(&format!("availability/stream/headers/{height}"))
+            .socket(&format!("v0/availability/stream/headers/{height}"))
             .subscribe::<Header>()
             .await
             .context("subscribing to Espresso headers")
@@ -119,6 +120,22 @@ impl SequencerClient {
         let balance = proof.elem().copied().unwrap_or(0.into());
         Ok(balance)
     }
+
+    pub async fn current_epoch(&self) -> anyhow::Result<u64> {
+        self.0
+            .get::<u64>("node/current_epoch")
+            .send()
+            .await
+            .context("getting epoch value")
+    }
+
+    pub async fn stake_table(&self, epoch: u64) -> anyhow::Result<Vec<StakeTableEntry<PubKey>>> {
+        self.0
+            .get::<_>(&format!("node/stake-table/{epoch}"))
+            .send()
+            .await
+            .context("getting epoch value")
+    }
 }
 
 #[cfg(test)]

data/genesis/demo-pos-base.toml

@@ -0,0 +1,21 @@
+base_version = "0.3"
+upgrade_version = "0.3"
+
+[stake_table]
+capacity = 10
+
+[chain_config]
+chain_id = 999999999
+max_block_size = '1mb'
+base_fee = '1 wei'
+fee_recipient = "0x0000000000000000000000000000000000000000"
+# bid_recipient = "0x0000000000000000000000000000000000000000"
+fee_contract = "0xa15bb66138824a1c7167f5e85b957d04dd34e468"
+stake_table_contract = "0x8ce361602b935680e8dec218b820ff5056beb7af"
+
+[header]
+timestamp = "1970-01-01T00:00:00Z"
+
+[l1_finalized]
+number = 0
+

data/genesis/demo-epoch.toml → data/genesis/demo-pos.toml

@@ -18,7 +18,7 @@ timestamp = "1970-01-01T00:00:00Z"
 number = 0
 
 [[upgrade]]
-version = "0.99"
+version = "0.3"
 start_proposing_view = 10
 stop_proposing_view = 60
 

hotshot-query-service/examples/simple-server.rs

@@ -216,7 +216,7 @@ async fn init_consensus(
         stop_proposing_time: 0,
         start_voting_time: 0,
         stop_voting_time: 0,
-        epoch_height: 0,
+        epoch_height: 150,
     };
 
     let nodes = join_all(priv_keys.into_iter().zip(data_sources).enumerate().map(

hotshot-query-service/src/testing/consensus.rs

@@ -148,7 +148,7 @@ impl<D: DataSourceLifeCycle + UpdateStatusData> MockNetwork<D> {
             stop_proposing_time: 0,
             start_voting_time: 0,
             stop_voting_time: 0,
-            epoch_height: 0,
+            epoch_height: 150,
         };
         update_config(&mut config);
 

justfile

@@ -1,3 +1,5 @@
+# export RUST_BACKTRACE := "1"
+
 default:
     just --list
 
@@ -29,6 +31,12 @@ build profile="test":
 demo-native-mp *args: build
     scripts/demo-native -f process-compose.yaml -f process-compose-mp.yml {{args}}
 
+demo-native-pos *args: build
+    ESPRESSO_SEQUENCER_GENESIS_FILE=data/genesis/demo-pos.toml scripts/demo-native -f process-compose.yaml {{args}}
+
+demo-native-pos-base *args: build
+    ESPRESSO_SEQUENCER_GENESIS_FILE=data/genesis/demo-pos-base.toml scripts/demo-native -f process-compose.yaml {{args}}
+
 demo-native-benchmark:
     cargo build --release --features benchmarking
     scripts/demo-native
@@ -71,10 +79,15 @@ test-all:
 test-integration:
 	@echo 'NOTE that demo-native must be running for this test to succeed.'
 	INTEGRATION_TEST_SEQUENCER_VERSION=2 cargo nextest run --all-features --nocapture --profile integration smoke
+
 test-integration-mp:
     @echo 'NOTE that demo-native-mp must be running for this test to succeed.'
     INTEGRATION_TEST_SEQUENCER_VERSION=99 cargo nextest run --all-features --nocapture --profile integration
 
+test-integration-pos:
+    @echo 'NOTE that demo-native-pos must be running for this test to succeed.'
+    INTEGRATION_TEST_SEQUENCER_VERSION=3 cargo nextest run --all-features --nocapture --profile integration smoke
+
 clippy:
     @echo 'features: "embedded-db"'
     cargo clippy --workspace --features embedded-db --all-targets -- -D warnings

process-compose.yaml

@@ -5,8 +5,8 @@ environment:
   - ESPRESSO_SEQUENCER_ORCHESTRATOR_URL=http://localhost:$ESPRESSO_ORCHESTRATOR_PORT
   - ESPRESSO_SEQUENCER_URL=http://localhost:$ESPRESSO_SEQUENCER_API_PORT
   - ESPRESSO_SEQUENCER_L1_PROVIDER=http://localhost:$ESPRESSO_SEQUENCER_L1_PORT
-  - ESPRESSO_SEQUENCER_GENESIS_FILE=data/genesis/demo.toml
-  - ESPRESSO_BUILDER_GENESIS_FILE=data/genesis/demo.toml
+  # - ESPRESSO_SEQUENCER_GENESIS_FILE=$DEMO_GENESIS_FILE
+  # - ESPRESSO_BUILDER_GENESIS_FILE=$DEMO_GENESIS_FILE
   - ESPRESSO_SEQUENCER_INITIAL_PERMISSIONED_STAKE_TABLE_PATH=data/initial_stake_table.toml
   - ESPRESSO_STATE_RELAY_SERVER_URL=http://localhost:$ESPRESSO_STATE_RELAY_SERVER_PORT
   - QUERY_SERVICE_URI=http://localhost:$ESPRESSO_SEQUENCER1_API_PORT/v0/
@@ -534,7 +534,7 @@ processes:
         host: localhost
         port: $ESPRESSO_BUILDER_SERVER_PORT
         path: /healthcheck
-      failure_threshold: 100
+      failure_threshold: 5
       period_seconds: 1
     availability:
       restart: "exit_on_failure"

sequencer/Cargo.toml

@@ -6,6 +6,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [features]
+default = ["pos"]
 testing = [
     "hotshot-testing",
     "marketplace-builder-core",
@@ -15,6 +16,9 @@ testing = [
 ]
 benchmarking = []
 embedded-db = ["hotshot-query-service/embedded-db"]
+fee = []
+pos = []
+marketplace = []
 
 [[bin]]
 name = "espresso-dev-node"

sequencer/api/node.toml

@@ -6,3 +6,7 @@ DOC = "Get the stake table for the current epoch"
 PATH = ["stake-table/:epoch_number"]
 ":epoch_number" = "Integer"
 DOC = "Get the stake table for the given epoch"
+
+[route.current_epoch]
+PATH = ["current_epoch"]
+DOC = "Get the current epoch"