281 lines
12 KiB
Python
281 lines
12 KiB
Python
from .autopath import do_autopath
|
|
do_autopath()
|
|
|
|
import warnings
|
|
from py.test import raises
|
|
from tiramisu import Config, IPOption, NetworkOption, NetmaskOption, \
|
|
PortOption, BroadcastOption, OptionDescription
|
|
from tiramisu.error import ValueWarning
|
|
from tiramisu.storage import list_sessions
|
|
|
|
|
|
def teardown_function(function):
|
|
assert list_sessions() == [], 'session list is not empty when leaving "{}"'.format(function.__name__)
|
|
|
|
|
|
def test_ip():
|
|
a = IPOption('a', '')
|
|
b = IPOption('b', '', private_only=True)
|
|
d = IPOption('d', '', warnings_only=True, private_only=True)
|
|
warnings.simplefilter("always", ValueWarning)
|
|
od = OptionDescription('od', '', [a, b, d])
|
|
config = Config(od)
|
|
config.option('a').value.set('192.168.1.1')
|
|
config.option('a').value.set('192.168.1.0')
|
|
config.option('a').value.set('88.88.88.88')
|
|
config.option('a').value.set('0.0.0.0')
|
|
raises(ValueError, "config.option('a').value.set('255.255.255.0')")
|
|
config.option('b').value.set('192.168.1.1')
|
|
config.option('b').value.set('192.168.1.0')
|
|
raises(ValueError, "config.option('b').value.set('88.88.88.88')")
|
|
config.option('b').value.set('0.0.0.0')
|
|
raises(ValueError, "config.option('b').value.set('255.255.255.0')")
|
|
raises(ValueError, "config.option('a').value.set('333.0.1.20')")
|
|
|
|
raises(ValueError, "IPOption('a', 'ip', default='192.000.023.01')")
|
|
with warnings.catch_warnings(record=True) as w:
|
|
config.option('d').value.set('88.88.88.88')
|
|
assert len(w) == 1
|
|
|
|
|
|
def test_ip_cidr():
|
|
b = IPOption('b', '', private_only=True, cidr=True)
|
|
c = IPOption('c', '', private_only=True)
|
|
warnings.simplefilter("always", ValueWarning)
|
|
od = OptionDescription('od', '', [b, c])
|
|
config = Config(od)
|
|
raises(ValueError, "config.option('b').value.set('192.168.1.1')")
|
|
config.option('b').value.set('192.168.1.1/24')
|
|
raises(ValueError, "config.option('b').value.set('192.168.1.1/32')")
|
|
#
|
|
config.option('c').value.set('192.168.1.1')
|
|
raises(ValueError, "config.option('c').value.set('192.168.1.1/24')")
|
|
raises(ValueError, "config.option('c').value.set('192.168.1.1/32')")
|
|
|
|
|
|
def test_ip_default():
|
|
a = IPOption('a', '', '88.88.88.88')
|
|
od = OptionDescription('od', '', [a])
|
|
c = Config(od)
|
|
c.option('a').value.get() == '88.88.88.88'
|
|
|
|
|
|
def test_ip_reserved():
|
|
a = IPOption('a', '')
|
|
b = IPOption('b', '', allow_reserved=True)
|
|
c = IPOption('c', '', warnings_only=True)
|
|
od = OptionDescription('od', '', [a, b, c])
|
|
warnings.simplefilter("always", ValueWarning)
|
|
cfg = Config(od)
|
|
raises(ValueError, "cfg.option('a').value.set('240.94.1.1')")
|
|
cfg.option('b').value.set('240.94.1.1')
|
|
with warnings.catch_warnings(record=True) as w:
|
|
cfg.option('c').value.set('240.94.1.1')
|
|
assert len(w) == 1
|
|
|
|
|
|
def test_network():
|
|
a = NetworkOption('a', '')
|
|
b = NetworkOption('b', '', warnings_only=True)
|
|
od = OptionDescription('od', '', [a, b])
|
|
warnings.simplefilter("always", ValueWarning)
|
|
cfg = Config(od)
|
|
cfg.option('a').value.set('192.168.1.1')
|
|
cfg.option('a').value.set('192.168.1.0')
|
|
cfg.option('a').value.set('88.88.88.88')
|
|
cfg.option('a').value.set('0.0.0.0')
|
|
raises(ValueError, "cfg.option('a').value.set(1)")
|
|
raises(ValueError, "cfg.option('a').value.set('1.1.1.1.1')")
|
|
raises(ValueError, "cfg.option('a').value.set('255.255.255.0')")
|
|
raises(ValueError, "cfg.option('a').value.set('192.168.001.0')")
|
|
raises(ValueError, "cfg.option('a').value.set('333.168.1.1')")
|
|
with warnings.catch_warnings(record=True) as w:
|
|
cfg.option('b').value.set('255.255.255.0')
|
|
assert len(w) == 1
|
|
|
|
|
|
def test_network_cidr():
|
|
a = NetworkOption('a', '', cidr=True)
|
|
od = OptionDescription('od', '', [a])
|
|
cfg = Config(od)
|
|
cfg.option('a').value.set('192.168.1.1/32')
|
|
cfg.option('a').value.set('192.168.1.0/24')
|
|
cfg.option('a').value.set('88.88.88.88/32')
|
|
cfg.option('a').value.set('0.0.0.0/0')
|
|
raises(ValueError, "cfg.option('a').value.set('192.168.1.1')")
|
|
raises(ValueError, "cfg.option('a').value.set('192.168.1.1/24')")
|
|
raises(ValueError, "cfg.option('a').value.set('2001:db00::0/24')")
|
|
|
|
|
|
def test_network_invalid():
|
|
raises(ValueError, "NetworkOption('a', '', default='toto')")
|
|
|
|
|
|
def test_netmask():
|
|
a = NetmaskOption('a', '')
|
|
od = OptionDescription('od', '', [a])
|
|
cfg = Config(od)
|
|
raises(ValueError, "cfg.option('a').value.set('192.168.1.1.1')")
|
|
raises(ValueError, "cfg.option('a').value.set('192.168.1.1')")
|
|
raises(ValueError, "cfg.option('a').value.set('192.168.1.0')")
|
|
raises(ValueError, "cfg.option('a').value.set('88.88.88.88')")
|
|
raises(ValueError, "cfg.option('a').value.set('255.255.255.000')")
|
|
raises(ValueError, "cfg.option('a').value.set(2)")
|
|
cfg.option('a').value.set('0.0.0.0')
|
|
cfg.option('a').value.set('255.255.255.0')
|
|
|
|
|
|
def test_broadcast():
|
|
a = BroadcastOption('a', '')
|
|
od = OptionDescription('od', '', [a])
|
|
cfg = Config(od)
|
|
raises(ValueError, "cfg.option('a').value.set('192.168.1.255.1')")
|
|
raises(ValueError, "cfg.option('a').value.set('192.168.001.255')")
|
|
raises(ValueError, "cfg.option('a').value.set('192.168.0.300')")
|
|
raises(ValueError, "cfg.option('a').value.set(1)")
|
|
raises(ValueError, "cfg.option('a').value.set(2)")
|
|
raises(ValueError, "cfg.option('a').value.set('2001:db8::1')")
|
|
cfg.option('a').value.set('0.0.0.0')
|
|
cfg.option('a').value.set('255.255.255.0')
|
|
|
|
|
|
def test_port():
|
|
a = PortOption('a', '')
|
|
b = PortOption('b', '', allow_zero=True)
|
|
c = PortOption('c', '', allow_zero=True, allow_registred=False)
|
|
d = PortOption('d', '', allow_zero=True, allow_wellknown=False, allow_registred=False)
|
|
e = PortOption('e', '', allow_zero=True, allow_private=True)
|
|
f = PortOption('f', '', allow_private=True)
|
|
od = OptionDescription('od', '', [a, b, c, d, e, f])
|
|
cfg = Config(od)
|
|
raises(ValueError, "cfg.option('a').value.set('0')")
|
|
cfg.option('a').value.set('1')
|
|
cfg.option('a').value.set('1023')
|
|
cfg.option('a').value.set('1024')
|
|
cfg.option('a').value.set('49151')
|
|
raises(ValueError, "cfg.option('a').value.set('49152')")
|
|
raises(ValueError, "cfg.option('a').value.set('65535')")
|
|
raises(ValueError, "cfg.option('a').value.set('65536')")
|
|
|
|
cfg.option('b').value.set('0')
|
|
cfg.option('b').value.set('1')
|
|
cfg.option('b').value.set('1023')
|
|
cfg.option('b').value.set('1024')
|
|
cfg.option('b').value.set('49151')
|
|
raises(ValueError, "cfg.option('b').value.set('49152')")
|
|
raises(ValueError, "cfg.option('b').value.set('65535')")
|
|
raises(ValueError, "cfg.option('b').value.set('65536')")
|
|
|
|
cfg.option('c').value.set('0')
|
|
cfg.option('c').value.set('1')
|
|
cfg.option('c').value.set('1023')
|
|
raises(ValueError, "cfg.option('c').value.set('1024')")
|
|
raises(ValueError, "cfg.option('c').value.set('49151')")
|
|
raises(ValueError, "cfg.option('c').value.set('49152')")
|
|
raises(ValueError, "cfg.option('c').value.set('65535')")
|
|
raises(ValueError, "cfg.option('c').value.set('65536')")
|
|
|
|
cfg.option('d').value.set('0')
|
|
raises(ValueError, "cfg.option('d').value.set('1')")
|
|
raises(ValueError, "cfg.option('d').value.set('1023')")
|
|
raises(ValueError, "cfg.option('d').value.set('1024')")
|
|
raises(ValueError, "cfg.option('d').value.set('49151')")
|
|
raises(ValueError, "cfg.option('d').value.set('49152')")
|
|
raises(ValueError, "cfg.option('d').value.set('65535')")
|
|
raises(ValueError, "cfg.option('d').value.set('65536')")
|
|
|
|
cfg.option('e').value.set('0')
|
|
cfg.option('e').value.set('1')
|
|
cfg.option('e').value.set('1023')
|
|
cfg.option('e').value.set('1024')
|
|
cfg.option('e').value.set('49151')
|
|
cfg.option('e').value.set('49152')
|
|
cfg.option('e').value.set('65535')
|
|
|
|
raises(ValueError, "cfg.option('f').value.set('0')")
|
|
cfg.option('f').value.set('1')
|
|
cfg.option('f').value.set('1023')
|
|
cfg.option('f').value.set('1024')
|
|
cfg.option('f').value.set('49151')
|
|
cfg.option('f').value.set('49152')
|
|
cfg.option('f').value.set('65535')
|
|
raises(ValueError, "cfg.option('f').value.set('65536')")
|
|
|
|
|
|
def test_port_range():
|
|
a = PortOption('a', '', allow_range=True)
|
|
b = PortOption('b', '', allow_range=True, allow_zero=True)
|
|
c = PortOption('c', '', allow_range=True, allow_zero=True, allow_registred=False)
|
|
d = PortOption('d', '', allow_range=True, allow_zero=True, allow_wellknown=False, allow_registred=False)
|
|
e = PortOption('e', '', allow_range=True, allow_zero=True, allow_private=True)
|
|
f = PortOption('f', '', allow_range=True, allow_private=True)
|
|
od = OptionDescription('od', '', [a, b, c, d, e, f])
|
|
cfg = Config(od)
|
|
raises(ValueError, "cfg.option('a').value.set('0')")
|
|
cfg.option('a').value.set('1')
|
|
cfg.option('a').value.set('1023')
|
|
cfg.option('a').value.set('1024')
|
|
cfg.option('a').value.set('49151')
|
|
raises(ValueError, "cfg.option('a').value.set('49152')")
|
|
raises(ValueError, "cfg.option('a').value.set('65535')")
|
|
raises(ValueError, "cfg.option('a').value.set('65536')")
|
|
cfg.option('a').value.set('1:49151')
|
|
raises(ValueError, "cfg.option('a').value.set('0:49151')")
|
|
raises(ValueError, "cfg.option('a').value.set('1:49152')")
|
|
|
|
cfg.option('b').value.set('0')
|
|
cfg.option('b').value.set('1')
|
|
cfg.option('b').value.set('1023')
|
|
cfg.option('b').value.set('1024')
|
|
cfg.option('b').value.set('49151')
|
|
raises(ValueError, "cfg.option('b').value.set('49152')")
|
|
raises(ValueError, "cfg.option('b').value.set('65535')")
|
|
raises(ValueError, "cfg.option('b').value.set('65536')")
|
|
cfg.option('b').value.set('0:49151')
|
|
raises(ValueError, "cfg.option('b').value.set('0:49152')")
|
|
|
|
cfg.option('c').value.set('0')
|
|
cfg.option('c').value.set('1')
|
|
cfg.option('c').value.set('1023')
|
|
raises(ValueError, "cfg.option('c').value.set('1024')")
|
|
raises(ValueError, "cfg.option('c').value.set('49151')")
|
|
raises(ValueError, "cfg.option('c').value.set('49152')")
|
|
raises(ValueError, "cfg.option('c').value.set('65535')")
|
|
raises(ValueError, "cfg.option('c').value.set('65536')")
|
|
cfg.option('c').value.set('0:1023')
|
|
raises(ValueError, "cfg.option('c').value.set('0:1024')")
|
|
|
|
cfg.option('d').value.set('0')
|
|
raises(ValueError, "cfg.option('d').value.set('1')")
|
|
raises(ValueError, "cfg.option('d').value.set('1023')")
|
|
raises(ValueError, "cfg.option('d').value.set('1024')")
|
|
raises(ValueError, "cfg.option('d').value.set('49151')")
|
|
raises(ValueError, "cfg.option('d').value.set('49152')")
|
|
raises(ValueError, "cfg.option('d').value.set('65535')")
|
|
raises(ValueError, "cfg.option('d').value.set('65536')")
|
|
raises(ValueError, "cfg.option('d').value.set('0:0')")
|
|
raises(ValueError, "cfg.option('d').value.set('0:1')")
|
|
|
|
cfg.option('e').value.set('0')
|
|
cfg.option('e').value.set('1')
|
|
cfg.option('e').value.set('1023')
|
|
cfg.option('e').value.set('1024')
|
|
cfg.option('e').value.set('49151')
|
|
cfg.option('e').value.set('49152')
|
|
cfg.option('e').value.set('65535')
|
|
cfg.option('e').value.set('0:65535')
|
|
raises(ValueError, "cfg.option('e').value.set('0:65536')")
|
|
|
|
raises(ValueError, "cfg.option('f').value.set('0')")
|
|
cfg.option('f').value.set('1')
|
|
cfg.option('f').value.set('1023')
|
|
cfg.option('f').value.set('1024')
|
|
cfg.option('f').value.set('49151')
|
|
cfg.option('f').value.set('49152')
|
|
cfg.option('f').value.set('65535')
|
|
raises(ValueError, "cfg.option('f').value.set('65536')")
|
|
cfg.option('f').value.set('1:65535')
|
|
cfg.option('f').value.set('3:4')
|
|
raises(ValueError, "cfg.option('f').value.set('0:65535')")
|
|
raises(ValueError, "cfg.option('f').value.set('4:3')")
|