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)