def test_get_provider_login_url_no_redirect_url(self, mock_tpa, mock_next_login_url): request = mock.Mock() provider_id = 'anything' login_url = get_provider_login_url(request, provider_id) assert mock_tpa.pipeline.get_login_url.return_value == login_url mock_tpa.pipeline.get_login_url.assert_called_once_with( provider_id, mock_tpa.pipeline.AUTH_ENTRY_LOGIN, redirect_url=mock_next_login_url.return_value, ) mock_next_login_url.assert_called_once_with(request)
def test_get_provider_login_url_with_redirect_url(self, mock_tpa, mock_next_login_url): request = mock.Mock() provider_id = 'anything' redirect_url = 'the-next-url' login_url = get_provider_login_url(request, provider_id, redirect_url=redirect_url) assert mock_tpa.pipeline.get_login_url.return_value == login_url mock_tpa.pipeline.get_login_url.assert_called_once_with( provider_id, mock_tpa.pipeline.AUTH_ENTRY_LOGIN, redirect_url=redirect_url, ) self.assertFalse(mock_next_login_url.called)