diff options
| -rwxr-xr-x | scripts/cli/test_vpn_anyconnect.py | 58 | 
1 files changed, 58 insertions, 0 deletions
diff --git a/scripts/cli/test_vpn_anyconnect.py b/scripts/cli/test_vpn_anyconnect.py new file mode 100755 index 000000000..dd8ab1609 --- /dev/null +++ b/scripts/cli/test_vpn_anyconnect.py @@ -0,0 +1,58 @@ +#!/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 re +import os +import unittest + +from psutil import process_iter +from vyos.configsession import ConfigSession, ConfigSessionError +from vyos.util import read_file + +OCSERV_CONF = '/run/ocserv/ocserv.conf' +base_path = ['vpn', 'anyconnect'] +cert = '/etc/ssl/certs/ssl-cert-snakeoil.pem' +cert_key = '/etc/ssl/private/ssl-cert-snakeoil.key' + +class TestVpnAnyconnect(unittest.TestCase): +    def setUp(self): +        self.session = ConfigSession(os.getpid()) + +    def tearDown(self): +        # Delete vpn anyconnect configuration +        self.session.delete(base_path) +        self.session.commit() + +        del self.session + +    def test_vpn(self): +        user = 'vyos_user' +        password = 'vyos_pass' +        self.session.delete(base_path) +        self.session.set(base_path + ["authentication", "local-users", "username", user, "password", password]) +        self.session.set(base_path + ["authentication", "mode", "local"]) +        self.session.set(base_path + ["network-settings", "client-ip-settings", "subnet", "192.0.2.0/24"]) +        self.session.set(base_path + ["ssl", "ca-cert-file", cert]) +        self.session.set(base_path + ["ssl", "cert-file", cert]) +        self.session.set(base_path + ["ssl", "key-file", cert_key]) + +        self.session.commit() + +        # Check for running process +        self.assertTrue("ocserv-main" in (p.name() for p in process_iter())) + +if __name__ == '__main__': +    unittest.main()  | 
