()
| 18 | |
| 19 | @patch('sshuttle.helpers.verbose', new=3) |
| 20 | def test_get_tcp_dstip(): |
| 21 | sock = Mock() |
| 22 | sock.getpeername.return_value = ("127.0.0.1", 1024) |
| 23 | sock.getsockname.return_value = ("127.0.0.2", 1025) |
| 24 | sock.family = AF_INET |
| 25 | |
| 26 | firewall = Mock() |
| 27 | firewall.pfile.readline.return_value = \ |
| 28 | b"QUERY_PF_NAT_SUCCESS 127.0.0.3,1026\n" |
| 29 | |
| 30 | method = get_method('pf') |
| 31 | method.set_firewall(firewall) |
| 32 | assert method.get_tcp_dstip(sock) == ('127.0.0.3', 1026) |
| 33 | |
| 34 | assert sock.mock_calls == [ |
| 35 | call.getpeername(), |
| 36 | call.getsockname(), |
| 37 | ] |
| 38 | assert firewall.mock_calls == [ |
| 39 | call.pfile.write(b'QUERY_PF_NAT 2,6,127.0.0.1,1024,127.0.0.2,1025\n'), |
| 40 | call.pfile.flush(), |
| 41 | call.pfile.readline() |
| 42 | ] |
| 43 | |
| 44 | |
| 45 | def test_recv_udp(): |
nothing calls this directly
no test coverage detected