class TestRoutingScreen(GoDjangoTestCase):
# Most of the functionality of this view lives in JS, so we just test that
# we're correctly injecting initial state into the template.
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.client = self.vumi_helper.get_client()
self.mock_rpc = MockRpc()
self.add_cleanup(self.mock_rpc.tearDown)
def make_routing_table(self, channels=(), conversations=(), routing=()):
routing_table = {
u'campaign_id': self.user_helper.account_key,
u'channels': [],
u'conversations': [],
u'routers': [],
u'routing_entries': [],
}
for channel in channels:
routing_table[u'channels'].append(
ChannelType.format_channel(channel))
for conv in conversations:
routing_table[u'conversations'].append(
ConversationType.format_conversation(conv))
# TODO: routers
def mkconn(thing):
if isinstance(thing, tuple):
# It's a tuple, so assume it's a tag.
return u'TRANSPORT_TAG:%s:%s' % thing
else:
# Assume it's a conversation.
return u'CONVERSATION:%s:%s' % (
thing.conversation_type, thing.key)
for src, dst in routing:
routing_table[u'routing_entries'].append(
RoutingEntryType.format_entry(
(mkconn(src), 'default'), (mkconn(dst), 'default')))
return routing_table
def check_model_data(self, response, routing_table):
model_data = response.context['model_data']
self.assertEqual(
json.loads(model_data),
json.loads(json.dumps(routing_table)))
self.assertContains(response, model_data)
def check_api_request(self):
request = self.mock_rpc.request
self.assertEqual(request['method'], 'routing_table')
self.assertEqual(request['params'], [self.user_helper.account_key])
def test_empty_routing(self):
routing_table = self.make_routing_table()
self.mock_rpc.set_response(result=routing_table)
response = self.client.get(reverse('routing'))
self.check_api_request()
self.check_model_data(response, routing_table)
self.assertContains(response, reverse('channels:new_channel'))
self.assertContains(
response, reverse('conversations:new_conversation'))
def test_non_empty_routing(self):
conv = self.user_helper.create_conversation(u'bulk_message')
tag = (u'pool', u'tag')
routing_table = self.make_routing_table(
channels=[self.user_helper.user_api.get_channel(tag)],
conversations=[conv], routing=[(conv, tag), (tag, conv)])
self.mock_rpc.set_response(result=routing_table)
response = self.client.get(reverse('routing'))
self.check_api_request()
self.check_model_data(response, routing_table)
self.assertContains(response, reverse('channels:new_channel'))
self.assertContains(
response, reverse('conversations:new_conversation'))
class TestDialogueViews(GoDjangoTestCase):
def setUp(self):
self.app_helper = self.add_helper(AppViewsHelper(u'dialogue'))
self.client = self.app_helper.get_client()
self.mock_rpc = MockRpc()
self.add_cleanup(self.mock_rpc.tearDown)
def setup_conversation(self, with_group=True, with_channel=True, **kw):
groups = []
if with_group:
groups.append(
self.app_helper.create_group_with_contacts(u'test_group', 0))
channel = None
if with_channel:
channel = self.app_helper.create_channel(
supports_generic_sends=True)
return self.app_helper.create_conversation_helper(
channel=channel, groups=groups, **kw)
def test_action_send_dialogue_get(self):
conv_helper = self.setup_conversation(started=True)
response = self.client.get(
conv_helper.get_action_view_url('send_jsbox'))
self.assertEqual([], self.app_helper.get_api_commands_sent())
self.assertContains(response, '<h1>Send Dialogue</h1>')
def test_action_send_dialogue_post(self):
conv_helper = self.setup_conversation(started=True)
response = self.client.post(
conv_helper.get_action_view_url('send_jsbox'), {}, follow=True)
self.assertRedirects(response, conv_helper.get_view_url('show'))
[send_jsbox_cmd] = self.app_helper.get_api_commands_sent()
conversation = conv_helper.get_conversation()
self.assertEqual(send_jsbox_cmd, VumiApiCommand.command(
'%s_application' % (conversation.conversation_type,),
'send_jsbox',
user_account_key=conversation.user_account.key,
conversation_key=conversation.key,
batch_id=conversation.batch.key))
def test_action_send_dialogue_no_group(self):
conv_helper = self.setup_conversation(started=True, with_group=False)
response = self.client.post(
conv_helper.get_action_view_url('send_jsbox'), {}, follow=True)
self.assertRedirects(response, conv_helper.get_view_url('show'))
[msg] = response.context['messages']
self.assertEqual(
str(msg), "Action disabled: This action needs a contact group.")
self.assertEqual([], self.app_helper.get_api_commands_sent())
def test_action_send_dialogue_not_running(self):
conv_helper = self.setup_conversation(started=False)
response = self.client.post(
conv_helper.get_action_view_url('send_jsbox'), {},
follow=True)
self.assertRedirects(response, conv_helper.get_view_url('show'))
[msg] = response.context['messages']
self.assertEqual(
str(msg),
"Action disabled: This action needs a running conversation.")
self.assertEqual([], self.app_helper.get_api_commands_sent())
def test_action_send_dialogue_no_channel(self):
conv_helper = self.setup_conversation(started=True, with_channel=False)
response = self.client.post(
conv_helper.get_action_view_url('send_jsbox'), {}, follow=True)
self.assertRedirects(response, conv_helper.get_view_url('show'))
[msg] = response.context['messages']
self.assertEqual(
str(msg),
"Action disabled: This action needs channels capable"
" of sending messages attached to this conversation.")
self.assertEqual([], self.app_helper.get_api_commands_sent())
def test_show_stopped(self):
"""
Test showing the conversation
"""
conv_helper = self.setup_conversation(started=False, name=u"myconv")
response = self.client.get(conv_helper.get_view_url('show'))
conversation = response.context[0].get('conversation')
self.assertEqual(conversation.name, u"myconv")
self.assertNotContains(
response, conv_helper.get_action_view_url('send_jsbox'))
def test_show_running(self):
"""
Test showing the conversation
"""
conv_helper = self.setup_conversation(started=True, name=u"myconv")
response = self.client.get(conv_helper.get_view_url('show'))
conversation = response.context[0].get('conversation')
self.assertEqual(conversation.name, u"myconv")
self.assertContains(response,
conv_helper.get_action_view_url('send_jsbox'))
def test_edit(self):
conv_helper = self.setup_conversation(
with_group=False, started=False, name=u"myconv")
conversation = conv_helper.get_conversation()
group1 = self.app_helper.create_group(u'group1')
group2 = self.app_helper.create_group(u'group2')
conversation.add_group(group1)
poll = {
'groups': [{
'key': '123',
'name': 'foo'
}]
}
self.mock_rpc.set_response(result={"poll": poll})
response = self.client.get(conv_helper.get_view_url('edit'))
self.assertContains(response, u"myconv")
self.assertContains(response, 'diagram')
self.assertContains(response, 'USSD')
self.assertContains(response, 'SMS')
self.assertContains(response, 'Google Talk')
self.assertContains(response, 'Mxit')
self.assertContains(response, 'WeChat')
self.assertContains(response, 'Twitter')
expected = poll.copy()
expected.update({
'campaign_id': conversation.user_account.key,
'conversation_key': conversation.key,
'urls': {"show": conv_helper.get_view_url('show')},
'groups': [group1.get_data(), group2.get_data()]
})
model_data = response.context["model_data"]
self.assertEqual(json.loads(model_data), expected)
def test_export_user_data(self):
conv_helper = self.setup_conversation()
response = self.client.get(conv_helper.get_view_url('user_data'))
self.assertEqual(response['Content-Type'], 'application/csv')
self.assertEqual(response.content, "TODO: write data export.")
class TestDialogueViews(GoDjangoTestCase):
def setUp(self):
self.app_helper = self.add_helper(AppViewsHelper(u'dialogue'))
self.client = self.app_helper.get_client()
self.mock_rpc = MockRpc()
self.add_cleanup(self.mock_rpc.tearDown)
def setup_conversation(self, with_group=True, with_channel=True, **kw):
groups = []
if with_group:
groups.append(
self.app_helper.create_group_with_contacts(u'test_group', 0))
channel = None
if with_channel:
channel = self.app_helper.create_channel(
supports_generic_sends=True)
return self.app_helper.create_conversation_helper(channel=channel,
groups=groups,
**kw)
def test_action_send_dialogue_get(self):
conv_helper = self.setup_conversation(started=True)
response = self.client.get(
conv_helper.get_action_view_url('send_jsbox'))
self.assertEqual([], self.app_helper.get_api_commands_sent())
self.assertContains(response, '<h1>Send Dialogue</h1>')
def test_action_send_dialogue_post(self):
conv_helper = self.setup_conversation(started=True)
response = self.client.post(
conv_helper.get_action_view_url('send_jsbox'), {}, follow=True)
self.assertRedirects(response, conv_helper.get_view_url('show'))
[send_jsbox_cmd] = self.app_helper.get_api_commands_sent()
conversation = conv_helper.get_conversation()
self.assertEqual(
send_jsbox_cmd,
VumiApiCommand.command(
'%s_application' % (conversation.conversation_type, ),
'send_jsbox',
user_account_key=conversation.user_account.key,
conversation_key=conversation.key,
batch_id=conversation.batch.key))
def test_action_send_dialogue_no_group(self):
conv_helper = self.setup_conversation(started=True, with_group=False)
response = self.client.post(
conv_helper.get_action_view_url('send_jsbox'), {}, follow=True)
self.assertRedirects(response, conv_helper.get_view_url('show'))
[msg] = response.context['messages']
self.assertEqual(
str(msg), "Action disabled: This action needs a contact group.")
self.assertEqual([], self.app_helper.get_api_commands_sent())
def test_action_send_dialogue_not_running(self):
conv_helper = self.setup_conversation(started=False)
response = self.client.post(
conv_helper.get_action_view_url('send_jsbox'), {}, follow=True)
self.assertRedirects(response, conv_helper.get_view_url('show'))
[msg] = response.context['messages']
self.assertEqual(
str(msg),
"Action disabled: This action needs a running conversation.")
self.assertEqual([], self.app_helper.get_api_commands_sent())
def test_action_send_dialogue_no_channel(self):
conv_helper = self.setup_conversation(started=True, with_channel=False)
response = self.client.post(
conv_helper.get_action_view_url('send_jsbox'), {}, follow=True)
self.assertRedirects(response, conv_helper.get_view_url('show'))
[msg] = response.context['messages']
self.assertEqual(
str(msg), "Action disabled: This action needs channels capable"
" of sending messages attached to this conversation.")
self.assertEqual([], self.app_helper.get_api_commands_sent())
def test_show_stopped(self):
"""
Test showing the conversation
"""
conv_helper = self.setup_conversation(started=False, name=u"myconv")
response = self.client.get(conv_helper.get_view_url('show'))
conversation = response.context[0].get('conversation')
self.assertEqual(conversation.name, u"myconv")
self.assertNotContains(response,
conv_helper.get_action_view_url('send_jsbox'))
def test_show_running(self):
"""
Test showing the conversation
"""
conv_helper = self.setup_conversation(started=True, name=u"myconv")
response = self.client.get(conv_helper.get_view_url('show'))
conversation = response.context[0].get('conversation')
self.assertEqual(conversation.name, u"myconv")
self.assertContains(response,
conv_helper.get_action_view_url('send_jsbox'))
def test_edit(self):
conv_helper = self.setup_conversation(with_group=False,
started=False,
name=u"myconv")
conversation = conv_helper.get_conversation()
group1 = self.app_helper.create_group(u'group1')
group2 = self.app_helper.create_group(u'group2')
conversation.add_group(group1)
poll = {'groups': [{'key': '123', 'name': 'foo'}]}
self.mock_rpc.set_response(result={"poll": poll})
response = self.client.get(conv_helper.get_view_url('edit'))
self.assertContains(response, u"myconv")
self.assertContains(response, 'diagram')
self.assertContains(response, 'USSD')
self.assertContains(response, 'SMS')
self.assertContains(response, 'Google Talk')
self.assertContains(response, 'Mxit')
self.assertContains(response, 'WeChat')
self.assertContains(response, 'Twitter')
expected = poll.copy()
expected.update({
'campaign_id': conversation.user_account.key,
'conversation_key': conversation.key,
'urls': {
"show": conv_helper.get_view_url('show')
},
'groups': [group1.get_data(), group2.get_data()]
})
model_data = response.context["model_data"]
self.assertEqual(json.loads(model_data), expected)
def test_export_user_data(self):
conv_helper = self.setup_conversation()
response = self.client.get(conv_helper.get_view_url('user_data'))
self.assertEqual(response['Content-Type'], 'application/csv')
self.assertEqual(response.content, "TODO: write data export.")
class TestRoutingScreen(GoDjangoTestCase):
# Most of the functionality of this view lives in JS, so we just test that
# we're correctly injecting initial state into the template.
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.client = self.vumi_helper.get_client()
self.mock_rpc = MockRpc()
self.add_cleanup(self.mock_rpc.tearDown)
def make_routing_table(self, channels=(), conversations=(), routing=()):
routing_table = {
u'campaign_id': self.user_helper.account_key,
u'channels': [],
u'conversations': [],
u'routers': [],
u'routing_entries': [],
}
for channel in channels:
routing_table[u'channels'].append(
ChannelType.format_channel(channel))
for conv in conversations:
routing_table[u'conversations'].append(
ConversationType.format_conversation(conv))
# TODO: routers
def mkconn(thing):
if isinstance(thing, tuple):
# It's a tuple, so assume it's a tag.
return u'TRANSPORT_TAG:%s:%s' % thing
else:
# Assume it's a conversation.
return u'CONVERSATION:%s:%s' % (thing.conversation_type,
thing.key)
for src, dst in routing:
routing_table[u'routing_entries'].append(
RoutingEntryType.format_entry((mkconn(src), 'default'),
(mkconn(dst), 'default')))
return routing_table
def check_model_data(self, response, routing_table):
model_data = response.context['model_data']
self.assertEqual(json.loads(model_data),
json.loads(json.dumps(routing_table)))
self.assertContains(response, model_data)
def check_api_request(self):
request = self.mock_rpc.request
self.assertEqual(request['method'], 'routing_table')
self.assertEqual(request['params'], [self.user_helper.account_key])
def test_empty_routing(self):
routing_table = self.make_routing_table()
self.mock_rpc.set_response(result=routing_table)
response = self.client.get(reverse('routing'))
self.check_api_request()
self.check_model_data(response, routing_table)
self.assertContains(response, reverse('channels:new_channel'))
self.assertContains(response,
reverse('conversations:new_conversation'))
def test_non_empty_routing(self):
conv = self.user_helper.create_conversation(u'bulk_message')
tag = (u'pool', u'tag')
routing_table = self.make_routing_table(
channels=[self.user_helper.user_api.get_channel(tag)],
conversations=[conv],
routing=[(conv, tag), (tag, conv)])
self.mock_rpc.set_response(result=routing_table)
response = self.client.get(reverse('routing'))
self.check_api_request()
self.check_model_data(response, routing_table)
self.assertContains(response, reverse('channels:new_channel'))
self.assertContains(response,
reverse('conversations:new_conversation'))