thread-language 0.1.1

// SPDX-FileCopyrightText: 2022 Herrington Darkholme <2883231+HerringtonDarkholme@users.noreply.github.com>
// SPDX-FileCopyrightText: 2025 Knitli Inc. <knitli@knit.li>
// SPDX-FileContributor: Adam Poulemanos <adam@knit.li>
//
// SPDX-License-Identifier: AGPL-3.0-or-later AND MIT
#![cfg(test)]
use super::*;

fn test_match(query: &str, source: &str) {
    use crate::test::test_match_lang;
    test_match_lang(query, source, Python);
}

fn test_non_match(query: &str, source: &str) {
    use crate::test::test_non_match_lang;
    test_non_match_lang(query, source, Python);
}

#[test]
fn test_python_str() {
    test_match("print($A)", "print(123)");
    test_match("print('123')", "print('123')");
    test_non_match("print('123')", "print('456')");
    test_non_match("'123'", "'456'");
    // https://github.com/ast-grep/ast-grep/issues/276
    // python has fixed the wrong parsing issue
    test_non_match(
        "getattr($O, \"__spec__\", None)",
        "getattr(response, \"render\", None)",
    );
    test_match(
        "getattr($O, \"render\", None)",
        "getattr(response, \"render\", None)",
    );
}

// https://github.com/ast-grep/ast-grep/issues/883
#[test]
fn test_issue_883() {
    test_match("r'^[A-Za-z0-9_-]+\\$'", "r'^[A-Za-z0-9_-]+\\$'");
}

#[test]
fn test_python_pattern() {
    test_match("$A = 0", "a = 0");
    // A test case from https://peps.python.org/pep-0636/#appendix-a-quick-intro
    test_match(
        r#"
match $A:
  case $B:
      $C
  case [$D(0, 0)]:
      $E
  case [$D($F, $G)]:
      $H
  case [$D(0, $I), $D(0, $J)]:
      $K
  case _:
      $L
"#,
        r#"
match points:
  case []:
      print("No points")
  case [Point(0, 0)]:
      print("The origin")
  case [Point(x, y)]:
      print(f"Single point {x}, {y}")
  case [Point(0, y1), Point(0, y2)]:
      print(f"Two on the Y axis at {y1}, {y2}")
  case _:
      print("Something else")
"#,
    );
}

fn test_replace(src: &str, pattern: &str, replacer: &str) -> String {
    use crate::test::test_replace_lang;
    test_replace_lang(src, pattern, replacer, Python)
}

#[test]
fn test_python_replace() {
    let ret = test_replace(
        r#"
if flag:
  a = value_pos
else:
  a = value_neg"#,
        r#"
if $FLAG:
  $VAR = $POS
else:
  $VAR = $NEG
"#,
        "$VAR = $POS if $FLAG else $NEG",
    );
    assert_eq!(ret, "\na = value_pos if flag else value_neg");

    let ret = test_replace(
        r#"
try:
  f = open(file_path, "r")
  file_content = f.read()
except:
  pass
finally:
  f.close()"#,
        r#"
try:
  $A = open($B, $C)
  $D = $A.read()
except:
  pass
finally:
  $A.close()"#,
        r#"
with open($B, $C) as $A:
  $D = $A.open()"#,
    );
    assert_eq!(
        ret,
        r#"

with open(file_path, "r") as f:
  file_content = f.open()"#
    );
}