def test_checksum(self):
""" ensures checksum of same config doesn't change """
o = OpenWisp({"general": {"hostname": "test"}})
# md5 is good enough and won't slow down test execution too much
checksum1 = md5(o.generate().getvalue()).hexdigest()
sleep(1)
checksum2 = md5(o.generate().getvalue()).hexdigest()
self.assertEqual(checksum1, checksum2)
def test_hostname_required(self):
o = OpenWisp({
"general": {
"timezone": "Coordinated Universal Time"
}
})
with self.assertRaises(ValidationError):
o.validate()
def test_checksum(self):
""" ensures checksum of same config doesn't change """
o = OpenWisp({"general": {"hostname": "test"}})
# md5 is good enough and won't slow down test execution too much
checksum1 = md5(o.generate().getvalue()).hexdigest()
sleep(1)
checksum2 = md5(o.generate().getvalue()).hexdigest()
self.assertEqual(checksum1, checksum2)
def test_ensure_tun_vpn_ignored(self):
config = deepcopy(self.config)
config['openvpn'][0]['dev_type'] = 'tun'
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
install = tar.getmember('install.sh')
contents = tar.extractfile(install).read().decode()
self.assertNotIn('openvpn --mktun --dev 2693 --dev-type tap', contents)
tar.close()
def test_ensure_tun_vpn_ignored(self):
config = deepcopy(self.config)
config["openvpn"][0]["dev_type"] = "tun"
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode="r")
install = tar.getmember("install.sh")
contents = tar.extractfile(install).read().decode()
self.assertNotIn("openvpn --mktun --dev 2693 --dev-type tap", contents)
tar.close()
def test_ensure_tun_vpn_ignored(self):
config = deepcopy(self.config)
config['openvpn'][0]['dev_type'] = 'tun'
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
install = tar.getmember('install.sh')
contents = tar.extractfile(install).read().decode()
self.assertNotIn('openvpn --mktun --dev 2693 --dev-type tap', contents)
tar.close()
def test_uninstall_script(self):
config = deepcopy(self.config)
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
uninstall = tar.getmember('uninstall.sh')
contents = tar.extractfile(uninstall).read().decode()
self.assertIn('openvpn --rmtun --dev 2693 --dev-type tap', contents)
self.assertNotIn('Stopping Cron', contents)
# esure is executable
self.assertEqual(uninstall.mode, 493)
tar.close()
def test_uninstall_script(self):
config = deepcopy(self.config)
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
uninstall = tar.getmember('uninstall.sh')
contents = tar.extractfile(uninstall).read().decode()
self.assertIn('openvpn --rmtun --dev 2693 --dev-type tap', contents)
self.assertNotIn('Stopping Cron', contents)
# esure is executable
self.assertEqual(uninstall.mode, 493)
tar.close()
def test_uci(self):
o = OpenWisp({"general": {"hostname": "openwisp-test"}})
tar = tarfile.open(fileobj=o.generate(), mode='r')
system = tar.getmember('uci/system.conf')
contents = tar.extractfile(system).read().decode()
expected = self._tabs("""package system
config system 'system'
option hostname 'openwisp-test'
""")
self.assertEqual(contents, expected)
tar.close()
def test_up_and_down_scripts(self):
config = deepcopy(self.config)
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
up = tar.getmember('openvpn/vpn_2693_script_up.sh')
contents = tar.extractfile(up).read().decode()
self.assertIn('rm -f /tmp/will_reboot', contents)
self.assertEqual(up.mode, 493) # esure is executable
down = tar.getmember('openvpn/vpn_2693_script_down.sh')
contents = tar.extractfile(down).read().decode()
self.assertIn('REBOOT_DELAY', contents)
self.assertEqual(down.mode, 493) # esure is executable
tar.close()
def test_up_and_down_scripts(self):
config = deepcopy(self.config)
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
up = tar.getmember('openvpn/vpn_2693_script_up.sh')
contents = tar.extractfile(up).read().decode()
self.assertIn('rm -f /tmp/will_reboot', contents)
self.assertEqual(up.mode, 493) # esure is executable
down = tar.getmember('openvpn/vpn_2693_script_down.sh')
contents = tar.extractfile(down).read().decode()
self.assertIn('REBOOT_DELAY', contents)
self.assertEqual(down.mode, 493) # esure is executable
tar.close()
def test_install_script(self):
config = deepcopy(self.config)
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
install = tar.getmember('install.sh')
contents = tar.extractfile(install).read().decode()
self.assertIn('openvpn --mktun --dev 2693 --dev-type tap', contents)
self.assertIn('ifup br-serv', contents)
self.assertIn('$(ip address show dev br-serv | grep 192.168.1.2)', contents)
self.assertIn('wifi up radio0', contents)
self.assertNotIn('Starting Cron', contents)
# esure is executable
self.assertEqual(install.mode, 493)
tar.close()
def test_cron(self):
config = deepcopy(self.config)
config["files"] = [
{"path": "/crontabs/root", "mode": "0644", "contents": "* * * * * echo 'test' > /tmp/test-cron"}
]
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode="r")
install = tar.getmember("install.sh")
contents = tar.extractfile(install).read().decode()
self.assertIn("Starting Cron", contents)
uninstall = tar.getmember("uninstall.sh")
contents = tar.extractfile(uninstall).read().decode()
self.assertIn("Stopping Cron", contents)
tar.close()
def test_tc_script(self):
config = deepcopy(self.config)
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
tc = tar.getmember('tc_script.sh')
contents = tar.extractfile(tc).read().decode()
self.assertIn('tc qdisc del dev tap0 root', contents)
self.assertIn('tc qdisc del dev tap0 ingress', contents)
self.assertIn('tc qdisc add dev tap0 root handle 1: htb default 2', contents)
self.assertIn('tc class add dev tap0 parent 1 classid 1:1 htb rate 1024kbit burst 191k', contents)
self.assertIn('tc class add dev tap0 parent 1:1 classid 1:2 htb rate 512kbit ceil 1024kbit', contents)
self.assertIn('tc qdisc add dev tap0 ingress', contents)
self.assertIn('tc filter add dev tap0 parent ffff: preference 0 u32 match u32 0x0 0x0 police rate 2048kbit burst 383k drop flowid :1', contents)
tar.close()
def test_install_script(self):
config = deepcopy(self.config)
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
install = tar.getmember('install.sh')
contents = tar.extractfile(install).read().decode()
self.assertIn('openvpn --mktun --dev 2693 --dev-type tap', contents)
self.assertIn('ifup br-serv', contents)
self.assertIn('$(ip address show dev br-serv | grep 192.168.1.2)', contents)
self.assertIn('wifi up radio0', contents)
self.assertNotIn('Starting Cron', contents)
# esure is executable
self.assertEqual(install.mode, 493)
tar.close()
def test_uci(self):
o = OpenWisp({"general": {"hostname": "openwisp-test"}})
tar = tarfile.open(fileobj=o.generate(), mode="r")
system = tar.getmember("uci/system.conf")
contents = tar.extractfile(system).read().decode()
expected = self._tabs(
"""package system
config system 'system'
option hostname 'openwisp-test'
option timezone 'UTC'
"""
)
self.assertEqual(contents, expected)
tar.close()
def test_tc_script(self):
config = deepcopy(self.config)
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
tc = tar.getmember('tc_script.sh')
contents = tar.extractfile(tc).read().decode()
self.assertIn('tc qdisc del dev tap0 root', contents)
self.assertIn('tc qdisc del dev tap0 ingress', contents)
self.assertIn('tc qdisc add dev tap0 root handle 1: htb default 2', contents)
self.assertIn('tc class add dev tap0 parent 1 classid 1:1 htb rate 1024kbit burst 191k', contents)
self.assertIn('tc class add dev tap0 parent 1:1 classid 1:2 htb rate 512kbit ceil 1024kbit', contents)
self.assertIn('tc qdisc add dev tap0 ingress', contents)
l = 'tc filter add dev tap0 parent ffff: preference 0 u32 match u32 0x0 0x0 police '\
'rate 2048kbit burst 383k drop flowid :1'
self.assertIn(l, contents)
tar.close()
def test_cron(self):
config = deepcopy(self.config)
config['files'] = [{
"path": "/crontabs/root",
"mode": "0644",
"contents": "* * * * * echo 'test' > /tmp/test-cron"
}]
o = OpenWisp(config)
tar = tarfile.open(fileobj=o.generate(), mode='r')
install = tar.getmember('install.sh')
contents = tar.extractfile(install).read().decode()
self.assertIn('Starting Cron', contents)
uninstall = tar.getmember('uninstall.sh')
contents = tar.extractfile(uninstall).read().decode()
self.assertIn('Stopping Cron', contents)
tar.close()
def test_hostname_required(self):
o = OpenWisp({"general": {"timezone": "UTC"}})
with self.assertRaises(ValidationError):
o.validate()
def test_double_generation(self):
o = OpenWisp(self.config)
o.generate()
o.generate()
def test_wireless_radio_disabled_0(self):
o = OpenWisp({'radios': self.config['radios']})
output = o.render()
self.assertIn("option disabled '0'", output)
def test_double_generation(self):
o = OpenWisp(self.config)
o.generate()
o.generate()
def test_wireless_radio_disabled_0(self):
o = OpenWisp({
'radios': self.config['radios']
})
output = o.render()
self.assertIn("option disabled '0'", output)
def test_hostname_required(self):
o = OpenWisp({"general": {"timezone": "UTC"}})
with self.assertRaises(ValidationError):
o.validate()