← Back to branch summary

~jaboing/canonical-identity-provider/new_tests

« back to all changes in this revision

Viewing changes to identityprovider/tests/test_models_account.py

  • Committer: Tarmac
  • Author(s): Ricardo Kirkner
  • Date: 2013-05-06 21:12:32 UTC
  • mfrom: (833.2.3 1176960-multiple-tokens)
  • Revision ID: tarmac-20130506211232-zkn0qx1xp21g2ldd
[r=nataliabidart] cope with pre-existing tokens causing multiple hits when looking tokens up by name

Show diffs side-by-side

added added

removed removed

Lines of Context:
identityprovider/tests/test_models_account.py
554
554
        self.assertNotEqual(token, token3)
555
555
        self.assertEqual(token3.consumer.id, consumer.id)
556
556
 
 
557
    def test_get_or_create_when_multiple_results(self):
 
558
        consumer, _ = Consumer.objects.get_or_create(user=self.user)
 
559
 
 
560
        # create multiple tokens
 
561
        token1 = consumer.token_set.create(name='new-token')
 
562
        token2 = consumer.token_set.create(name='new-token')
 
563
        token2.created_at = token1.created_at + timedelta(seconds=10)
 
564
        token2.save()
 
565
 
 
566
        token, created = self.account.get_or_create_oauth_token('new-token')
 
567
 
 
568
        self.assertEqual(token, token2)
 
569
        self.assertFalse(created)
 
570
 
557
571
    def test_get_when_account_has_no_associated_consumer(self):
558
572
        Consumer.objects.filter(user=self.user).delete()
559
573