@@ -524,6 +524,7 @@ test_nc_connect_ssh_interactive_succesfull(void **state)
524524 session = nc_connect_ssh ("127.0.0.1" , 8080 , NULL );
525525 assert_non_null (session );
526526
527+ will_return (__wrap_ssh_channel_poll_timeout , 0 );
527528 nc_session_free (session , NULL );
528529}
529530
@@ -574,6 +575,7 @@ test_nc_connect_ssh_password_succesfull(void **state)
574575 assert_non_null (session );
575576
576577 /* disconnect */
578+ will_return (__wrap_ssh_channel_poll_timeout , 0 );
577579 nc_session_free (session , NULL );
578580}
579581
@@ -613,6 +615,7 @@ test_nc_connect_ssh_pubkey_succesfull(void **state)
613615 assert_non_null (session );
614616
615617 /* disconnect */
618+ will_return (__wrap_ssh_channel_poll_timeout , 0 );
616619 nc_session_free (session , NULL );
617620}
618621
@@ -625,6 +628,7 @@ test_nc_connect_connection_failed(void **state)
625628 errno = ECONNREFUSED ;
626629 will_return (__wrap_connect , -1 );
627630 will_return (__wrap_ssh_is_connected , 0 );
631+ will_return (__wrap_ssh_channel_poll_timeout , 0 );
628632
629633 session = nc_connect_ssh ("127.0.0.1" , 8080 , NULL );
630634 assert_null (session );
@@ -654,6 +658,7 @@ test_nc_connect_ssh_bad_hello(void **state)
654658 will_return (__wrap_ssh_channel_open_session , 0 );
655659 will_return (__wrap_ssh_channel_request_subsystem , 0 );
656660 will_return (__wrap_nc_handshake_io , 4 );
661+ will_return (__wrap_ssh_channel_poll_timeout , 0 );
657662
658663 session = nc_connect_ssh ("127.0.0.1" , 8080 , NULL );
659664 assert_null (session );
0 commit comments