impl faulthandler

Author: youknowone

Lib/test/test_faulthandler.py

@@ -411,8 +411,6 @@ def test_dump_ext_modules(self):
         for name in ('sys', 'faulthandler'):
             self.assertIn(name, modules)
 
-    # TODO: RUSTPYTHON, AttributeError: module 'faulthandler' has no attribute 'is_enabled'
-    @unittest.expectedFailure
     def test_is_enabled(self):
         orig_stderr = sys.stderr
         try:
@@ -435,8 +433,6 @@ def test_is_enabled(self):
         finally:
             sys.stderr = orig_stderr
 
-    # TODO: RUSTPYTHON, subprocess.CalledProcessError: Command ... returned non-zero exit status 1.
-    @unittest.expectedFailure
     @support.requires_subprocess()
     def test_disabled_by_default(self):
         # By default, the module should be disabled

crates/stdlib/src/faulthandler.rs

@@ -3,6 +3,9 @@ pub(crate) use decl::make_module;
 #[pymodule(name = "faulthandler")]
 mod decl {
     use crate::vm::{VirtualMachine, frame::Frame, function::OptionalArg, stdlib::sys::PyStderr};
+    use std::sync::atomic::{AtomicBool, Ordering};
+
+    static ENABLED: AtomicBool = AtomicBool::new(false);
 
     fn dump_frame(frame: &Frame, vm: &VirtualMachine) {
         let stderr = PyStderr(vm);
@@ -39,8 +42,18 @@ mod decl {
     }
 
     #[pyfunction]
-    const fn enable(_args: EnableArgs) {
-        // TODO
+    fn enable(_args: EnableArgs) {
+        ENABLED.store(true, Ordering::Relaxed);
+    }
+
+    #[pyfunction]
+    fn disable() {
+        ENABLED.store(false, Ordering::Relaxed);
+    }
+
+    #[pyfunction]
+    fn is_enabled() -> bool {
+        ENABLED.load(Ordering::Relaxed)
     }
 
     #[derive(FromArgs)]