diff options
| -rwxr-xr-x | scripts/cli/test_service_ssh.py | 19 |
1 files changed, 18 insertions, 1 deletions
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 <http://www.gnu.org/licenses/>. +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())) |