From 329f5c918abe63f70ede2edc8fa52946e6d89e84 Mon Sep 17 00:00:00 2001
From: Guilherme Andrade <g@gandrade.net>
Date: Sat, 20 Jul 2024 17:22:24 +0100
Subject: [PATCH] Commit unfinished work needed to fix test downloads from
 MaxMind

---
 test/locus_remote_sources_SUITE.erl | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/test/locus_remote_sources_SUITE.erl b/test/locus_remote_sources_SUITE.erl
index b0471da..21c2dd2 100644
--- a/test/locus_remote_sources_SUITE.erl
+++ b/test/locus_remote_sources_SUITE.erl
@@ -71,7 +71,9 @@ groups() ->
 
 should_run_remote_http_tests() ->
     currently_checkedout_commit_is_likely_tagged()
-    andalso license_key_from_environment_is_defined().
+    andalso license_key_from_environment_is_defined()
+    % FIXME 2024/07/20 got rate limited by MaxMind when fixing tests
+    andalso list_to_atom("false").
 
 currently_checkedout_commit_is_likely_tagged() ->
     {ok, _} = application:ensure_all_started(locus),
@@ -205,12 +207,16 @@ cacheless_loading_test(Config) ->
 
         {maxmind, _} ->
             ?assertRecv({locus, Loader, {request_sent, _URL, _Headers}}),
+            ?assertRecv({locus, Loader, {download_redirected, _Info}}),
+            ?assertRecv({locus, Loader, {request_sent, _, _}}),
             ?assertRecv({locus, Loader, {download_started, _Headers}}),
             ?assertRecv({locus, Loader, {download_finished, _BytesReceived,
                                          {ok, _TrailingHeaders}}}),
 
             ?assertRecv({locus, Loader, {checksum, {request_sent, _ChecksumURL,
                                                     _ChecksumReqHeaders}}}),
+            ?assertRecv({locus, Loader, {checksum, {download_redirected, _Info}}}),
+            ?assertRecv({locus, Loader, {checksum, {request_sent, _, _}}}),
             ?assertRecv({locus, Loader, {checksum, {download_started,
                                                     _ChecksumRespHeaders}}}),
             ?assertRecv({locus, Loader, {checksum, {download_finished,
@@ -247,6 +253,8 @@ cold_remote_loading_test(Config) ->
 
         {maxmind, _} ->
             ?assertRecv({locus, Loader, {request_sent, _URL, _Headers}}),
+            ?assertRecv({locus, Loader, {download_redirected, _Info}}),
+            ?assertRecv({locus, Loader, {request_sent, _, _}}),
             ?assertRecv({locus, Loader, {download_started, _Headers}}),
             ?assertRecv({locus, Loader, {download_finished, _BytesReceived,
                                          {ok, _TrailingHeaders}}}),
@@ -461,6 +469,14 @@ idle_download_timeout_httptest(Config) ->
     (fun F(AttemptsLeft) ->
              ok = locus:start_loader(Loader, LoadFrom, LoaderOpts),
              ?assertRecv({locus, Loader, {request_sent, _URL, _Headers}}),
+
+             case LoadFrom of
+                 {maxmind, _} ->
+                     ?assertRecv({locus, Loader, {download_redirected, _Info}}),
+                     ?assertRecv({locus, Loader, {request_sent, _, _}});
+                _ -> ok
+             end,
+
              ?assertRecv({locus, Loader, {download_started, _Headers}}),
              try ?assertRecv({locus, Loader, {download_finished, _BytesReceived,
                                               {error, timeout}}})