From d90dfa8db777221503ecac101d1a167d84c4e723 Mon Sep 17 00:00:00 2001 From: Dashgin Date: Sun, 22 May 2022 20:05:48 +0400 Subject: [PATCH] added test for `remove_prefix` method --- defender/tests.py | 7 +++++++ 1 file changed, 7 insertions(+) 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")