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())