def test_disconnect_with_partial_pipeline(self):
self.strategy.set_settings({
'SOCIAL_AUTH_DISCONNECT_PIPELINE': (
'social.pipeline.partial.save_status_to_session',
'social.tests.pipeline.ask_for_password',
'social.tests.pipeline.set_password',
'social.pipeline.disconnect.allowed_to_disconnect',
'social.pipeline.disconnect.get_entries',
'social.pipeline.disconnect.revoke_tokens',
'social.pipeline.disconnect.disconnect'
)
})
self.do_login()
user = User.get(self.expected_username)
redirect = do_disconnect(self.backend, user)
url = self.strategy.build_absolute_uri('/password')
expect(redirect.url).to.equal(url)
HTTPretty.register_uri(HTTPretty.GET, redirect.url, status=200,
body='foobar')
HTTPretty.register_uri(HTTPretty.POST, redirect.url, status=200)
password = '******'
requests.get(url)
requests.post(url, data={'password': password})
data = parse_qs(HTTPretty.last_request.body)
expect(data['password']).to.equal(password)
self.strategy.session_set('password', data['password'])
redirect = do_disconnect(self.backend, user)
expect(len(user.social)).to.equal(0)
def test_disconnect_with_partial_pipeline(self):
self.strategy.set_settings({
'SOCIAL_AUTH_DISCONNECT_PIPELINE':
('social.pipeline.partial.save_status_to_session',
'social.tests.pipeline.ask_for_password',
'social.tests.pipeline.set_password',
'social.pipeline.disconnect.allowed_to_disconnect',
'social.pipeline.disconnect.get_entries',
'social.pipeline.disconnect.revoke_tokens',
'social.pipeline.disconnect.disconnect')
})
self.do_login()
user = User.get(self.expected_username)
redirect = do_disconnect(self.backend, user)
url = self.strategy.build_absolute_uri('/password')
self.assertEqual(redirect.url, url)
HTTPretty.register_uri(HTTPretty.GET,
redirect.url,
status=200,
body='foobar')
HTTPretty.register_uri(HTTPretty.POST, redirect.url, status=200)
password = '******'
requests.get(url)
requests.post(url, data={'password': password})
data = parse_qs(HTTPretty.last_request.body)
self.assertEqual(data['password'], password)
self.strategy.session_set('password', data['password'])
redirect = do_disconnect(self.backend, user)
self.assertEqual(len(user.social), 0)
def test_revoke_token(self):
self.strategy.set_settings(
{'SOCIAL_AUTH_REVOKE_TOKENS_ON_DISCONNECT': True})
self.do_login()
user = User.get(self.expected_username)
user.password = '******'
HTTPretty.register_uri(self._method(self.backend.REVOKE_TOKEN_METHOD),
self.backend.REVOKE_TOKEN_URL,
status=200)
do_disconnect(self.backend, user)
def test_disconnect_with_association_id(self):
self.do_login()
user = User.get(self.expected_username)
user.password = '******'
association_id = user.social[0].id
second_usa = TestUserSocialAuth(user, user.social[0].provider, "uid2")
self.assertEqual(len(user.social), 2)
do_disconnect(self.backend, user, association_id)
self.assertEqual(len(user.social), 1)
self.assertEqual(user.social[0], second_usa)
def test_disconnect_with_association_id(self):
self.do_login()
user = User.get(self.expected_username)
user.password = '******'
association_id = user.social[0].id
second_usa = TestUserSocialAuth(user, user.social[0].provider, "uid2")
self.assertEqual(len(user.social), 2)
do_disconnect(self.backend, user, association_id)
self.assertEqual(len(user.social), 1)
self.assertEqual(user.social[0], second_usa)
def test_revoke_token(self):
self.strategy.set_settings({
'SOCIAL_AUTH_REVOKE_TOKENS_ON_DISCONNECT': True
})
self.do_login()
user = User.get(self.expected_username)
user.password = '******'
HTTPretty.register_uri(self._method(self.backend.REVOKE_TOKEN_METHOD),
self.backend.REVOKE_TOKEN_URL,
status=200)
do_disconnect(self.backend, user)
def test_disconnect(self):
self.do_login()
user = User.get(self.expected_username)
user.password = '******'
do_disconnect(self.backend, user)
self.assertEqual(len(user.social), 0)
def test_not_allowed_to_disconnect(self):
self.do_login()
user = User.get(self.expected_username)
with self.assertRaises(NotAllowedToDisconnect):
do_disconnect(self.backend, user)
def test_disconnect(self):
self.do_login()
user = User.get(self.expected_username)
user.password = '******'
do_disconnect(self.backend, user)
expect(len(user.social)).to.equal(0)
def test_not_allowed_to_disconnect(self):
self.do_login()
user = User.get(self.expected_username)
do_disconnect.when.called_with(self.backend, user).should.throw(
NotAllowedToDisconnect
)
def test_not_allowed_to_disconnect(self):
self.do_login()
user = User.get(self.expected_username)
do_disconnect.when.called_with(
self.backend, user).should.throw(NotAllowedToDisconnect)
def test_not_allowed_to_disconnect(self):
self.do_login()
user = User.get(self.expected_username)
with self.assertRaises(NotAllowedToDisconnect):
do_disconnect(self.backend, user)
def test_disconnect(self):
self.do_login()
user = User.get(self.expected_username)
user.password = '******'
do_disconnect(self.strategy, user)
expect(len(user.social)).to.equal(0)
