Make tests run independently from cwdlarge-tests3

1 files changed, 24 insertions, 1 deletions

diff --git a/verifycert.erl b/verifycert.erl
index fa0e4e4..a606d1c 100755
--- a/verifycert.erl
+++ b/verifycert.erl
@@ -1,6 +1,5 @@
 #!/usr/bin/env escript
 %% -*- erlang -*-
-%%! -pa lib/catlfish-0.9.0-dev.ez/catlfish-0.9.0-dev/ebin -pa lib/lager-2.1.1.ez/lager-2.1.1/ebin -pa lib/plop-0.9.0-dev.ez/plop-0.9.0-dev/ebin
 
 write_reply(Bin) ->
     Length = size(Bin),
@@ -34,6 +33,30 @@ loop(RootCerts) ->
             loop(RootCerts)
     end.
 
+add_lib(Dirs, Name) ->
+    add_lib(Dirs, Dirs, Name).
+add_lib(AllDirs, [], Name) ->
+    {ok, Cwd} = file:get_cwd(),
+    io:format(standard_error,
+              "Could not add lib ~p: tried directories ~p cwd: ~p~n",
+              [Name, AllDirs, Cwd]),
+    halt(1);
+add_lib(AllDirs, [Dir | Rest], Name) ->
+    Path = Dir ++ "/" ++ Name ++ ".ez/" ++ Name ++ "/ebin",
+    case code:add_path(Path) of
+        true ->
+            ok;
+        {error, bad_directory} ->
+            add_lib(AllDirs, Rest, Name)
+    end.
+
 main([KnownRoots]) ->
+    [ScriptFile | _] = init:get_plain_arguments(),
+    BaseDir = filename:dirname(ScriptFile),
+    LibDirs = [BaseDir ++ "/../lib", "../lib"],
+    add_lib(LibDirs, "catlfish-0.9.0-dev"),
+    add_lib(LibDirs, "lager-2.1.1"),
+    add_lib(LibDirs, "plop-0.9.0-dev"),
+
     Certs = x509:read_pemfiles_from_dir(KnownRoots),
     loop(Certs).