class AppViewsHelper(object):
implements(IHelper)
def __init__(self, conversation_type):
self.conversation_type = conversation_type
self.vumi_helper = DjangoVumiApiHelper()
self._app_helper = ApplicationHelper(
conversation_type, self.vumi_helper)
# Proxy methods from our helpers.
generate_proxies(self, self._app_helper)
generate_proxies(self, self.vumi_helper)
def setup(self):
# Create the things we need to create
self.vumi_helper.setup()
self.vumi_helper.make_django_user()
def cleanup(self):
return self.vumi_helper.cleanup()
def get_new_view_url(self):
return reverse('conversations:new_conversation')
def get_conversation_helper(self, conversation):
return ConversationViewHelper(self, conversation.key)
def create_conversation_helper(self, *args, **kw):
conversation = self.create_conversation(*args, **kw)
return self.get_conversation_helper(conversation)
def get_api_commands_sent(self):
return base_utils.connection.get_commands()
class AppViewsHelper(object):
implements(IHelper)
def __init__(self, conversation_type):
self.conversation_type = conversation_type
self.vumi_helper = DjangoVumiApiHelper()
self._app_helper = ApplicationHelper(conversation_type,
self.vumi_helper)
# Proxy methods from our helpers.
generate_proxies(self, self._app_helper)
generate_proxies(self, self.vumi_helper)
def setup(self):
# Create the things we need to create
self.vumi_helper.setup()
self.vumi_helper.make_django_user()
def cleanup(self):
return self.vumi_helper.cleanup()
def get_new_view_url(self):
return reverse('conversations:new_conversation')
def get_conversation_helper(self, conversation):
return ConversationViewHelper(self, conversation.key)
def create_conversation_helper(self, *args, **kw):
conversation = self.create_conversation(*args, **kw)
return self.get_conversation_helper(conversation)
def get_api_commands_sent(self):
return base_utils.connection.get_commands()
def __init__(self, conversation_type):
self.conversation_type = conversation_type
self.vumi_helper = DjangoVumiApiHelper()
self._app_helper = ApplicationHelper(conversation_type,
self.vumi_helper)
# Proxy methods from our helpers.
generate_proxies(self, self._app_helper)
generate_proxies(self, self.vumi_helper)
def __init__(self, router_type):
self.router_type = router_type
self.vumi_helper = DjangoVumiApiHelper()
self._router_helper = RouterHelper(
router_type, self.vumi_helper)
# Proxy methods from our helpers.
generate_proxies(self, self._router_helper)
generate_proxies(self, self.vumi_helper)
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.vumi_helper.setup_tagpool(
u'longcode', [u'default1000%s' % i for i in [1, 2, 3, 4]])
self.user_helper.add_tagpool_permission(u'longcode')
self.client = self.vumi_helper.get_client()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.command = go_manage_routing_table.Command()
self.command.stdout = StringIO()
self.command.stderr = StringIO()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.redis = self.vumi_helper.get_vumi_api().redis
self.stdout = StringIO()
self.stderr = StringIO()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.diamondash_api = FakeDiamondashApiClient()
layout = ToyDashboardLayout([{
'type': 'lvalue',
'time_range': '1d',
'title': 'Spam (24h)',
'target': {
'metric_type': 'foo',
'name': 'spam',
},
}, {
'type': 'lvalue',
'time_range': '1d',
'title': 'Ham (24h)',
'target': {
'metric_type': 'foo',
'name': 'ham',
},
}])
self.monkey_patch(client, 'get_diamondash_api',
lambda: self.diamondash_api)
self.dashboard = Dashboard('ackbar-the-dashboard', layout)
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.contact_store = self.user_helper.user_api.contact_store
self.command = go_list_accounts.Command()
self.command.stdout = StringIO()
self.command.stderr = StringIO()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.command = go_send_app_worker_command.Command()
self.command.stdout = StringIO()
self.command.stderr = StringIO()
self.command.allowed_commands = ['good_command']
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.user_pk = self.user_helper.get_django_user().pk
self.client = Client()
self.tm = DjangoTokenManager(TokenManager(
self.vumi_helper.get_redis_manager().sub_manager('token_manager')))
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.user_email = self.user_helper.get_django_user().email
self.command = go_manage_http_api.Command()
self.command.stdout = StringIO()
self.command.stderr = StringIO()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.command = go_assign_tagpool.Command()
self.command.stdout = StringIO()
self.command.stderr = StringIO()
self.vumi_helper.setup_tagpool(u"pool", [u"tag1", u"tag2"])
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.user_email = self.user_helper.get_django_user().email
self.command = go_start_conversation.Command()
self.command.stdout = StringIO()
self.command.stderr = StringIO()
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.diamondash_api = FakeDiamondashApiClient()
self.monkey_patch(client, 'get_diamondash_api',
lambda: self.diamondash_api)
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.user = self.user_helper.get_django_user()
self.tag_pool = TagPool(name=u"pool", description=u"description")
self.tag_pool.save()
self.account = Account.objects.get(user=self.user)
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.contact_store = self.user_helper.user_api.contact_store
self.command = go_import_contacts.Command()
self.command.stdout = StringIO()
self.command.stderr = StringIO()
self.vumi_helper.setup_tagpool(u"pool", [u"tag1", u"tag2"])
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper(),
setup_vumi_api=False)
self.monkey_patch(go.base.utils, 'get_conversation_pkg',
self._get_conversation_pkg)
self.vumi_helper.patch_config(
VUMI_INSTALLED_APPS=DUMMY_CONVERSATION_SETTINGS)
self.vumi_helper.setup_vumi_api()
self.user_helper = self.vumi_helper.make_django_user()
self.client = self.vumi_helper.get_client()
def __init__(self, conversation_type):
self.conversation_type = conversation_type
self.vumi_helper = DjangoVumiApiHelper()
self._app_helper = ApplicationHelper(
conversation_type, self.vumi_helper)
# Proxy methods from our helpers.
generate_proxies(self, self._app_helper)
generate_proxies(self, self.vumi_helper)
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_api = self.vumi_helper.make_django_user().user_api
conv_store = self.user_api.conversation_store
self.old_conv_model = conv_store.manager.proxy(ConversationV1)
self.command = go_migrate_conversations.Command()
self.command.stdout = StringIO()
self.command.stderr = StringIO()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.command = go_manage_application.Command()
self.vumi_helper.patch_config(VUMI_INSTALLED_APPS={
'go.apps.dummy': {
'namespace': 'dummy',
'display_name': 'dummy'
}
})
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.superuser_helper = self.vumi_helper.make_django_user(
superuser=True, email='*****@*****.**')
self.ordinary_user_helper = self.vumi_helper.make_django_user(
email='*****@*****.**')
self.target_user_helper = self.vumi_helper.make_django_user(
email='*****@*****.**')
self.superuser_client = self.vumi_helper.get_client(
username='******')
self.ordinary_user_client = self.vumi_helper.get_client(
username='******')
class RouterViewsHelper(object):
implements(IHelper)
def __init__(self, router_type):
self.router_type = router_type
self.vumi_helper = DjangoVumiApiHelper()
self._router_helper = RouterHelper(
router_type, self.vumi_helper)
# Proxy methods from our helpers.
generate_proxies(self, self._router_helper)
generate_proxies(self, self.vumi_helper)
def setup(self):
self.vumi_helper.setup()
self.vumi_helper.make_django_user()
def cleanup(self):
return self.vumi_helper.cleanup()
def get_new_view_url(self):
return reverse('routers:new_router')
def get_router_helper(self, router):
return self.get_router_helper_by_key(router.key)
def create_router_helper(self, *args, **kw):
router = self.create_router(*args, **kw)
return self.get_router_helper(router)
def get_router_helper_by_key(self, router_key):
return RouterViewHelper(self, router_key)
def get_api_commands_sent(self):
return base_utils.connection.get_commands()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.conv = self.user_helper.create_conversation(u'dummy')
self.layout = ConversationReportsLayout(self.conv)
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper(),
setup_vumi_api=False)
self.factory = RequestFactory()
self.mw = ResponseTimeMiddleware()
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 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()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.command = go_create_user.Command()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user = self.vumi_helper.make_django_user().get_django_user()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.superuser_helper = self.vumi_helper.make_django_user(
email='*****@*****.**', superuser=True)
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.user_helper = self.vumi_helper.make_django_user()
self.contact_store = self.user_helper.user_api.contact_store
self.command = go_manage_contact_group.Command()
def setUp(self):
self.vumi_helper = self.add_helper(DjangoVumiApiHelper())
self.command = go_setup_env.Command()
self.command.stdout = StringIO()
self.command.stderr = StringIO()
# do whatever setup command.handle() does manually
self.command.setup_backend(self.vumi_helper.mk_config({}))
self.tagpool = self.command.tagpool
self.command.file_name_template = 'go_%(file_name)s.%(suffix)s'
self.command.dest_dir = 'setup_env'
self.command.config = {
'redis_manager': {
'key_prefix': 'test'
},
'riak_manager': {
'bucket_prefix': 'test.'
}
}
self.command.contact_group_info = []
self.command.conversation_info = []
self.command.router_info = []
self.command.transport_names = []
self.command.router_names = []
self.command.application_names = []
self.tagpool_file = tmp_yaml_file([
'pools:',
' pool1:',
' tags: "[\'default%d\' % i for i in range(10)]"',
' metadata:',
' display_name: "Pool 1"',
' pool2:',
' tags: ["a", "b", "c"]',
' metadata:',
' display_name: "Pool 2"',
])
self.workers_file = tmp_yaml_file([
'transports:',
' sms_transport:',
' class: vumi.transports.telnet.TelnetServerTransport',
' config:',
' telnet_port: 8080',
' ussd_transport:',
' class: vumi.transports.telnet.TelnetServerTransport',
' config:',
' telnet_port: 8081',
'',
'routers:',
' keyword:',
' class: go.routers.keyword.vumi_app.KeywordRouter',
' config:',
' redis_manager: {}',
' riak_manager: {}',
'',
'applications:',
' bulk_message:',
' class: go.apps.bulk_message.vumi_app.BulkMessageApplication',
' config:',
' redis_manager: {}',
' riak_manager: {}',
])
self.account_1_file = tmp_yaml_file([
'account:',
' email: "*****@*****.**"',
' password: foo',
' first_name: First',
' last_name: Last',
' applications:',
' - go.apps.surveys',
' - go.apps.bulk_message',
' tagpools:',
' - ["pool1", 10]',
' - ["pool2", 15]',
'',
'channels:',
' - "pool1:default0"',
' - "pool1:default1"',
'',
'conversations:',
' - key: "conv1"',
' conversation_type: survey',
' name: foo',
' config:',
' foo: bar',
' - key: conv2',
' conversation_type: wikipedia',
' name: Wikipedia',
' config: {}',
'',
'routers:',
' - key: "router1"',
' router_type: keyword',
' name: foo',
' config:',
' keyword_endpoint_mapping:',
' foo: keyword_foo',
'',
'routing_entries:',
' - ["conv1", "default", "pool1:default0", "default"]',
' - ["pool1:default0", "default", "conv1", "default"]',
' - ["router1:INBOUND", "default", "pool1:default1", "default"]',
' - ["pool1:default1", "default", "router1:INBOUND", "default"]',
' - ["conv2", "default", "router1:OUTBOUND", "default"]',
' - ["router1:OUTBOUND", "default", "conv2", "default"]',
'',
'contact_groups:',
' - key: group1',
' name: group1',
' contacts_csv: contacts.csv',
])
self.account_2_file = tmp_yaml_file([
'account:',
' email: "*****@*****.**"',
' password: bar',
' applications:',
' - go.apps.bulk_message',
' - go.apps.jsbox',
' tagpools:',
' - ["pool1", null]',
' - ["pool2", null]',
'',
'conversations:',
' - key: "conversation-key-2"',
' conversation_type: bulk_message',
' name: oof',
' config:',
' oof: rab',
'',
'contact_groups:',
' - key: group2',
' name: group2',
' contacts_csv: contacts.csv',
])
