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
39
40
41
42
43
44
45
46
47
|
from __future__ import annotations
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) -> None:
cmd = "powershell.exe" if sys.platform == "win32" else "pwsh"
super().__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-8-sig"
def _get_test_lines(self, activate_script):
return super()._get_test_lines(activate_script)
def invoke_script(self):
return [self.cmd, "-File"]
def print_prompt(self):
return "prompt"
def quote(self, s):
"""
Tester will pass strings to native commands on Windows so extra
parsing rules are used. Check `PowerShellActivator.quote` for more
details.
"""
text = PowerShellActivator.quote(s)
return text.replace('"', '""') if sys.platform == "win32" else text
def activate_call(self, script):
# Commands are called without quotes in PowerShell
cmd = self.activate_cmd
scr = self.quote(str(script))
return f"{cmd} {scr}".strip()
activation_tester(PowerShell)
|
