diff options
Diffstat (limited to 'scripts/cli/test_interfaces_wireguard.py')
-rwxr-xr-x | scripts/cli/test_interfaces_wireguard.py | 68 |
1 files changed, 0 insertions, 68 deletions
diff --git a/scripts/cli/test_interfaces_wireguard.py b/scripts/cli/test_interfaces_wireguard.py deleted file mode 100755 index 0c32a4696..000000000 --- a/scripts/cli/test_interfaces_wireguard.py +++ /dev/null @@ -1,68 +0,0 @@ -#!/usr/bin/env python3 -# -# Copyright (C) 2020 VyOS maintainers and contributors -# -# This program is free software; you can redistribute it and/or modify -# it under the terms of the GNU General Public License version 2 or later as -# published by the Free Software Foundation. -# -# This program is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU General Public License for more details. -# -# You should have received a copy of the GNU General Public License -# along with this program. If not, see <http://www.gnu.org/licenses/>. - -import os -import unittest - -from vyos.configsession import ConfigSession, ConfigSessionError -from base_interfaces_test import BasicInterfaceTest - -# Generate WireGuard default keypair -if not os.path.isdir('/config/auth/wireguard/default'): - os.system('sudo /usr/libexec/vyos/op_mode/wireguard.py --genkey') - -base_path = ['interfaces', 'wireguard'] - -class WireGuardInterfaceTest(unittest.TestCase): - def setUp(self): - self.session = ConfigSession(os.getpid()) - self._test_addr = ['192.0.2.1/26', '192.0.2.255/31', '192.0.2.64/32', - '2001:db8:1::ffff/64', '2001:db8:101::1/112'] - self._interfaces = ['wg0', 'wg1'] - - def tearDown(self): - self.session.delete(base_path) - self.session.commit() - del self.session - - def test_peer_setup(self): - """ - Create WireGuard interfaces with associated peers - """ - for intf in self._interfaces: - peer = 'foo-' + intf - psk = 'u2xdA70hkz0S1CG0dZlOh0aq2orwFXRIVrKo4DCvHgM=' - pubkey = 'n6ZZL7ph/QJUJSUUTyu19c77my1dRCDHkMzFQUO9Z3A=' - - for addr in self._test_addr: - self.session.set(base_path + [intf, 'address', addr]) - - self.session.set(base_path + [intf, 'peer', peer, 'address', '127.0.0.1']) - self.session.set(base_path + [intf, 'peer', peer, 'port', '1337']) - - # Allow different prefixes to traverse the tunnel - allowed_ips = ['10.0.0.0/8', '172.16.0.0/12', '192.168.0.0/16'] - for ip in allowed_ips: - self.session.set(base_path + [intf, 'peer', peer, 'allowed-ips', ip]) - - self.session.set(base_path + [intf, 'peer', peer, 'preshared-key', psk]) - self.session.set(base_path + [intf, 'peer', peer, 'pubkey', pubkey]) - self.session.commit() - - self.assertTrue(os.path.isdir(f'/sys/class/net/{intf}')) - -if __name__ == '__main__': - unittest.main() |