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