X-Git-Url: https://git.squeep.com/?a=blobdiff_plain;f=test%2Fdocs%2Fgenerator_test.exs;h=43877244d28fe801e9c1c1b47295df31f96a2800;hb=8539e386c3f00537f120487e717ec7b25fe6c572;hp=9c9f4357b2836a116d84c0d05d528e4aee2484ab;hpb=503d966e9f7d4c41a1bfbd215b375443a858564a;p=akkoma

diff --git a/test/docs/generator_test.exs b/test/docs/generator_test.exs
index 9c9f4357b..43877244d 100644
--- a/test/docs/generator_test.exs
+++ b/test/docs/generator_test.exs
@@ -1,3 +1,7 @@
+# Pleroma: A lightweight social networking server
+# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
+# SPDX-License-Identifier: AGPL-3.0-only
+
 defmodule Pleroma.Docs.GeneratorTest do
   use ExUnit.Case, async: true
   alias Pleroma.Docs.Generator
@@ -13,21 +17,13 @@ defmodule Pleroma.Docs.GeneratorTest do
           key: :uploader,
           type: :module,
           description: "",
-          suggestions:
-            Generator.list_modules_in_dir(
-              "lib/pleroma/upload/filter",
-              "Elixir.Pleroma.Upload.Filter."
-            )
+          suggestions: {:list_behaviour_implementations, Pleroma.Upload.Filter}
         },
         %{
           key: :filters,
           type: {:list, :module},
           description: "",
-          suggestions:
-            Generator.list_modules_in_dir(
-              "lib/pleroma/web/activity_pub/mrf",
-              "Elixir.Pleroma.Web.ActivityPub.MRF."
-            )
+          suggestions: {:list_behaviour_implementations, Pleroma.Web.ActivityPub.MRF}
         },
         %{
           key: Pleroma.Upload,