Function parse_function_params

dossier-py/src/function.rs:151–193

()

    #[test]
    fn parse_function_params() {
        let source = indoc! {r#"
            def foo(bar, baz: int) -> bool:
                pass
        "#};

        let mut ctx = ParserContext::new(Path::new("test.py"), source);
        let tree = crate::init_parser().parse(source, None).unwrap();
        let mut cursor = tree.root_node().walk();
        cursor.goto_first_child();

        let symbol = Function::parse_symbol(cursor.node(), &mut ctx).unwrap();
        assert_eq!(symbol.fqn.as_deref(), Some("test.py::foo"));

        let function = symbol.as_function().unwrap();
        assert_eq!(function.title, "foo");

        let params = function.parameters().collect::<Vec<_>>();
        assert_eq!(params.len(), 2);

        let param = params[0].as_parameter().unwrap();
        assert_eq!(params[0].context, Some(SymbolContext::Parameter));
        assert_eq!(params[0].fqn.as_deref(), Some("test.py::foo::bar"));
        assert_eq!(param.title, "bar");
        assert_eq!(param.the_type(), None);

        let param = params[1].as_parameter().unwrap();
        assert_eq!(params[1].context, Some(SymbolContext::Parameter));
        assert_eq!(params[1].fqn.as_deref(), Some("test.py::foo::baz"));
        assert_eq!(param.title, "baz");
        let the_type = param.the_type();
        assert_eq!(
            the_type.unwrap().as_type().unwrap(),
            &Type::BuiltIn("int".to_owned())
        );

        let return_type = function.return_type().unwrap();
        assert_eq!(return_type.context, Some(SymbolContext::ReturnType));
        assert_eq!(
            return_type.as_type().unwrap(),
            &Type::BuiltIn("bool".to_owned())
        );
    }
}