def fetch_access_token(self, *args, **kwargs):
if self.error:
raise self.error('')
response = Mock(['key', 'secret'])
response.key = self.key
response.secret = self.secret
return response
def test_bind(self, mock_gofer_agent):
context = Mock()
context.agent_id = '123'
context.secret = 'test-secret'
context.url = 'http://broker.com'
context.transport = 'qpid'
context.authenticator = Mock()
context.details = {'task_id': '4567'}
context.reply_queue = 'pulp.task'
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_consumer = Mock()
mock_agent.Consumer = Mock(return_value=mock_consumer)
bindings = []
options = {}
Consumer.bind(context, bindings, options)
mock_gofer_agent.assert_called_with(
context.agent_id,
ctag=context.reply_queue,
url=context.url,
transport=context.transport,
secret=context.secret,
authenticator=context.authenticator,
any=context.details)
mock_consumer.bind.assert_called_with(bindings, options)
def test_request_token_fake(self):
c = Mock()
c.key = self.accepted_consumer.key
c.secret = 'mom'
self.client = OAuthClient(c)
res = self.client.get(self.url)
eq_(res.status_code, 401)
def test_request_token_fake(self):
"""Try with a phony consumer key"""
c = Mock()
c.key = 'yer'
c.secret = 'mom'
r = oclient.get('oauth.request_token', c, callback=True)
eq_(r.content, 'Invalid Consumer.')
def test_request_token_fake(self):
c = Mock()
c.key = self.accepted_consumer.key
c.secret = 'mom'
self.client = OAuthClient(c)
res = self.client.get(self.url)
eq_(res.status_code, 401)
def test_unregistered(self, mock_gofer_agent):
context = Mock()
context.agent_id = "123"
context.secret = "test-secret"
context.url = "http://broker.com"
context.transport = "qpid"
context.authenticator = Mock()
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_consumer = Mock()
mock_agent.Consumer = Mock(return_value=mock_consumer)
# test capability
Consumer.unregistered(context)
# validation
mock_gofer_agent.assert_called_with(
context.agent_id,
url=context.url,
secret=context.secret,
authenticator=context.authenticator,
transport=context.transport,
async=True,
)
mock_consumer.unregistered.assert_called_with()
def test_request_token_fake(self):
c = Mock()
c.key = self.access.key
c.secret = 'mom'
ok_(not self.auth.authenticate(
Request(self.call(client=OAuthClient(c)))))
ok_(not this_thread_is_pinned())
def test_request_token_fake(self):
c = Mock()
c.key = self.access.key
c.secret = 'mom'
res = self.auth.is_authenticated(self.call(client=OAuthClient(c)))
eq_(res.status_code, 401)
eq_(json.loads(res.content)['reason'], errors['headers'])
def test_unbind(self, mock_gofer_agent):
context = Mock()
context.route = '123'
context.secret = 'test-secret'
context.url = 'http://broker.com'
context.authenticator = Mock()
context.details = {'task_id': '4567'}
context.reply_queue = 'pulp.task'
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_consumer = Mock()
mock_agent.Consumer = Mock(return_value=mock_consumer)
bindings = []
options = {}
Consumer.unbind(context, bindings, options)
mock_gofer_agent.assert_called_with(
context.url,
context.route,
reply=context.reply_queue,
secret=context.secret,
authenticator=context.authenticator,
data=context.details)
mock_consumer.unbind.assert_called_with(bindings, options)
def test_unregistered(self, mock_gofer_agent):
context = Mock()
context.agent_id = '123'
context.secret = 'test-secret'
context.url = 'http://broker.com'
context.transport = 'qpid'
context.authenticator = Mock()
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_consumer = Mock()
mock_agent.Consumer = Mock(return_value=mock_consumer)
# test capability
Consumer.unregistered(context)
# validation
mock_gofer_agent.assert_called_with(
context.agent_id,
url=context.url,
secret=context.secret,
authenticator=context.authenticator,
transport=context.transport,
async=True)
mock_consumer.unregistered.assert_called_with()
def test_request_token_fake(self):
c = Mock()
c.key = self.access.key
c.secret = 'mom'
res = self.auth.is_authenticated(self.call(client=OAuthClient(c)))
eq_(res.status_code, 401)
eq_(json.loads(res.content)['reason'], errors['headers'])
def test_unregistered(self, mock_gofer_agent):
context = Mock()
context.route = '123'
context.secret = 'test-secret'
context.url = 'http://broker.com'
context.authenticator = Mock()
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_consumer = Mock()
mock_agent.Consumer = Mock(return_value=mock_consumer)
# test capability
Consumer.unregistered(context)
# validation
mock_gofer_agent.assert_called_with(
context.url,
context.route,
secret=context.secret,
authenticator=context.authenticator,
wait=0)
mock_consumer.unregistered.assert_called_with()
def test_request_token_fake(self):
c = Mock()
c.key = self.access.key
c.secret = 'mom'
ok_(not self.auth.authenticate(
Request(self.call(client=OAuthClient(c)))))
ok_(not this_thread_is_pinned())
def test_request_token_fake(self):
"""Try with a phony consumer key"""
c = Mock()
c.key = 'yer'
c.secret = 'mom'
r = client.get('oauth.request_token', c)
eq_(r.content, 'Invalid consumer.')
def test_request_token_fake(self):
"""Try with a phony consumer key"""
c = Mock()
c.key = "yer"
c.secret = "mom"
r = client.get("oauth.request_token", c)
eq_(r.content, "Invalid consumer.")
def test_unbind(self, mock_gofer_agent):
context = Mock()
context.agent_id = "123"
context.secret = "test-secret"
context.url = "http://broker.com"
context.transport = "qpid"
context.authenticator = Mock()
context.details = {"task_id": "4567"}
context.reply_queue = "pulp.task"
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_consumer = Mock()
mock_agent.Consumer = Mock(return_value=mock_consumer)
bindings = []
options = {}
Consumer.unbind(context, bindings, options)
mock_gofer_agent.assert_called_with(
context.agent_id,
ctag=context.reply_queue,
url=context.url,
transport=context.transport,
secret=context.secret,
authenticator=context.authenticator,
any=context.details,
)
mock_consumer.unbind.assert_called_with(bindings, options)
def test_request_token_fake(self):
c = Mock()
c.key = self.access.key
c.secret = 'mom'
self.client = OAuthClient(c)
res = self.client.get(self.url)
eq_(res.status_code, 401)
eq_(json.loads(res.content)['reason'], errors['headers'])
def test_request_token_fake(self):
c = Mock()
c.key = self.access.key
c.secret = 'mom'
self.client = OAuthClient(c)
res = self.client.get(self.url)
eq_(res.status_code, 401)
eq_(json.loads(res.content)['reason'], errors['headers'])
def setUp(self):
# mock request object
request = Request()
request.method = 'GET'
request.url = 'http://example.com/'
request.headers = {}
request.params = {}
request.data = {}
request.params_and_data = {}
self.request = request
# mock consumer object
consumer = Mock()
consumer.key = '123'
consumer.secret = '456'
self.consumer = consumer
# mock token object
token = Mock()
token.key = '321'
token.secret = '456'
self.token = token
def setUp(self):
# mock request object
request = Request()
request.method = 'GET'
request.url = 'http://example.com/'
request.headers = {}
request.params = {}
request.data = {}
request.params_and_data = {}
self.request = request
# mock response object
response = Mock()
response.content = 'access_token=321'
response.headers = {'content-type': 'text/html; charset=UTF-8'}
response.ok = True
response.status_code = 200
response.raise_for_status = lambda: None
self.response = response
# mock raise_for_status with an error
def raise_for_status():
raise Exception('Response not OK!')
self.raise_for_status = raise_for_status
# mock consumer object
consumer = Mock()
consumer.key = '123'
consumer.secret = '456'
self.consumer = consumer
# mock token object
token = Mock()
token.key = '321'
token.secret = '456'
self.token = token
def test_send(self, mock_gofer_agent):
context = Mock()
context.route = '123'
context.secret = 'test-secret'
context.url = 'http://broker.com'
context.authenticator = Mock()
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_profile = Mock()
mock_agent.Profile = Mock(return_value=mock_profile)
Profile.send(context)
mock_gofer_agent.assert_called_with(
context.url,
context.route,
secret=context.secret,
authenticator=context.authenticator)
mock_profile.send.assert_called_with()
def test_send(self, mock_gofer_agent):
context = Mock()
context.agent_id = "123"
context.secret = "test-secret"
context.url = "http://broker.com"
context.transport = "qpid"
context.authenticator = Mock()
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_profile = Mock()
mock_agent.Profile = Mock(return_value=mock_profile)
Profile.send(context)
mock_gofer_agent.assert_called_with(
context.agent_id,
url=context.url,
transport=context.transport,
secret=context.secret,
authenticator=context.authenticator,
)
mock_profile.send.assert_called_with()
def test_request_token_fake(self):
c = Mock()
c.key = self.access.key
c.secret = 'mom'
ok_(not self.auth.authenticate(self.call(client=OAuthClient(c))))
def test_request_token_fake(self):
c = Mock()
c.key = self.access.key
c.secret = 'mom'
ok_(not self.auth.authenticate(self.call(client=OAuthClient(c))))
def setUp(self):
consumer = Mock(spec=oauth.OAuthConsumer)
consumer.secret = 'abc'
token = Mock(spec=oauth.OAuthToken)
token.secret = 'abc'
self.stream = Stream(consumer, token)
