From 0ffc1fa39944a202dc8cfd1a77a2493e59c84755 Mon Sep 17 00:00:00 2001 From: Christian Poessinger Date: Sun, 9 Feb 2020 15:45:59 +0100 Subject: ssh: basic test to see if SSH server launches --- scripts/cli/test_service_ssh.py | 47 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 47 insertions(+) create mode 100755 scripts/cli/test_service_ssh.py diff --git a/scripts/cli/test_service_ssh.py b/scripts/cli/test_service_ssh.py new file mode 100755 index 000000000..b5247d77a --- /dev/null +++ b/scripts/cli/test_service_ssh.py @@ -0,0 +1,47 @@ +#!/usr/bin/env python3 +# +# Copyright (C) 2019-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 . + +import os +import unittest + +from psutil import process_iter +from vyos.config import Config +from vyos.configsession import ConfigSession, ConfigSessionError + +base_path = ['service', 'ssh'] + +class TestServiceSSH(unittest.TestCase): + def setUp(self): + self.session = ConfigSession(os.getpid()) + env = self.session.get_session_env() + self.config = Config(session_env=env) + + def tearDown(self): + # Delete SSH configuration + self.session.delete(base_path) + self.session.commit() + + del self.session + + def test_ssh(self): + """ Check if SSH service can be configured and runs """ + self.session.set(base_path) + + # Check for running process + self.assertTrue("sshd" in (p.name() for p in process_iter())) + +if __name__ == '__main__': + unittest.main() -- cgit v1.2.3