def test_lookup_token_registered(self, mock_lookup_user): # Lookup token for a user with registered email # basket returns response with data, lookup_basket_token returns the token user = User(email='*****@*****.**') profile = UserProfile(user=user) mock_lookup_user.return_value = {'status': 'ok', 'token': 'FAKETOKEN'} result = profile.lookup_basket_token() eq_('FAKETOKEN', result)
def test_lookup_token_unregistered(self, mock_lookup_user): # Lookup token for a user with no registered email # Basket raises unknown user exception, then lookup-token returns None user = User(email='*****@*****.**') profile = UserProfile(user=user) mock_lookup_user.side_effect = basket.BasketException( code=basket.errors.BASKET_UNKNOWN_EMAIL) result = profile.lookup_basket_token() ok_(result is None)
def test_lookup_token_exceptions(self, mock_lookup_user): # If basket raises any exception other than BASKET_UNKNOWN_EMAIL when # we call lookup_basket_token, lookup_basket_token passes it up the chain class SomeException(Exception): pass user = User(email='*****@*****.**') profile = UserProfile(user=user) mock_lookup_user.side_effect = SomeException with self.assertRaises(SomeException): profile.lookup_basket_token()