style: make main more readable

MilesCranmer · Apr 14, 2024 · c0e3092 · c0e3092
1 parent 9b0e61d
commit c0e3092
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -21,7 +21,12 @@ fn main() -> ExitCode {
         None => {}
     }
 
-    if let Err(ref e) = rip2::run(cli, mode, &mut stream) {
+    ////////////////////////////////////////////////////////////
+    // Main code ///////////////////////////////////////////////
+    let result = rip2::run(cli, mode, &mut stream);
+    ////////////////////////////////////////////////////////////
+
+    if let Err(ref e) = result {
         println!("Exception: {}", e);
         return ExitCode::FAILURE;
     }

diff --git a/tests/unit_tests.rs b/tests/unit_tests.rs
@@ -175,7 +175,9 @@ fn test_completions(
             assert!(result.is_err());
             let err_msg = result.unwrap_err().to_string();
             assert!(err_msg.contains("Invalid shell specification: fake"));
-            assert!(err_msg.contains("Available shells: bash, elvish, fish, powershell, zsh, nushell"));
+            assert!(
+                err_msg.contains("Available shells: bash, elvish, fish, powershell, zsh, nushell")
+            );
         }
         _ => {}
     }