def setUp(self):
# Reuse the consumer from piston for the tests
settings.API_AUTH_FIELDS = utils.config('fields').copy()
settings.API_AUTH_APP = "piston"
settings.API_AUTH_MODEL = "consumer"
self.c = utils.consumer_model().objects.create(
name = "Test", key = "consumer1", secret = "secret1")
def get_secret_key(self):
"""
Queries the database to find the secret key for this consumer.
"""
fields = config('fields')
ConsumerModel = consumer_model()
query = {fields['consumer_key']: self.consumer_key}
try:
q = ConsumerModel.objects.values(fields['consumer_secret'])
consumer_secret = q.get(**query)
except FieldError:
raise _AuthException("Invalid database fields.")
except ConsumerModel.DoesNotExist:
raise _AuthException("Invalid consumer key: %s." % self.consumer_key)
except ConsumerModel.MultipleObjectsReturned:
raise _AuthException(
"Invalid database. "
"Consumer keys are not unique: %s." % self.consumer_key
)
return consumer_secret[fields['consumer_secret']]
def test_secret_multiple(self):
c = utils.consumer_model().objects.create(
name = "Test", key = "consumer1", secret = "secret1")
consumer = auth.Consumer("consumerx")
self.assertRaises(auth._AuthException, consumer.get_secret_key)
c.delete()
