diff --git a/lib/tlitest/test_tlog_rec_session.py b/lib/tlitest/test_tlog_rec_session.py index eba34d83..0154b40d 100644 --- a/lib/tlitest/test_tlog_rec_session.py +++ b/lib/tlitest/test_tlog_rec_session.py @@ -217,10 +217,14 @@ def test_session_record_pipe_io_stdin(self): """ text_in_stdio = 'print("hello world")\n' text_out = "hello world" - p = Popen(['sshpass', '-p', 'Secret123', 'ssh', '-o', - 'StrictHostKeyChecking=no', - 'tlitestlocaluser2@localhost', 'python3'], - stdout=PIPE, stdin=PIPE, stderr=PIPE, encoding='utf8') + + sessionclass = TlogRecSessionConfig(writer="syslog") + sessionclass.generate_config(SYSTEM_TLOG_REC_SESSION_CONF) + + p = Popen(['ssh', '-i', f'~/.ssh/id_rsa_{self.user}', + '-o', 'StrictHostKeyChecking=no', + f'{self.user}@localhost', 'python3'], + stdout=PIPE, stdin=PIPE, stderr=PIPE, encoding='utf8') stdout_data = p.communicate(input=text_in_stdio)[0] assert text_out in stdout_data diff --git a/src/tlitest/tlitest-setup b/src/tlitest/tlitest-setup index 3749b952..7250a22e 100755 --- a/src/tlitest/tlitest-setup +++ b/src/tlitest/tlitest-setup @@ -8,7 +8,6 @@ python3-pytest python3-pexpect python3-systemd tcsh -sshpass " [[ -z "${CONTAINER_ENV}" ]] && PKGS+="tlog" @@ -61,3 +60,12 @@ echo "%wheel ALL=(ALL) NOPASSWD: ALL" > \ usermod tlitestlocaladmin1 -aG wheel,systemd-journal usermod tlitestlocaluser1 -aG systemd-journal usermod tlitestlocaluser2 -s /usr/bin/tlog-rec-session + +# Generate ssh key for testing +user="tlitestlocaluser2" +rm -rf ~/.ssh/id_rsa_${user}* +ssh-keygen -t rsa -b 2048 -N "" -C stdin_test -f ~/.ssh/id_rsa_${user} +user_dir=$(eval echo "~${user}") +mkdir -p ${user_dir}/.ssh +cat ~/.ssh/id_rsa_${user}.pub >> ${user_dir}/.ssh/authorized_keys +chown -R ${user}:${user} ${user_dir}/.ssh diff --git a/src/tlitest/tlitest-teardown b/src/tlitest/tlitest-teardown index aa0fa8a7..bc8397a3 100755 --- a/src/tlitest/tlitest-teardown +++ b/src/tlitest/tlitest-teardown @@ -48,3 +48,8 @@ if [ -f /etc/sudoers.d/01_wheel_nopass_tlitest ]; then echo "Found test sudoers file...removing" rm /etc/sudoers.d/01_wheel_nopass_tlitest fi + +user="tlitestlocaluser2" +if [ -f ~/.ssh/id_rsa_${user} ]; then + rm -rf ~/.ssh/id_rsa_${user}* +fi