-
-
Notifications
You must be signed in to change notification settings - Fork 1k
/
test_powershell.py
38 lines (27 loc) · 1.29 KB
/
test_powershell.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from __future__ import absolute_import, unicode_literals
import pipes
import sys
import pytest
from virtualenv.activation import PowerShellActivator
@pytest.mark.slow
def test_powershell(activation_tester_class, activation_tester, monkeypatch):
monkeypatch.setenv("TERM", "xterm")
class PowerShell(activation_tester_class):
def __init__(self, session):
cmd = "powershell.exe" if sys.platform == "win32" else "pwsh"
super(PowerShell, self).__init__(PowerShellActivator, session, cmd, "activate.ps1", "ps1")
self._version_cmd = [cmd, "-c", "$PSVersionTable"]
self._invoke_script = [cmd, "-ExecutionPolicy", "ByPass", "-File"]
self.activate_cmd = "."
self.script_encoding = "utf-16"
def quote(self, s):
"""powershell double double quote needed for quotes within single quotes"""
return pipes.quote(s).replace('"', '""')
def _get_test_lines(self, activate_script):
# for BATCH utf-8 support need change the character code page to 650001
return super(PowerShell, self)._get_test_lines(activate_script)
def invoke_script(self):
return [self.cmd, "-File"]
def print_prompt(self):
return "prompt"
activation_tester(PowerShell)