diff --git a/defender/tests.py b/defender/tests.py index 69d87fe..1ca1b72 100644 --- a/defender/tests.py +++ b/defender/tests.py @@ -1169,3 +1169,10 @@ class TestUtils(DefenderTestCase): req = HttpRequest() req.META["HTTP_X_FORWARDED_FOR"] = "[2001:db8::1]:123456" self.assertEqual(utils.get_ip(req), "2001:db8::1") + + def test_remove_prefix(self): + """ test the remove_prefix() method """ + self.assertEqual(utils.remove_prefix( + "defender:blocked:ip:192.168.24.24", "defender:blocked:"), "ip:192.168.24.24") + self.assertEqual(utils.remove_prefix( + "defender:blocked:username:johndoe", "defender:blocked:"), "username:johndoe")