def setUp(self):
self.callback_populated_list = []
self.ssmi_client = SSMIClient()
self.protocol_transport = proto_helpers.StringTransport()
self.ssmi_client.app_setup('user', 'pass', self.ussd_test_callback,
None, None)
self.ssmi_client.makeConnection(self.protocol_transport)
def set_handler(self, handler):
# set the variables needed by SSMIClient so I don't have to
# specify callbacks manually for each function
self.handler = handler
self._ussd_callback = self.handler.ussd_callback
self._sms_callback = self.handler.sms_callback
self._errback = self.handler.errback
# ugh, can't do normal super() call because twisted's protocol.Factory
# is an old style class that doesn't subclass object.
SSMIClient.__init__(self)
def setUp(self):
self.callback_populated_list = []
self.ssmi_client = SSMIClient()
self.protocol_transport = proto_helpers.StringTransport()
self.ssmi_client.app_setup('user', 'pass', self.ussd_test_callback,
None, None)
self.ssmi_client.makeConnection(self.protocol_transport)
class SSMIClientTestCase(TestCase):
def setUp(self):
self.callback_populated_list = []
self.ssmi_client = SSMIClient()
self.protocol_transport = proto_helpers.StringTransport()
self.ssmi_client.app_setup('user', 'pass', self.ussd_test_callback,
None, None)
self.ssmi_client.makeConnection(self.protocol_transport)
def tearDown(self):
self.protocol_transport.loseConnection()
self.ssmi_client.updateCall.cancel()
def ussd_test_callback(self,
msisdn,
ussd_type,
phase,
message,
genfields=None):
if genfields is None:
genfields = {}
self.callback_populated_list.append({
"msisdn": msisdn,
"ussd_type": ussd_type,
"phase": phase,
"message": message,
"genfields": genfields
})
def test_default_link_check_period(self):
self.assertEqual(self.ssmi_client._link_check_period, LINKCHECK_PERIOD)
def test_non_default_link_check_period(self):
self.ssmi_client.app_setup('user', 'pass', link_check_period=15)
self.assertEqual(self.ssmi_client._link_check_period, 15)
def test_extended_ussd_msg_recieved(self):
self.ssmi_client.dataReceived(
'SSMI,111,27831112222,1,0,655011234567890:1::,*156#\r')
self.assertTrue(len(self.callback_populated_list) > 0)
list_0 = self.callback_populated_list[0]
self.assertEqual(list_0['msisdn'], '27831112222')
self.assertEqual(list_0['ussd_type'], '1')
self.assertEqual(list_0['phase'], '0')
self.assertEqual(list_0['message'], '*156#')
genfields = list_0['genfields']
self.assertEqual(genfields['IMSI'], '655011234567890')
self.assertEqual(genfields['Subscriber type'], '1')
self.assertEqual(genfields['OperatorID'], '')
self.assertEqual(genfields['SessionID'], '')
self.assertEqual(genfields['ValiPort'], '')
def test_ussd_message_recieved(self):
self.ssmi_client.dataReceived('SSMI,110,27831112222,1,0,*156#\r')
self.assertTrue(len(self.callback_populated_list) > 0)
list_0 = self.callback_populated_list[0]
self.assertEqual(list_0['msisdn'], '27831112222')
self.assertEqual(list_0['ussd_type'], '1')
self.assertEqual(list_0['phase'], '0')
self.assertEqual(list_0['message'], '*156#')
self.assertEqual(list_0['genfields'], {})
def test_multi_line_receive(self):
lines = ('SSMI,110,27831112222,1,0,*156#\r'
'SSMI,110,27831112222,1,0,*157#\r')
self.ssmi_client.dataReceived(lines)
list_0, list_1 = self.callback_populated_list
self.assertEqual(list_0['message'], '*156#')
self.assertEqual(list_1['message'], '*157#')
def test_split_line_receive(self):
self.ssmi_client.dataReceived('SSMI,110,27831112222')
self.ssmi_client.dataReceived(',1,0,*156#\r')
[list_0] = self.callback_populated_list
self.assertEqual(list_0['message'], '*156#')
def test_bad_line_receive(self):
self.ssmi_client.dataReceived('Garbage\r')
[err] = self.flushLoggedErrors(SSMIRemoteServerError)
self.assertEqual(err.value.args, ("No SSMI header. Skipping bad line"
" 'Garbage'", ))
def test_ussd_msg_with_comma(self):
self.ssmi_client.dataReceived('SSMI,110,27831112222,1,0,Hello, you.\r')
[list_0] = self.callback_populated_list
self.assertEqual(list_0['message'], 'Hello, you.')
def test_extended_ussd_msg_with_comma(self):
self.ssmi_client.dataReceived(
'SSMI,111,27831112222,1,0,655011234567890:1::,Hello, you.\r')
[list_0] = self.callback_populated_list
self.assertEqual(list_0['message'], 'Hello, you.')
def test_send_ussd_with_unicode(self):
self.ssmi_client.send_ussd('27831112222', u'Hello Zoë')
[login, ussd_cmd, blank
] = self.protocol_transport.value().split(self.ssmi_client.delimiter)
message = ussd_cmd.split(',')[-1]
self.assertTrue(isinstance(message, basestring))
self.assertEqual(message, 'Hello Zo\xc3\xab')
def connectionLost(self, *args, **kwargs):
SSMIClient.connectionLost(self, *args, **kwargs)
self.factory.onConnectionLost.callback(self)
class SSMIClientTestCase(TestCase):
def setUp(self):
self.callback_populated_list = []
self.ssmi_client = SSMIClient()
self.protocol_transport = proto_helpers.StringTransport()
self.ssmi_client.app_setup('user', 'pass', self.ussd_test_callback,
None, None)
self.ssmi_client.makeConnection(self.protocol_transport)
def tearDown(self):
self.protocol_transport.loseConnection()
self.ssmi_client.updateCall.cancel()
def ussd_test_callback(self, msisdn, ussd_type, phase, message,
genfields=None):
if genfields is None:
genfields = {}
self.callback_populated_list.append({
"msisdn": msisdn,
"ussd_type": ussd_type,
"phase": phase,
"message": message,
"genfields": genfields})
def test_default_link_check_period(self):
self.assertEqual(self.ssmi_client._link_check_period, LINKCHECK_PERIOD)
def test_non_default_link_check_period(self):
self.ssmi_client.app_setup('user', 'pass', link_check_period=15)
self.assertEqual(self.ssmi_client._link_check_period, 15)
def test_extended_ussd_msg_recieved(self):
self.ssmi_client.dataReceived(
'SSMI,111,27831112222,1,0,655011234567890:1::,*156#\r')
self.assertTrue(len(self.callback_populated_list) > 0)
list_0 = self.callback_populated_list[0]
self.assertEqual(list_0['msisdn'], '27831112222')
self.assertEqual(list_0['ussd_type'], '1')
self.assertEqual(list_0['phase'], '0')
self.assertEqual(list_0['message'], '*156#')
genfields = list_0['genfields']
self.assertEqual(genfields['IMSI'], '655011234567890')
self.assertEqual(genfields['Subscriber type'], '1')
self.assertEqual(genfields['OperatorID'], '')
self.assertEqual(genfields['SessionID'], '')
self.assertEqual(genfields['ValiPort'], '')
def test_ussd_message_recieved(self):
self.ssmi_client.dataReceived('SSMI,110,27831112222,1,0,*156#\r')
self.assertTrue(len(self.callback_populated_list) > 0)
list_0 = self.callback_populated_list[0]
self.assertEqual(list_0['msisdn'], '27831112222')
self.assertEqual(list_0['ussd_type'], '1')
self.assertEqual(list_0['phase'], '0')
self.assertEqual(list_0['message'], '*156#')
self.assertEqual(list_0['genfields'], {})
def test_multi_line_receive(self):
lines = ('SSMI,110,27831112222,1,0,*156#\r'
'SSMI,110,27831112222,1,0,*157#\r')
self.ssmi_client.dataReceived(lines)
list_0, list_1 = self.callback_populated_list
self.assertEqual(list_0['message'], '*156#')
self.assertEqual(list_1['message'], '*157#')
def test_split_line_receive(self):
self.ssmi_client.dataReceived('SSMI,110,27831112222')
self.ssmi_client.dataReceived(',1,0,*156#\r')
[list_0] = self.callback_populated_list
self.assertEqual(list_0['message'], '*156#')
def test_bad_line_receive(self):
self.ssmi_client.dataReceived('Garbage\r')
[err] = self.flushLoggedErrors(SSMIRemoteServerError)
self.assertEqual(err.value.args, ("No SSMI header. Skipping bad line"
" 'Garbage'",))
def test_ussd_msg_with_comma(self):
self.ssmi_client.dataReceived('SSMI,110,27831112222,1,0,Hello, you.\r')
[list_0] = self.callback_populated_list
self.assertEqual(list_0['message'], 'Hello, you.')
def test_extended_ussd_msg_with_comma(self):
self.ssmi_client.dataReceived(
'SSMI,111,27831112222,1,0,655011234567890:1::,Hello, you.\r')
[list_0] = self.callback_populated_list
self.assertEqual(list_0['message'], 'Hello, you.')
def test_send_ussd_with_unicode(self):
self.ssmi_client.send_ussd('27831112222', u'Hello Zoë')
[login, ussd_cmd, blank] = self.protocol_transport.value().split(
self.ssmi_client.delimiter)
message = ussd_cmd.split(',')[-1]
self.assertTrue(isinstance(message, basestring))
self.assertEqual(message, 'Hello Zo\xc3\xab')