def setUp(self):
self.tx_helper = self.add_helper(TransportHelper(TruteqTransport))
self.fake_server = FakeServer.for_protocol(SSMIServerProtocol)
self.config = {"username": "******", "password": "******", "twisted_endpoint": self.fake_server.endpoint}
self.transport = yield self.tx_helper.get_transport(self.config)
self.conn = yield self.fake_server.await_connection()
yield self.conn.await_connected()
self.server = self.conn.server_protocol
yield self.process_login_commands(self.server, "username", "password")
def setUp(self):
self.tx_helper = self.add_helper(TransportHelper(TruteqTransport))
self.fake_server = FakeServer.for_protocol(SSMIServerProtocol)
self.config = {
'username': '******',
'password': '******',
'twisted_endpoint': self.fake_server.endpoint,
}
self.transport = yield self.tx_helper.get_transport(self.config)
self.conn = yield self.fake_server.await_connection()
yield self.conn.await_connected()
self.server = self.conn.server_protocol
yield self.process_login_commands(self.server, 'username', 'password')
def test_server_for_protocol(self):
"""
A FakeServer can also be constructed from a a protocol class instead of
a factory.
"""
class MyProtocol(Protocol):
pass
fake_server = FakeServer.for_protocol(MyProtocol)
self.assertEqual(fake_server.server_factory.protocol, MyProtocol)
self.assertNotEqual(fake_server.server_factory.protocol, Protocol)
yield fake_server.endpoint.connect(self.client_factory)
conn = yield fake_server.await_connection()
self.assertIsInstance(conn.server_protocol, MyProtocol)
def test_server_for_protocol(self):
"""
A FakeServer can also be constructed from a a protocol class instead of
a factory.
"""
class MyProtocol(Protocol):
pass
fake_server = FakeServer.for_protocol(MyProtocol)
self.assertEqual(fake_server.server_factory.protocol, MyProtocol)
self.assertNotEqual(fake_server.server_factory.protocol, Protocol)
yield fake_server.endpoint.connect(self.client_factory)
conn = yield fake_server.await_connection()
self.assertIsInstance(conn.server_protocol, MyProtocol)
