Fix crash when format_kind_reason is executed with TestModule (#14901)

Closes #14900.

Author: lukaszsamson

lib/ex_unit/lib/ex_unit/formatter.ex

@@ -338,7 +338,14 @@ defmodule ExUnit.Formatter do
         {formatted_reason, wrapped_stack} =
           format_exception(test, struct, wrapped_stack, width, formatter, @counter_padding)
 
-        formatted_stack = format_stacktrace(wrapped_stack, test.module, test.name, formatter)
+        formatted_stack =
+          format_stacktrace(
+            wrapped_stack,
+            stacktrace_case(test),
+            stacktrace_test(test),
+            formatter
+          )
+
         {error_info(header, formatter) <> pad(formatted_reason <> formatted_stack), stack}
 
       :error ->
@@ -367,6 +374,14 @@ defmodule ExUnit.Formatter do
 
   defp linked_or_trapped_exit(_kind, _reason), do: :error
 
+  defp stacktrace_case(%ExUnit.Test{module: module}), do: module
+  defp stacktrace_case(%ExUnit.TestModule{name: name}), do: name
+  defp stacktrace_case(_), do: nil
+
+  defp stacktrace_test(%ExUnit.Test{name: name}), do: name
+  defp stacktrace_test(%ExUnit.TestModule{}), do: nil
+  defp stacktrace_test(_), do: nil
+
   defp format_exception(test, %ExUnit.AssertionError{} = struct, stack, width, formatter, pad) do
     label_padding_size = if has_value?(struct.right), do: 7, else: 6
     padding_size = label_padding_size + byte_size(@counter_padding)

lib/ex_unit/test/ex_unit/formatter_test.exs

@@ -465,6 +465,21 @@ defmodule ExUnit.FormatterTest do
            """
   end
 
+  test "formats setup_all exit stacktraces using module context" do
+    stacktrace = [
+      {Hello, :setup_all, [], [file: __ENV__.file, line: 123]},
+      {Hello, :some_fun, [], [file: __ENV__.file, line: 456]}
+    ]
+
+    failure = [{{:EXIT, self()}, {:function_clause, stacktrace}, []}]
+
+    formatted =
+      format_test_all_failure(test_module(), failure, 1, 80, &formatter/2)
+
+    assert formatted =~ "Hello.setup_all/0"
+    assert formatted =~ "Hello.some_fun()"
+  end
+
   test "formats assertions with operators with no limit" do
     failure = [{:error, catch_assertion(assert [1, 2, 3] == [4, 5, 6]), []}]