def test_create_friend_request_successful(self, db_controller_name_mock):
        db_controller_mock = db_controller_name_mock.return_value
        side_effects = [mock.Mock(), MongoController.get_mongodb_instance(MONGODB_USER, MONGODB_PASSWD),
                        MongoController.get_mongodb_instance(MONGODB_USER, MONGODB_PASSWD),
                        MongoController.get_mongodb_instance(MONGODB_USER, MONGODB_PASSWD),
                        MongoController.get_mongodb_instance(MONGODB_USER, MONGODB_PASSWD),
                        MongoController.get_mongodb_instance(MONGODB_USER, MONGODB_PASSWD)]
        db_controller_mock.get_mongodb_instance = mock.Mock(side_effect=side_effects)
        FriendRequestModel.friend_request_exists = mock.MagicMock(return_value= False)

        request = FriendRequestModel.create_friend_request(1, 5, 'hello', 'picId')

        self.assertEqual(request["user_id_sender"], friends_requests_creation_successful_mock["user_id_sender"])
        self.assertEqual(request["user_id_rcv"], friends_requests_creation_successful_mock["user_id_rcv"])
        self.assertEqual(request["message"], friends_requests_creation_successful_mock["message"])
        self.assertEqual(request["picture"], friends_requests_creation_successful_mock["picture"])
Ejemplo n.º 2
0
	def _create_be_friend_request(self, user_sender_id, user_rcv_id, msg, picture):
		friend_request = FriendRequestModel.create_friend_request(user_sender_id, user_rcv_id, msg, picture)
		return friend_request