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()