36 lines
916 B
Python
36 lines
916 B
Python
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!'
|
|
|