relink.py: New fn process_args.

Author: samsrabin

relink.py

@@ -192,21 +192,36 @@ def parse_arguments():
         help="Measure and display the execution time",
     )
 
-    return parser.parse_args()
+    args = parser.parse_args()
 
-def main():
+    process_args(args)
 
-    args = parse_arguments()
+    return args
 
+
+def process_args(args):
+    """
+    Process parsed arguments and set derived attributes.
+
+    Sets the log_level attribute on args based on verbosity flags.
+
+    Args:
+        args (argparse.Namespace): Parsed command-line arguments.
+    """
     # Configure logging based on verbosity flags
     if args.quiet:
-        log_level = logging.WARNING
+        args.log_level = logging.WARNING
     elif args.verbose:
-        log_level = logging.DEBUG
+        args.log_level = logging.DEBUG
     else:
-        log_level = logging.INFO
+        args.log_level = logging.INFO
+
 
-    logging.basicConfig(level=log_level, format="%(message)s", stream=sys.stdout)
+def main():
+
+    args = parse_arguments()
+
+    logging.basicConfig(level=args.log_level, format="%(message)s", stream=sys.stdout)
 
     my_username = os.environ["USER"]
 
@@ -221,5 +236,6 @@ def main():
         elapsed_time = time.time() - start_time
         logger.info("Execution time: %.2f seconds", elapsed_time)
 
+
 if __name__ == "__main__":
     main()

tests/test_relink.py

@@ -615,6 +615,7 @@ def mock_symlink(src, dst):
                 relink.find_and_replace_owned_files(source_dir, target_dir, username)
             assert "Error creating symlink" in caplog.text
 
+
 class TestEdgeCases:
     """Test edge cases and error handling."""
 
@@ -726,6 +727,7 @@ def mock_symlink(src, dst):
             assert "Error creating symlink" in caplog.text
             assert source_file in caplog.text
 
+
 class TestTiming:
     """Test suite for timing functionality."""
 
@@ -989,3 +991,36 @@ def test_command_line_execution_actual_run(self, mock_dirs):
 
         # Verify success messages in output
         assert "Created symbolic link:" in result.stdout
+
+
+class TestProcessArgs:
+    """Test suite for process_args function."""
+
+    # pylint: disable=no-member
+
+    def test_process_args_quiet_sets_warning_level(self):
+        """Test that quiet flag sets log level to WARNING."""
+        args = argparse.Namespace(quiet=True, verbose=False)
+        relink.process_args(args)
+        assert args.log_level == logging.WARNING
+
+    def test_process_args_verbose_sets_debug_level(self):
+        """Test that verbose flag sets log level to DEBUG."""
+        args = argparse.Namespace(quiet=False, verbose=True)
+        relink.process_args(args)
+        assert args.log_level == logging.DEBUG
+
+    def test_process_args_default_sets_info_level(self):
+        """Test that default (no flags) sets log level to INFO."""
+        args = argparse.Namespace(quiet=False, verbose=False)
+        relink.process_args(args)
+        assert args.log_level == logging.INFO
+
+    def test_process_args_modifies_args_in_place(self):
+        """Test that process_args modifies the args object in place."""
+        args = argparse.Namespace(quiet=False, verbose=False)
+        original_args = args
+        relink.process_args(args)
+        # Should be the same object, modified in place
+        assert args is original_args
+        assert hasattr(args, "log_level")