wip lease block

diff --git a/source4/torture/smb2/multichannel.c b/source4/torture/smb2/multichannel.c
index ac1a116fc0a4def9f287e07a60dd2156b86b72d7..eb1105087f7901ad47186244f0005fc2756a22fe 100644 (file)
--- a/source4/torture/smb2/multichannel.c
+++ b/source4/torture/smb2/multichannel.c
@@ -1195,6 +1195,14 @@ static bool test_multichannel_lease_break(struct torture_context *tctx,
 
        torture_reset_lease_break_info(tctx, &lease_break_info);
 
+
+       block_ok = torture_block_tcp_transport(tctx, transport2A);
+       torture_comment(tctx, "sleeping for 15000\n");
+       smb_msleep(15000);
+       block_ok = torture_block_tcp_transport(tctx, transport2B);
+       torture_comment(tctx, "sleeping for 15000\n");
+       smb_msleep(15000);
+
        /* block tcp connection of transport2C */
        block_ok = torture_block_tcp_transport(tctx, transport2C);
        torture_assert(tctx, block_ok, "we could not block tcp transport");
@@ -1259,6 +1267,8 @@ static bool test_multichannel_lease_break(struct torture_context *tctx,
        CHECK_PTR(lease_break_info.lease_transport, transport1);
 
        unblock_ok = torture_unblock_tcp_transport(tctx, transport2C);
+       unblock_ok = torture_unblock_tcp_transport(tctx, transport2A);
+       unblock_ok = torture_unblock_tcp_transport(tctx, transport2B);
        torture_assert(tctx, unblock_ok, "we could not unblock tcp transport");
 
        /* next test: disconnect 2C and trigger break */
@@ -1348,6 +1358,8 @@ static bool test_multichannel_lease_break(struct torture_context *tctx,
        if (block_ok && !unblock_ok) {
                /* unblock tcp connection of transport2C */
                unblock_ok = torture_unblock_tcp_transport(tctx, transport2C);
+               unblock_ok = torture_unblock_tcp_transport(tctx, transport2A);
+               unblock_ok = torture_unblock_tcp_transport(tctx, transport2B);
        }
 
        tree1->session = session1;