def test_handle_system_service_message_pickup(self):
system_service = mcchannel._SystemService(mcchannel._SYSTEM_SERVICE, '1')
memcache.set(mcchannel._PICKUP_MEMCACHE_PREFIX + 'abc',
{'service1': ['message1']})
yield system_service.handle_message({'pickup': 'abc'})
self.assertEquals(mcchannel._outgoing_messages,
{'1': {'service1': ['message1']}})
def test_handle_system_service_message_get_token(self):
system_service = mcchannel._SystemService(mcchannel._SYSTEM_SERVICE, '1')
yield system_service.handle_message({'getToken': None})
channel_stub = self.testbed.get_stub('channel')
token = mcchannel._outgoing_messages['1'][mcchannel._SYSTEM_SERVICE][-1][
'token']
self.assertEquals('1', channel_stub.client_id_from_token(token))
