def test_maybe_tls_plaintext(mocker): mock_config(mocker) tls_callback = mock.MagicMock() p, t = gen_debug_protocol(mocker) start_maybe_tls(p.transport, 'www.foo.faketld') p.dataReceived("Hello world!") assert p.transport.pop_value() == "Hello world!"
def test_maybe_tls_use_tls(mocker): mock_config(mocker) tls_callback = mock.MagicMock() p, t = gen_debug_protocol(mocker) start_maybe_tls(p.transport, 'www.foo.faketld') maybe_tls_prot = t.protocol assert isinstance(maybe_tls_prot, MaybeTLSProtocol) assert maybe_tls_prot.state == MaybeTLSProtocol.STATE_DECIDING t.protocol.dataReceived("\x16") assert not isinstance(t.protocol, MaybeTLSProtocol) assert maybe_tls_prot.state == MaybeTLSProtocol.STATE_PASSTHROUGH