more resolver tests

Author: abroooo

src/resolver/tests/resolve_expressions_tests.rs

@@ -5960,7 +5960,45 @@ fn global_namespace_operator_is_not_resolved() {
 }
 
 #[test]
-fn is_this_there() {
+fn this_in_assignments_in_methods() {
+    let id_provider = IdProvider::default();
+    let (unit, mut index) = index_with_ids(
+        "
+        FUNCTION_BLOCK fb
+            VAR
+                myvar : INT;
+            END_VAR
+            METHOD foo : INT
+                this^.myvar := 8;
+                myvar := this^.myvar;
+            END_FUNCTION_BLOCK
+        END_FUNCTION_BLOCK
+        ",
+        id_provider.clone(),
+    );
+
+    let annotations = annotate_with_ids(&unit, &mut index, id_provider);
+    let AstStatement::Assignment(statement_1) = unit.implementations[0].statements[0].get_stmt() else {
+        unreachable!()
+    };
+    let AstStatement::Assignment(statement_2) = unit.implementations[0].statements[1].get_stmt() else {
+        unreachable!()
+    };
+    assert!(index.find_type("fb.__THIS").is_some());
+    assert_type_and_hint!(&annotations, &index, &statement_1.left, "INT", None);
+    assert_type_and_hint!(&annotations, &index, &statement_2.right, "INT", Some("INT"));
+    let AstStatement::ReferenceExpr(ReferenceExpr { base: Some(deref), .. }) = statement_1.left.get_stmt()
+    else {
+        unreachable!();
+    };
+    let AstStatement::ReferenceExpr(ReferenceExpr { base: Some(this), .. }) = deref.get_stmt() else {
+        unreachable!();
+    };
+    assert_type_and_hint!(&annotations, &index, this, "fb.__THIS", None);
+}
+
+#[test]
+fn this_in_assignments() {
     let id_provider = IdProvider::default();
     let (unit, mut index) = index_with_ids(
         "
@@ -6131,5 +6169,71 @@ fn this_call() {
     let statement = &unit.implementations[0].statements[0];
     dbg!(&statement);
     assert!(index.find_type("fb.__THIS").is_some());
-    assert_type_and_hint!(&annotations, &index, statement, "fb.__THIS", None);
+    // assert_type_and_hint!(&annotations, &index, statement, "fb.__THIS", None);
+}
+
+#[test]
+fn this_as_function_parameter() {
+    let id_provider = IdProvider::default();
+    let (unit, mut index) = index_with_ids(
+        "
+        FUNCTION_BLOCK FB_Test
+            foo2(this);
+        END_FUNCTION_BLOCK
+        ",
+        id_provider.clone(),
+    );
+
+    let annotations = annotate_with_ids(&unit, &mut index, id_provider);
+    let statement = &unit.implementations[0].statements[0];
+    let AstStatement::CallStatement(CallStatement { parameters: Some(param), .. }) = statement.get_stmt()
+    else {
+        unreachable!();
+    };
+    let Some(StatementAnnotation::Value { resulting_type, .. }) = annotations.get(param) else {
+        unreachable!()
+    };
+    assert_eq!(resulting_type, "FB_Test.__THIS");
+    assert!(index.find_type("fb_test.__THIS").is_some());
+}
+
+#[test]
+fn this_in_conditionals() {
+    let id_provider = IdProvider::default();
+    let (unit, mut index) = index_with_ids(
+        "
+        FUNCTION_BLOCK FB_Test
+            VAR
+                x : INT;
+                bo: BOOL;
+            END_VAR
+            IF this^.bo THEN
+                x := 1;
+            END_IF
+        END_FUNCTION_BLOCK
+        ",
+        id_provider.clone(),
+    );
+
+    let annotations = annotate_with_ids(&unit, &mut index, id_provider);
+    let statement = &unit.implementations[0].statements[0];
+    let AstStatement::ControlStatement(AstControlStatement::If(IfStatement { blocks, .. })) =
+        unit.implementations[0].statements[0].get_stmt()
+    else {
+        unreachable!();
+    };
+    let AstStatement::ReferenceExpr(ReferenceExpr { base: Some(deref), .. }) = blocks[0].condition.get_stmt()
+    else {
+        unreachable!();
+    };
+    let AstStatement::ReferenceExpr(ReferenceExpr { base: Some(base), .. }) = deref.get_stmt() else {
+        unreachable!()
+    };
+    let StatementAnnotation::Value { resulting_type, .. } = annotations.get(base).expect("damn") else {
+        unreachable!()
+    };
+    assert_eq!(resulting_type, "FB_Test.__THIS");
+    dbg!(resulting_type);
+    // dbg!(&statement);
+    assert!(index.find_type("fb_test.__THIS").is_some());
 }