diff options
Diffstat (limited to 'tests/test_rdap.py')
| -rw-r--r-- | tests/test_rdap.py | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/tests/test_rdap.py b/tests/test_rdap.py new file mode 100644 index 0000000..681e2bb --- /dev/null +++ b/tests/test_rdap.py @@ -0,0 +1,56 @@ +# Copyright (C) 2026 Danilo M. <danix@danix.xyz> +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License version 2 as +# published by the Free Software Foundation. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. +"""Tests for the RDAP protocol module.""" + +import email +import unittest +import urllib.error +import urllib.request + +from abusectl import rdap + + +class RedirectPolicy(unittest.TestCase): + """A redirect is remote data directing our next request. + + urllib.request.Request is used rather than a hand-rolled fake, because + HTTPRedirectHandler reads attributes (origin_req_host, unverifiable, + timeout) that a fake would have to reproduce exactly to prove anything. + """ + + def _request(self): + return urllib.request.Request("https://rdap.example.invalid/ip/192.0.2.1") + + def test_an_https_to_http_downgrade_is_refused(self): + handler = rdap._NoDowngradeRedirectHandler() + with self.assertRaises(urllib.error.HTTPError): + handler.redirect_request( + self._request(), None, 302, "Found", + email.message_from_string(""), + "http://rdap.example.invalid/ip/192.0.2.1", + ) + + def test_an_https_to_https_redirect_is_allowed(self): + handler = rdap._NoDowngradeRedirectHandler() + result = handler.redirect_request( + self._request(), None, 302, "Found", + email.message_from_string(""), + "https://other.example.invalid/ip/192.0.2.1", + ) + self.assertIsNotNone(result) + + +if __name__ == "__main__": + unittest.main() |
