本文整理汇总了Python中twisted.test.proto_helpers.StringTransportWithDisconnection.write方法的典型用法代码示例。如果您正苦于以下问题：Python StringTransportWithDisconnection.write方法的具体用法？Python StringTransportWithDisconnection.write怎么用？Python StringTransportWithDisconnection.write使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类twisted.test.proto_helpers.StringTransportWithDisconnection的用法示例。
在下文中一共展示了StringTransportWithDisconnection.write方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: TestTruteqTransport
# 需要导入模块: from twisted.test.proto_helpers import StringTransportWithDisconnection [as 别名]
# 或者: from twisted.test.proto_helpers.StringTransportWithDisconnection import write [as 别名]
class TestTruteqTransport(VumiTestCase):
@inlineCallbacks
def setUp(self):
self.tx_helper = self.add_helper(TransportHelper(TruteqTransport))
self.config = {
'username': 'username',
'password': 'password',
}
self.string_transport = StringTransportWithDisconnection()
# NOTE: pausing the transport before starting so we can
# start the SSMIProtocol, which expects the vumi transport
# as an argument.
self.transport = yield self.tx_helper.get_transport(
self.config, start=False)
st_endpoint = StringTransportEndpoint(self._ste_connect_callback)
def truteq_service_maker(endpoint, factory):
return ReconnectingClientService(st_endpoint, factory)
self.transport.service_class = truteq_service_maker
yield self.transport.startWorker()
yield self.process_login_commands('username', 'password')
def _ste_connect_callback(self, protocol):
self.protocol = protocol
self.string_transport = protocol.transport
@inlineCallbacks
def process_login_commands(self, username, password):
[cmd] = yield self.receive(1)
self.assertEqual(cmd.command_name, 'LOGIN')
self.assertEqual(cmd.username, username)
self.assertEqual(cmd.password, password)
self.send(Ack(ack_type='1'))
[link_check] = yield self.receive(1)
self.assertEqual(link_check.command_name, 'LINK_CHECK')
returnValue(True)
def send(self, command):
return self.protocol.lineReceived(str(command))
def receive(self, count, clear=True):
d = Deferred()
def check_for_input():
if not self.string_transport.value():
reactor.callLater(0, check_for_input)
return
lines = self.string_transport.value().split(
self.protocol.delimiter)
commands = map(SSMIRequest.parse, filter(None, lines))
if len(commands) >= count:
if clear:
self.string_transport.clear()
self.string_transport.write(
self.protocol.delimiter.join(
map(str, commands[count:])))
d.callback(commands[:count])
check_for_input()
return d
def incoming_ussd(self, msisdn="12345678", ussd_type=c.USSD_RESPONSE,
phase="ignored", message="Hello"):
return self.transport.handle_raw_inbound_message(
USSDMessage(msisdn=msisdn, type=ussd_type,
phase=c.USSD_PHASE_UNKNOWN,
message=message))
@inlineCallbacks
def start_ussd(self, message="*678#", **kw):
yield self.incoming_ussd(ussd_type=c.USSD_NEW, message=message, **kw)
self.tx_helper.clear_dispatched_inbound()
@inlineCallbacks
def check_msg(self, from_addr="+12345678", to_addr="*678#", content=None,
session_event=None, helper_metadata=None):
default_hmd = {'truteq': {'genfields': {}}}
[msg] = yield self.tx_helper.wait_for_dispatched_inbound(1)
self.assertEqual(msg['transport_name'], self.tx_helper.transport_name)
self.assertEqual(msg['transport_type'], 'ussd')
self.assertEqual(msg['transport_metadata'], {})
self.assertEqual(
msg['helper_metadata'], helper_metadata or default_hmd)
self.assertEqual(msg['from_addr'], from_addr)
self.assertEqual(msg['to_addr'], to_addr)
self.assertEqual(msg['content'], content)
self.assertEqual(msg['session_event'], session_event)
self.tx_helper.clear_dispatched_inbound()
@inlineCallbacks
def test_handle_inbound_ussd_new(self):
yield self.send(USSDMessage(msisdn='27000000000', type=c.USSD_NEW,
message='*678#', phase=c.USSD_PHASE_1))
[msg] = yield self.tx_helper.wait_for_dispatched_inbound(1)
#.........这里部分代码省略.........