def test_clean_url(self): from oauth2_provider.provider import AuthorizationProvider provider = AuthorizationProvider() verify_client_id = Mock(return_value = True) verify_redirect_uri = Mock(return_value = True) verify_scope = Mock(return_value = True) save_data = Mock() provider.verify_client_id = verify_client_id provider.verify_redirect_uri = verify_redirect_uri provider.verify_scope = verify_scope provider.save_data = save_data provider.generate_authorization_code = Mock(return_value='bar') redirect_uri = 'http://google.com?this=bad&so=is_this' client_id = 'client_id' scope = 'foo,bar' state = 'omgz' results = provider.verify_auth_request('code', client_id, redirect_uri=redirect_uri, scope=scope, state=state) save_data.assert_called_with(client_id, 'bar', scope, redirect_uri) assert state in results['redirect_uri'] assert not 'this' in results['redirect_uri'] assert not 'bad' in results['redirect_uri'] assert not 'so' in results['redirect_uri']
def test_save_data(self): from oauth2_provider.provider import AuthorizationProvider provider = AuthorizationProvider() with self.assertRaises(NotImplementedError): provider.save_data('foo', 'bar', 'baz', 'boom!')