fix test

microsoft · Nov 15, 2024 · 3d117a1 · 3d117a1
1 parent 9b26fde
commit 3d117a1
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python_files/tests/test_shell_integration.py b/python_files/tests/test_shell_integration.py
@@ -1,17 +1,20 @@
 import importlib
+import platform
 import sys
 from unittest.mock import Mock
 
 import pythonrc
 
+is_wsl = "microsoft-standard-WSL" in platform.release()
+
 
 def test_decoration_success():
     importlib.reload(pythonrc)
     ps1 = pythonrc.PS1()
 
     ps1.hooks.failure_flag = False
     result = str(ps1)
-    if sys.platform != "win32":
+    if sys.platform != "win32" and (not is_wsl):
         assert (
             result
             == "\x1b]633;E;None\x07\x1b]633;D;0\x07\x1b]633;A\x07>>> \x1b]633;B\x07\x1b]633;C\x07"
@@ -26,7 +29,7 @@ def test_decoration_failure():
 
     ps1.hooks.failure_flag = True
     result = str(ps1)
-    if sys.platform != "win32":
+    if sys.platform != "win32" and (not is_wsl):
         assert (
             result
             == "\x1b]633;E;None\x07\x1b]633;D;1\x07\x1b]633;A\x07>>> \x1b]633;B\x07\x1b]633;C\x07"