def test_add_application_allowed(self, env_get):
self.add_application = tables.AddApplication()
self.add_application.table = mock.Mock()
self.add_application.table.kwargs.get.return_value = "env_id"
env_get.return_value = {'status': 'good', 'version': '1'}
self.assertTrue(self.add_application.allowed("test", "test"))
def test_add_application_get_link_url(self, reverse):
self.add_application = tables.AddApplication()
reverse.return_value = "reversed url"
self.add_application.table = mock.Mock()
self.add_application.table.kwargs = {'environment_id': 'id'}
self.assertEqual('reversed url?next=reversed url',
self.add_application.get_link_url())