From b528da0558d7ab4217e18255705182dae0497afe Mon Sep 17 00:00:00 2001
From: devdanzin <74280297+devdanzin@users.noreply.github.com>
Date: Tue, 15 Aug 2023 23:56:41 -0300
Subject: [PATCH] Fix tests, add new test for graph with multiple paths.

---
 test/unit/test_cli.py | 38 +++++++++++++++++++++++++++++---------
 1 file changed, 29 insertions(+), 9 deletions(-)

diff --git a/test/unit/test_cli.py b/test/unit/test_cli.py
index 0300c671..40c1f8a0 100644
--- a/test/unit/test_cli.py
+++ b/test/unit/test_cli.py
@@ -287,12 +287,32 @@ def test_graph():
     with patch("wily.__main__.exists", return_value=True) as check_cache:
         with patch("wily.commands.graph.graph") as graph:
             runner = CliRunner()
-            result = runner.invoke(main.cli, ["graph", "foo.py", "example_metric"])
+            result = runner.invoke(
+                main.cli, ["graph", "foo.py", "-m", "example_metric"]
+            )
+            assert result.exit_code == 0
+            assert graph.called_once
+            assert check_cache.called_once
+            assert graph.call_args[1]["path"] == ("foo.py",)
+            assert graph.call_args[1]["metrics"] == "example_metric"
+
+
+def test_graph_multiple_paths():
+    """
+    Test that graph calls the graph command with multiple paths
+    """
+    with patch("wily.__main__.exists", return_value=True) as check_cache:
+        with patch("wily.commands.graph.graph") as graph:
+            runner = CliRunner()
+            result = runner.invoke(
+                main.cli,
+                ["graph", "foo.py", "bar.py", "baz.py", "-m", "example_metric"],
+            )
             assert result.exit_code == 0
             assert graph.called_once
             assert check_cache.called_once
-            assert graph.call_args[1]["path"] == "foo.py"
-            assert graph.call_args[1]["metrics"] == ("example_metric",)
+            assert graph.call_args[1]["path"] == ("foo.py", "bar.py", "baz.py")
+            assert graph.call_args[1]["metrics"] == "example_metric"
 
 
 def test_graph_multiple_metrics():
@@ -303,13 +323,13 @@ def test_graph_multiple_metrics():
         with patch("wily.commands.graph.graph") as graph:
             runner = CliRunner()
             result = runner.invoke(
-                main.cli, ["graph", "foo.py", "example_metric", "another_metric"]
+                main.cli, ["graph", "foo.py", "-m", "example_metric,another_metric"]
             )
             assert result.exit_code == 0
             assert graph.called_once
             assert check_cache.called_once
-            assert graph.call_args[1]["path"] == "foo.py"
-            assert graph.call_args[1]["metrics"] == ("example_metric", "another_metric")
+            assert graph.call_args[1]["path"] == ("foo.py",)
+            assert graph.call_args[1]["metrics"] == "example_metric,another_metric"
 
 
 def test_graph_with_output():
@@ -320,13 +340,13 @@ def test_graph_with_output():
         with patch("wily.commands.graph.graph") as graph:
             runner = CliRunner()
             result = runner.invoke(
-                main.cli, ["graph", "foo.py", "example_metric", "-o", "foo.html"]
+                main.cli, ["graph", "foo.py", "-m", "example_metric", "-o", "foo.html"]
             )
             assert result.exit_code == 0
             assert graph.called_once
             assert check_cache.called_once
-            assert graph.call_args[1]["path"] == "foo.py"
-            assert graph.call_args[1]["metrics"] == ("example_metric",)
+            assert graph.call_args[1]["path"] == ("foo.py",)
+            assert graph.call_args[1]["metrics"] == "example_metric"
             assert graph.call_args[1]["output"] == "foo.html"