From de00a4c0f1cc5d61d0a821a2d0a292f8bd95e3f1 Mon Sep 17 00:00:00 2001
From: lain <lain@soykaf.club>
Date: Fri, 7 Aug 2020 15:27:41 +0200
Subject: [PATCH] Mix Task Frontend Test: Capture IO.

---
 test/tasks/frontend_test.exs | 34 ++++++++++++++++++++++------------
 1 file changed, 22 insertions(+), 12 deletions(-)

diff --git a/test/tasks/frontend_test.exs b/test/tasks/frontend_test.exs
index 5cd4594e2..6a9a931eb 100644
--- a/test/tasks/frontend_test.exs
+++ b/test/tasks/frontend_test.exs
@@ -6,6 +6,8 @@ defmodule Pleroma.FrontendTest do
   use Pleroma.DataCase
   alias Mix.Tasks.Pleroma.Frontend
 
+  import ExUnit.CaptureIO, only: [capture_io: 1]
+
   @dir "test/frontend_static_test"
 
   setup do
@@ -31,7 +33,10 @@ test "it downloads and unzips a known frontend" do
       %Tesla.Env{status: 200, body: File.read!("test/fixtures/tesla_mock/frontend.zip")}
     end)
 
-    Frontend.run(["install", "pleroma"])
+    capture_io(fn ->
+      Frontend.run(["install", "pleroma"])
+    end)
+
     assert File.exists?(Path.join([@dir, "frontends", "pleroma", "fantasy", "test.txt"]))
   end
 
@@ -44,7 +49,10 @@ test "it also works given a file" do
       }
     })
 
-    Frontend.run(["install", "pleroma", "--file", "test/fixtures/tesla_mock/frontend.zip"])
+    capture_io(fn ->
+      Frontend.run(["install", "pleroma", "--file", "test/fixtures/tesla_mock/frontend.zip"])
+    end)
+
     assert File.exists?(Path.join([@dir, "frontends", "pleroma", "fantasy", "test.txt"]))
   end
 
@@ -53,16 +61,18 @@ test "it downloads and unzips unknown frontends" do
       %Tesla.Env{status: 200, body: File.read!("test/fixtures/tesla_mock/frontend.zip")}
     end)
 
-    Frontend.run([
-      "install",
-      "unknown",
-      "--ref",
-      "baka",
-      "--build-url",
-      "http://gensokyo.2hu/madeup.zip",
-      "--build-dir",
-      ""
-    ])
+    capture_io(fn ->
+      Frontend.run([
+        "install",
+        "unknown",
+        "--ref",
+        "baka",
+        "--build-url",
+        "http://gensokyo.2hu/madeup.zip",
+        "--build-dir",
+        ""
+      ])
+    end)
 
     assert File.exists?(Path.join([@dir, "frontends", "unknown", "baka", "test.txt"]))
   end