Python User.add_friend Examples

Example #1

File: test_trip_service.py Project: Steefkroon/tripservice-kata-python

    def test_return_trips_if_users_are_friends(self, mock_get_user,
                                               mock_get_trips_from_user):
        friend_one = User()
        friend_one.add_friend(self.A_USER)
        friend_one.add_trip(Trip())
        trip_service = TripService()
        result = trip_service.get_trips_by_user(friend_one)

        self.assertEquals(len(result), 1)

Example #2

File: app.py Project: benjaminjenkins1/magic_frame

def add_friend():
    if session.get('logged_in') is None:
        return redirect('/signin')
    phone_number = re.sub('[^0-9]', '', session['phone'])
    friend_phone = re.sub('[^0-9]', '', request.form['phone-number'])
    if len(friend_phone) != 10:
        flash(
            'Oops, that didn\'t look like a phone number.<br>Please try again')
        return redirect('/people')
    if phone_number == friend_phone:
        flash('That\'s you!')
        return redirect('/people')
    user = User(phone_number)
    user.add_friend(friend_phone)
    return redirect('/people')