Procházet zdrojové kódy

feat(cth-suite): use cheaper heuristic for schema modules

Andrew Mayorov před 2 roky
rodič
revize
24dfa41722
1 změnil soubory, kde provedl 6 přidání a 3 odebrání
  1. 6 3
      apps/emqx/test/emqx_cth_suite.erl

+ 6 - 3
apps/emqx/test/emqx_cth_suite.erl

@@ -470,9 +470,12 @@ clean_suite_state() ->
 
 
 app_schema(App) ->
 app_schema(App) ->
     Mod = list_to_atom(atom_to_list(App) ++ "_schema"),
     Mod = list_to_atom(atom_to_list(App) ++ "_schema"),
-    try is_list(Mod:roots()) of
-        true -> {ok, Mod};
-        false -> {error, schema_no_roots}
+    try
+        Exports = Mod:module_info(exports),
+        case lists:member({roots, 0}, Exports) of
+            true -> {ok, Mod};
+            false -> {error, schema_no_roots}
+        end
     catch
     catch
         error:undef ->
         error:undef ->
             {error, schema_not_found}
             {error, schema_not_found}