104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
# coding=utf-8
|
|
import re
|
|
import pytest
|
|
xfail = pytest.mark.xfail
|
|
|
|
|
|
def test_unicode_roundtrip(protocol_real):
|
|
shell_id = protocol_real.open_shell(codepage=65001)
|
|
command_id = protocol_real.run_command(
|
|
shell_id, u'PowerShell', arguments=['-Command', 'Write-Host', u'こんにちは'])
|
|
|
|
try:
|
|
std_out, std_err, status_code = protocol_real.get_command_output(
|
|
shell_id, command_id)
|
|
assert status_code == 0
|
|
assert len(std_err) == 0
|
|
# std_out will be returned as UTF-8, but PEP8 won't let us store a
|
|
# UTF-8 string literal, so we'll convert it on the fly
|
|
assert std_out == (u'こんにちは\n'.encode('utf-8'))
|
|
|
|
finally:
|
|
protocol_real.cleanup_command(shell_id, command_id)
|
|
protocol_real.close_shell(shell_id)
|
|
|
|
|
|
def test_open_shell_and_close_shell(protocol_real):
|
|
shell_id = protocol_real.open_shell()
|
|
assert re.match(r'^\w{8}-\w{4}-\w{4}-\w{4}-\w{12}$', shell_id)
|
|
|
|
protocol_real.close_shell(shell_id)
|
|
|
|
|
|
def test_run_command_with_arguments_and_cleanup_command(protocol_real):
|
|
shell_id = protocol_real.open_shell()
|
|
command_id = protocol_real.run_command(shell_id, 'ipconfig', ['/all'])
|
|
assert re.match(r'^\w{8}-\w{4}-\w{4}-\w{4}-\w{12}$', command_id)
|
|
|
|
protocol_real.cleanup_command(shell_id, command_id)
|
|
protocol_real.close_shell(shell_id)
|
|
|
|
|
|
def test_run_command_without_arguments_and_cleanup_command(protocol_real):
|
|
shell_id = protocol_real.open_shell()
|
|
command_id = protocol_real.run_command(shell_id, 'hostname')
|
|
assert re.match(r'^\w{8}-\w{4}-\w{4}-\w{4}-\w{12}$', command_id)
|
|
|
|
protocol_real.cleanup_command(shell_id, command_id)
|
|
protocol_real.close_shell(shell_id)
|
|
|
|
|
|
def test_run_command_with_env(protocol_real):
|
|
|
|
shell_id = protocol_real.open_shell(env_vars=dict(TESTENV1='hi mom', TESTENV2='another var'))
|
|
command_id = protocol_real.run_command(shell_id, 'echo', ['%TESTENV1%', '%TESTENV2%'])
|
|
std_out, std_err, status_code = protocol_real.get_command_output(shell_id, command_id)
|
|
assert re.search(b'hi mom another var', std_out)
|
|
|
|
protocol_real.cleanup_command(shell_id, command_id)
|
|
protocol_real.close_shell(shell_id)
|
|
|
|
|
|
def test_get_command_output(protocol_real):
|
|
shell_id = protocol_real.open_shell()
|
|
command_id = protocol_real.run_command(shell_id, 'ipconfig', ['/all'])
|
|
std_out, std_err, status_code = protocol_real.get_command_output(
|
|
shell_id, command_id)
|
|
|
|
assert status_code == 0
|
|
assert b'Windows IP Configuration' in std_out
|
|
assert len(std_err) == 0
|
|
|
|
protocol_real.cleanup_command(shell_id, command_id)
|
|
protocol_real.close_shell(shell_id)
|
|
|
|
|
|
def test_run_command_taking_more_than_operation_timeout_sec(protocol_real):
|
|
shell_id = protocol_real.open_shell()
|
|
command_id = protocol_real.run_command(
|
|
shell_id, 'PowerShell -Command Start-Sleep -s {0}'.format(protocol_real.operation_timeout_sec * 2))
|
|
assert re.match(r'^\w{8}-\w{4}-\w{4}-\w{4}-\w{12}$', command_id)
|
|
std_out, std_err, status_code = protocol_real.get_command_output(
|
|
shell_id, command_id)
|
|
|
|
assert status_code == 0
|
|
assert len(std_err) == 0
|
|
|
|
protocol_real.cleanup_command(shell_id, command_id)
|
|
protocol_real.close_shell(shell_id)
|
|
|
|
|
|
@xfail()
|
|
def test_set_timeout(protocol_real):
|
|
raise NotImplementedError()
|
|
|
|
|
|
@xfail()
|
|
def test_set_max_env_size(protocol_real):
|
|
raise NotImplementedError()
|
|
|
|
|
|
@xfail()
|
|
def test_set_locale(protocol_real):
|
|
raise NotImplementedError()
|