Merge pull request #1413 from ChJR/hotfix/exit

youknowone · web-flow · commit f209929dbca7 · 2019-09-26T20:16:08.000+09:00
Fix exit()
diff --git a/tests/snippets/exit.py b/tests/snippets/exit.py
@@ -0,0 +1,4 @@
+from testutils import assert_raises
+
+with assert_raises(SystemExit):
+    exit()
diff --git a/vm/src/builtins.rs b/vm/src/builtins.rs
@@ -883,6 +883,7 @@ pub fn make_module(vm: &VirtualMachine, module: PyObjectRef) {
         "UserWarning" => ctx.exceptions.user_warning.clone(),
 
         "KeyboardInterrupt" => ctx.exceptions.keyboard_interrupt.clone(),
+        "SystemExit" => ctx.exceptions.system_exit.clone(),
     });
 }
 
diff --git a/vm/src/exceptions.rs b/vm/src/exceptions.rs
@@ -252,6 +252,7 @@ pub struct ExceptionZoo {
     pub user_warning: PyClassRef,
 
     pub keyboard_interrupt: PyClassRef,
+    pub system_exit: PyClassRef,
 }
 
 impl ExceptionZoo {
@@ -303,6 +304,7 @@ impl ExceptionZoo {
         let user_warning = create_type("UserWarning", &type_type, &warning);
 
         let keyboard_interrupt = create_type("KeyboardInterrupt", &type_type, &base_exception_type);
+        let system_exit = create_type("SystemExit", &type_type, &base_exception_type);
 
         ExceptionZoo {
             arithmetic_error,
@@ -347,6 +349,7 @@ impl ExceptionZoo {
             reference_error,
             user_warning,
             keyboard_interrupt,
+            system_exit,
         }
     }
 }

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +from testutils import assert_raises
++
 +with assert_raises(SystemExit):
 +    exit()
Original file line number	Diff line number	Diff line change
`@@ -883,6 +883,7 @@ pub fn make_module(vm: &VirtualMachine, module: PyObjectRef) {`
`883`	`883`	`"UserWarning" => ctx.exceptions.user_warning.clone(),`
`884`	`884`
`885`	`885`	`"KeyboardInterrupt" => ctx.exceptions.keyboard_interrupt.clone(),`
	`886`	`+ "SystemExit" => ctx.exceptions.system_exit.clone(),`
`886`	`887`	`});`
`887`	`888`	`}`
`888`	`889`
Original file line number	Diff line number	Diff line change
`@@ -252,6 +252,7 @@ pub struct ExceptionZoo {`
`252`	`252`	`pub user_warning: PyClassRef,`
`253`	`253`
`254`	`254`	`pub keyboard_interrupt: PyClassRef,`
	`255`	`+ pub system_exit: PyClassRef,`
`255`	`256`	`}`
`256`	`257`
`257`	`258`	`impl ExceptionZoo {`
`@@ -303,6 +304,7 @@ impl ExceptionZoo {`
`303`	`304`	`let user_warning = create_type("UserWarning", &type_type, &warning);`
`304`	`305`
`305`	`306`	`let keyboard_interrupt = create_type("KeyboardInterrupt", &type_type, &base_exception_type);`
	`307`	`+ let system_exit = create_type("SystemExit", &type_type, &base_exception_type);`
`306`	`308`
`307`	`309`	`ExceptionZoo {`
`308`	`310`	`arithmetic_error,`
`@@ -347,6 +349,7 @@ impl ExceptionZoo {`
`347`	`349`	`reference_error,`
`348`	`350`	`user_warning,`
`349`	`351`	`keyboard_interrupt,`
	`352`	`+ system_exit,`
`350`	`353`	`}`
`351`	`354`	`}`
`352`	`355`	`}`