You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
916 B
37 lines
916 B
9 years ago
|
import pytest
|
||
|
|
|
||
|
|
from proxy import ProxyClient, ProxyClientFactory, ProxyServer
|
||
|
|
from testutil import mock_deferred
|
||
|
|
from twisted.internet.protocol import ServerFactory
|
||
|
|
from twisted.test import proto_helpers
|
||
|
|
from twisted.internet import defer
|
||
|
|
|
||
|
|
####################
|
||
|
|
## Fixtures
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def proxyserver():
|
||
|
|
factory = ServerFactory()
|
||
|
|
factory.protocol = ProxyServer
|
||
|
|
protocol = factory.buildProtocol(('127.0.0.1', 0))
|
||
|
|
transport = proto_helpers.StringTransport()
|
||
|
|
protocol.makeConnection(transport)
|
||
|
|
return (protocol, transport)
|
||
|
|
|
||
|
|
####################
|
||
|
|
## Basic tests
|
||
|
|
|
||
|
|
def test_proxy_server_fixture(proxyserver):
|
||
|
|
prot = proxyserver[0]
|
||
|
|
tr = proxyserver[1]
|
||
|
|
prot.transport.write('hello')
|
||
|
|
print tr.value()
|
||
|
|
assert tr.value() == 'hello'
|
||
|
|
|
||
|
|
@pytest.inlineCallbacks
|
||
|
|
def test_mock_deferreds(mock_deferred):
|
||
|
|
d = mock_deferred('Hello!')
|
||
|
|
r = yield d
|
||
|
|
assert r == 'Hello!'
|
||
|
|
|