Add integration test for PoolSv2

.github/workflows/test.yaml

@@ -40,6 +40,10 @@ jobs:
           cargo build --manifest-path=roles/Cargo.toml
           cargo build --manifest-path=utils/Cargo.toml
 
+      - name: Roles Integration Tests
+        run: |
+          cargo test --manifest-path=roles/Cargo.toml --verbose --test '*' -- --nocapture
+
       - name: Run sv1-client-and-server example
         run: |
           cargo run --manifest-path=examples/sv1-client-and-server/Cargo.toml --bin client_and_server -- 60

roles/tests-integration/tests/common/mod.rs

@@ -168,11 +168,11 @@ impl Drop for TemplateProvider {
     }
 }
 
-pub fn is_port_open(address: SocketAddr) -> bool {
+fn is_port_open(address: SocketAddr) -> bool {
     TcpListener::bind(address).is_err()
 }
 
-pub fn get_available_port() -> u16 {
+fn get_available_port() -> u16 {
     let mut unique_ports = UNIQUE_PORTS.lock().unwrap();
 
     loop {
@@ -214,7 +214,11 @@ impl TestPoolSv2 {
         template_provider_address: Option<SocketAddr>,
     ) -> Self {
         use pool_sv2::mining_pool::{CoinbaseOutput, Configuration};
-        let pool_port = get_available_port();
+        let pool_port = if let Some(listen_addr) = listening_address {
+            listen_addr.port()
+        } else {
+            get_available_port()
+        };
         let listening_address = listening_address
             .unwrap_or(SocketAddr::from_str(&format!("127.0.0.1:{}", pool_port)).unwrap());
         let is_pool_port_open = is_port_open(listening_address);
@@ -247,8 +251,10 @@ impl TestPoolSv2 {
             cert_validity_sec,
             pool_signature,
         );
-        let template_provider_config =
-            pool_sv2::mining_pool::TemplateProviderConfig::new(tp_address, None);
+        let template_provider_config = pool_sv2::mining_pool::TemplateProviderConfig::new(
+            tp_address,
+            Some(authority_public_key),
+        );
         let authority_config =
             pool_sv2::mining_pool::AuthorityConfig::new(authority_public_key, authority_secret_key);
         let config = Configuration::new(

roles/tests-integration/tests/pool_integration.rs

@@ -0,0 +1,16 @@
+mod common;
+
+#[tokio::test]
+async fn success_pool_template_provider_connection() {
+    let sniffer_addr = common::get_available_address();
+    let tp_addr = common::get_available_address();
+    let pool_addr = common::get_available_address();
+    let _tp = common::start_template_provider(tp_addr.port()).await;
+    let sniffer = common::start_sniffer(tp_addr, sniffer_addr).await;
+    let _ = common::start_pool(Some(pool_addr), None, Some(sniffer_addr)).await;
+    assert!(sniffer.expect_downstream_setup_connection());
+    assert!(sniffer.expect_upstream_setup_connection_success());
+    assert!(sniffer.expect_downstream_coinbase_output_data_size());
+    assert!(sniffer.expect_upstream_new_template());
+    assert!(sniffer.expect_upstream_set_new_prev_hash());
+}