From 50315abd38be71554a2daf18f8c3c23bf8690bcf Mon Sep 17 00:00:00 2001 From: Christian Poessinger Date: Sun, 9 Feb 2020 18:43:08 +0100 Subject: ssh: add test for proper port and host validation --- scripts/cli/test_service_ssh.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) (limited to 'scripts/cli') diff --git a/scripts/cli/test_service_ssh.py b/scripts/cli/test_service_ssh.py index e272872ea..f0ad2e565 100755 --- a/scripts/cli/test_service_ssh.py +++ b/scripts/cli/test_service_ssh.py @@ -14,15 +14,23 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see . +import re import os import unittest from psutil import process_iter from vyos.config import Config from vyos.configsession import ConfigSession, ConfigSessionError +from vyos.util import read_file +SSHD_CONF = '/etc/ssh/sshd_config' base_path = ['service', 'ssh'] +def get_config_value(key): + tmp = read_file(SSHD_CONF) + return re.findall(r'\n?{}\s+(.*)'.format(key), tmp) + + class TestServiceSSH(unittest.TestCase): def setUp(self): self.session = ConfigSession(os.getpid()) @@ -38,11 +46,20 @@ class TestServiceSSH(unittest.TestCase): def test_ssh(self): """ Check if SSH service can be configured and runs """ - self.session.set(base_path) + self.session.set(base_path + ['port', '2222']) + self.session.set(base_path + ['disable-host-validation']) # commit changes self.session.commit() + # Check configured port + port = get_config_value('Port')[0] + self.assertTrue("2222" in port) + + # Check DNS usage + dns = get_config_value('UseDNS')[0] + self.assertTrue("no" in dns) + # Check for running process self.assertTrue("sshd" in (p.name() for p in process_iter())) -- cgit v1.2.3