X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=test%2Fdocs%2Fgenerator_test.exs;h=43877244d28fe801e9c1c1b47295df31f96a2800;hb=8bacdc36806efd01a7897359ff0fd2c8e24730d2;hp=9c9f4357b2836a116d84c0d05d528e4aee2484ab;hpb=7605fe5065b62dcb8aaf7db07e5d168138b80d3e;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,