From b10318f32cc8c8336bd5709210c37e74d76cf7bd Mon Sep 17 00:00:00 2001
From: Alexandre Hamez <alexandre.hamez@gmail.com>
Date: Wed, 2 Mar 2022 22:17:28 +0100
Subject: [PATCH] Add an ending newline to generated files

---
 lib/protox/generate.ex        | 12 ++----------
 test/code_generation_test.exs | 12 ++++++++++++
 2 files changed, 14 insertions(+), 10 deletions(-)

diff --git a/lib/protox/generate.ex b/lib/protox/generate.ex
index 040da713..6fabff39 100644
--- a/lib/protox/generate.ex
+++ b/lib/protox/generate.ex
@@ -61,18 +61,10 @@ defmodule Protox.Generate do
   end
 
   defp generate_file_content(code) do
-    code_str = Macro.to_string(code)
-
-    formatted_code =
-      try do
-        Code.format_string!(code_str)
-      rescue
-        _ -> code_str
-      end
-
     [
       "# credo:disable-for-this-file\n",
-      formatted_code
+      code |> Macro.to_string() |> Code.format_string!(),
+      "\n"
     ]
   end
 end
diff --git a/test/code_generation_test.exs b/test/code_generation_test.exs
index 18ce6000..b99b2fc2 100644
--- a/test/code_generation_test.exs
+++ b/test/code_generation_test.exs
@@ -145,6 +145,18 @@ defmodule Protox.CodeGenerationTest do
       )
 
     assert credo_exit_status == 0
+
+    {_, mix_format_exit_status} =
+      System.cmd(
+        "mix",
+        [
+          "format",
+          "--check-formatted"
+        ],
+        cd: code_generation_path
+      )
+
+    assert mix_format_exit_status == 0
   end
 
   defp mix_exs() do