Skip to content
Snippets Groups Projects
Commit ce40618d authored by ale's avatar ale
Browse files

fix the source_ip blacklist

parent 134389bf
No related branches found
No related tags found
No related merge requests found
...@@ -10,7 +10,7 @@ from flask import Flask, request, abort, make_response ...@@ -10,7 +10,7 @@ from flask import Flask, request, abort, make_response
@blacklist_on_auth_failure(key_from_args(0), count=5, period=600, ttl=43200) @blacklist_on_auth_failure(key_from_args(0), count=5, period=600, ttl=43200)
@blacklist_on_auth_failure(key_from_args(4), count=5, period=600, ttl=43200, @blacklist_on_auth_failure(key_from_args(5), count=5, period=600, ttl=43200,
check_wl=True) check_wl=True)
def _auth(username, service, shard, password, otp_token, source_ip): def _auth(username, service, shard, password, otp_token, source_ip):
user = app.userdb.get_user(username, service, shard) user = app.userdb.get_user(username, service, shard)
......
...@@ -157,7 +157,7 @@ class ServerTest(unittest.TestCase): ...@@ -157,7 +157,7 @@ class ServerTest(unittest.TestCase):
'service': 'svc', 'source_ip': '1.2.3.4'}) 'service': 'svc', 'source_ip': '1.2.3.4'})
self.assertEquals(200, response.status_code) self.assertEquals(200, response.status_code)
self.assertEquals(protocol.ERR_AUTHENTICATION_FAILURE, response.data, self.assertEquals(protocol.ERR_AUTHENTICATION_FAILURE, response.data,
'failed at %d (t %d)' % (i, self.tick)) 'failed at %d (t %d): %s' % (i, self.tick, response.data))
def test_blacklist_by_source_ip_whitelisted(self): def test_blacklist_by_source_ip_whitelisted(self):
self._create_many_users(60) self._create_many_users(60)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment