Merge pull request #249 from Glutexo/with-module

Help message on invalid component
Glutexo · Nov 9, 2024 · ded9aa9 · ded9aa9
2 parents fcfea46 + 6811b1a
commit ded9aa9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/lib/cli.ex b/lib/cli.ex
@@ -10,9 +10,12 @@ defmodule Onigumo.CLI do
            strict: [working_dir: :string]
          ) do
       {switches, [component], []} ->
-        {:ok, module} = Map.fetch(@components, String.to_atom(component))
-        working_dir = Keyword.get(switches, :working_dir, File.cwd!())
-        module.main(working_dir)
+        with {:ok, module} <- Map.fetch(@components, String.to_atom(component)) do
+          working_dir = Keyword.get(switches, :working_dir, File.cwd!())
+          module.main(working_dir)
+        else
+          :error -> usage_message()
+        end
 
       _ ->
         usage_message()

diff --git a/test/onigumo_cli_test.exs b/test/onigumo_cli_test.exs
@@ -21,7 +21,7 @@ defmodule OnigumoCLITest do
   describe("Onigumo.CLI.main/1") do
     for argument <- @invalid_arguments do
       test("run CLI with invalid argument #{inspect(argument)}") do
-        assert_raise(MatchError, fn -> Onigumo.CLI.main([unquote(argument)]) end)
+        assert usage_message_printed?(fn -> Onigumo.CLI.main([unquote(argument)]) end)
       end
     end