mirror of
https://github.com/kasmtech/KasmVNC.git
synced 2024-11-24 09:03:30 +01:00
119 lines
3.8 KiB
Python
119 lines
3.8 KiB
Python
import os
|
|
import shutil
|
|
import subprocess
|
|
from path import Path
|
|
from mamba import description, context, it, before, after
|
|
from expects import expect, equal
|
|
|
|
vncserver_cmd = 'vncserver :1 -cert /etc/ssl/certs/ssl-cert-snakeoil.pem -key /etc/ssl/private/ssl-cert-snakeoil.key -sslOnly -FrameRate=24 -interface 0.0.0.0 -httpd /usr/share/kasmvnc/www -depth 24 -geometry 1280x1050'
|
|
|
|
|
|
def clean_env():
|
|
home_dir = os.environ['HOME']
|
|
password_file = os.path.join(home_dir, ".kasmpasswd")
|
|
Path(password_file).remove_p()
|
|
|
|
vnc_dir = os.path.join(home_dir, ".vnc")
|
|
Path(vnc_dir).rmtree(ignore_errors=True)
|
|
|
|
|
|
def run_cmd(cmd, **kwargs):
|
|
completed_process = subprocess.run(cmd, shell=True, text=True,
|
|
capture_output=True,
|
|
executable='/bin/bash', **kwargs)
|
|
if completed_process.returncode != 0:
|
|
print(completed_process.stdout)
|
|
print(completed_process.stderr)
|
|
|
|
return completed_process
|
|
|
|
|
|
def add_kasmvnc_user_docker():
|
|
completed_process = run_cmd('echo -e "password\\npassword\\n" | vncpasswd -u docker')
|
|
expect(completed_process.returncode).to(equal(0))
|
|
|
|
|
|
def kill_xvnc():
|
|
run_cmd('vncserver -kill :1')
|
|
|
|
|
|
def select_de(de_name):
|
|
try:
|
|
cmd = f'{vncserver_cmd} -select-de {de_name}'
|
|
completed_process = run_cmd(cmd)
|
|
expect(completed_process.returncode).to(equal(0))
|
|
finally:
|
|
kill_xvnc()
|
|
|
|
|
|
def check_de_was_setup_to_run(de_name):
|
|
completed_process = run_cmd(f'grep -q {de_name} ~/.vnc/xstartup')
|
|
expect(completed_process.returncode).to(equal(0))
|
|
|
|
|
|
with description('vncserver') as self:
|
|
with before.each:
|
|
clean_env()
|
|
with after.each:
|
|
kill_xvnc()
|
|
|
|
with context('on the first run'):
|
|
with it('asks user to select a DE on the first run'):
|
|
add_kasmvnc_user_docker()
|
|
|
|
choose_cinnamon = "1\n"
|
|
completed_process = run_cmd(vncserver_cmd, input=choose_cinnamon)
|
|
expect(completed_process.returncode).to(equal(0))
|
|
|
|
check_de_was_setup_to_run('cinnamon')
|
|
|
|
with it('asks to select a DE, when ran with -select-de'):
|
|
add_kasmvnc_user_docker()
|
|
|
|
cmd = f'{vncserver_cmd} -select-de'
|
|
choose_cinnamon_and_answer_yes = "1\ny\n"
|
|
completed_process = run_cmd(cmd, input=choose_cinnamon_and_answer_yes)
|
|
expect(completed_process.returncode).to(equal(0))
|
|
|
|
check_de_was_setup_to_run('cinnamon')
|
|
|
|
with it('selects passed DE with -s'):
|
|
add_kasmvnc_user_docker()
|
|
|
|
cmd = f'{vncserver_cmd} -select-de mate'
|
|
completed_process = run_cmd(cmd)
|
|
expect(completed_process.returncode).to(equal(0))
|
|
|
|
check_de_was_setup_to_run('mate')
|
|
|
|
with context('after DE was selected'):
|
|
with it("don't ask to choose DE by default"):
|
|
add_kasmvnc_user_docker()
|
|
select_de('mate')
|
|
|
|
completed_process = run_cmd(vncserver_cmd)
|
|
expect(completed_process.returncode).to(equal(0))
|
|
|
|
check_de_was_setup_to_run('mate')
|
|
|
|
with it('asks to select a DE, when ran with -select-de'):
|
|
add_kasmvnc_user_docker()
|
|
select_de('mate')
|
|
|
|
cmd = f'{vncserver_cmd} -select-de'
|
|
choose_cinnamon_and_answer_yes = "1\ny\n"
|
|
completed_process = run_cmd(cmd, input=choose_cinnamon_and_answer_yes)
|
|
expect(completed_process.returncode).to(equal(0))
|
|
|
|
check_de_was_setup_to_run('cinnamon')
|
|
|
|
with it('selects passed DE with -s'):
|
|
add_kasmvnc_user_docker()
|
|
select_de('mate')
|
|
|
|
cmd = f'{vncserver_cmd} -select-de cinnamon'
|
|
completed_process = run_cmd(cmd)
|
|
expect(completed_process.returncode).to(equal(0))
|
|
|
|
check_de_was_setup_to_run('cinnamon')
|