tests: add a test to ensure py-spy does not fails on cyrillic (ucs-2) strings

Author: nariman

tests/integration_test.rs

@@ -229,6 +229,27 @@ fn test_unicode() {
     assert!(!traces[0].owns_gil);
 }
 
+#[test]
+fn test_cyrillic() {
+    #[cfg(target_os = "macos")]
+    {
+        if unsafe { libc::geteuid() } != 0 {
+            return;
+        }
+    }
+    let mut runner = TestRunner::new(Config::default(), "./tests/scripts/cyrillic.py");
+
+    let traces = runner.spy.get_stack_traces().unwrap();
+    assert_eq!(traces.len(), 1);
+    let trace = &traces[0];
+
+    assert_eq!(trace.frames[0].name, "кириллица");
+    assert_eq!(trace.frames[0].line, 4);
+
+    assert_eq!(trace.frames[1].name, "<module>");
+    assert_eq!(trace.frames[1].line, 7);
+}
+
 #[test]
 fn test_local_vars() {
     #[cfg(target_os = "macos")]

tests/scripts/cyrillic.py

@@ -1,15 +1,7 @@
 import time
 
-
-def f(seconds):
-    time.sleep(seconds)
-
-
 def кириллица(seconds):
-    f(seconds)
-
+    time.sleep(seconds)
 
 if __name__ == "__main__":
-    f(3)
-    кириллица(3)
-    f(3)
+    кириллица(10)