def test_run_cmd(protocol_fake):
# TODO this test should cover __init__ method
s = Session('windows-host', auth=('john.smith', 'secret'))
s.protocol = protocol_fake
r = s.run_cmd('ipconfig', ['/all'])
assert r.status_code == 0
assert b'Windows IP Configuration' in r.std_out
assert len(r.std_err) == 0
def test_run_cmd(protocol_fake):
# TODO this test should cover __init__ method
s = Session('windows-host', auth=('john.smith', 'secret'))
s.protocol = protocol_fake
r = s.run_cmd('ipconfig', ['/all'])
assert r.status_code == 0
assert 'Windows IP Configuration' in r.std_out
assert len(r.std_err) == 0
def test_run_ps_with_error(protocol_fake):
# TODO this test should cover __init__ method
s = Session('windows-host', auth=('john.smith', 'secret'))
s.protocol = protocol_fake
r = s.run_ps('Write-Error "Error"')
assert r.status_code == 1
assert b'Write-Error "Error"' in r.std_err
assert len(r.std_out) == 0
def test_run_ps(protocol_fake):
s = Session('windows-host', auth=('john.smith', 'secret'))
s.protocol = protocol_fake
script = """
$strComputer = $Host
Clear
$RAM = WmiObject Win32_ComputerSystem
$MB = 1048576
"Installed Memory: " + [int]($RAM.TotalPhysicalMemory /$MB) + " MB"
"""
r = s.run_ps(script)
assert r.status_code == 0
assert b'Installed Memory: 2044 MB' in r.std_out
assert len(r.std_err) == 0
