43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
# pyruse is intended as a replacement to both fail2ban and epylog
|
||
# Copyright © 2017–2018 Y. Gablin
|
||
# Full licensing information in the LICENSE file, or gnu.org/licences/gpl-3.0.txt if the file is missing.
|
||
from pyruse.filters.filter_inNetworks import Filter
|
||
|
||
def whenIp4InNet4ThenTrue():
|
||
assert Filter({"field": "ip", "nets": ["34.56.78.90/12"]}).filter({"ip": "34.48.0.1"})
|
||
|
||
def whenIp4NotInNet4ThenFalse():
|
||
assert not Filter({"field": "ip", "nets": ["34.56.78.90/12"]}).filter({"ip": "34.47.255.254"})
|
||
|
||
def whenIp6InNet6ThenTrue():
|
||
assert Filter({"field": "ip", "nets": ["2001:db8:1:1a0::/59"]}).filter({"ip": "2001:db8:1:1a0::1"})
|
||
|
||
def whenIp6NotInNet6ThenFalse():
|
||
assert not Filter({"field": "ip", "nets": ["2001:db8:1:1a0::/59"]}).filter({"ip": "2001:db8:1:19f:ffff:ffff:ffff:fffe"})
|
||
|
||
def whenNumericIp6InNet4ThenFalse():
|
||
assert not Filter({"field": "ip", "nets": ["34.56.78.90/12"]}).filter({"ip": "::2230:1"})
|
||
|
||
def whenNumericIp4InNet6ThenFalse():
|
||
assert not Filter({"field": "ip", "nets": ["::2230:1/108"]}).filter({"ip": "34.48.0.1"})
|
||
|
||
def whenIpInOneNetworkThenTrue():
|
||
assert Filter({"field": "ip", "nets": ["::2230:1/108", "10.0.0.0/8", "34.56.78.90/12", "2001:db8:1:1a0::/59"]}).filter({"ip": "34.48.0.1"})
|
||
|
||
def whenNoIpThenFalse():
|
||
assert not Filter({"field": "ip", "nets": ["::2230:1/108", "10.0.0.0/8"]}).filter({"no_ip": "Hi!"})
|
||
|
||
def whenNoNetworkThenFalse():
|
||
assert not Filter({"field": "ip", "nets": []}).filter({"ip": "34.48.0.1"})
|
||
|
||
def unitTests():
|
||
whenIp4InNet4ThenTrue()
|
||
whenIp4NotInNet4ThenFalse()
|
||
whenIp6InNet6ThenTrue()
|
||
whenIp6NotInNet6ThenFalse()
|
||
whenNumericIp6InNet4ThenFalse()
|
||
whenNumericIp4InNet6ThenFalse()
|
||
whenIpInOneNetworkThenTrue()
|
||
whenNoIpThenFalse()
|
||
whenNoNetworkThenFalse()
|