darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-04 02:11:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use ipaddress library for the masked bits

Browse Source
This commit is contained in:
venusbb
2017-09-13 14:08:23 +01:00
parent c285ab0b45
commit 9efc17a941
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/app/authentication/test_authentication.py
View File

@@ -313,7 +313,7 @@ def __create_token(service_id):
def restrict_ip_sms_app():
app = flask.Flask(__name__)
app.config['TESTING'] = True
app.config['SMS_INBOUND_WHITELIST'] = ['111.111.111.111', '100.100.100.100']
app.config['SMS_INBOUND_WHITELIST'] = ['111.111.111.111/32', '200.200.200.200/24']
blueprint = flask.Blueprint('restrict_ip_sms_app', __name__)
@blueprint.route('/')
@@ -369,7 +369,7 @@ def test_allow_valid_ips_24bits(restrict_ip_sms_app):
response = restrict_ip_sms_app.get(
path='/',
headers=[
('X-Forwarded-For', '111.111.111.119, 222.222.222.222, 127.0.0.1'),
('X-Forwarded-For', '200.200.200.222, 222.222.222.222, 127.0.0.1'),
]
)