aboutsummaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_rdap.py56
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()