Run stubtest on windows for colorama (#9168)

Avasam · web-flow · commit 24277cedf41e · 2022-11-12T16:14:15.000-08:00
diff --git a/stubs/colorama/@tests/stubtest_allowlist.txt b/stubs/colorama/@tests/stubtest_allowlist.txt
@@ -44,13 +44,6 @@ colorama.initialise.wrapped_stdout
 
 # These are re-exports that are implementation detail:
 colorama.ansitowin32.BEL
-colorama.ansitowin32.windll
-colorama.winterm.get_osfhandle
-
-# These are only available on Windows:
-colorama.winterm.WinColor
-colorama.winterm.WinStyle
-colorama.winterm.WinTerm
 
 # Not planning on writing stubs for tests:
 colorama.tests.*
diff --git a/stubs/colorama/@tests/stubtest_allowlist_linux.txt b/stubs/colorama/@tests/stubtest_allowlist_linux.txt
@@ -0,0 +1,8 @@
+# These are only available on Windows:
+colorama.winterm.WinColor
+colorama.winterm.WinStyle
+colorama.winterm.WinTerm
+
+# These are re-exports that are implementation detail:
+colorama.ansitowin32.windll
+colorama.winterm.get_osfhandle
diff --git a/stubs/colorama/METADATA.toml b/stubs/colorama/METADATA.toml
@@ -2,3 +2,4 @@ version = "0.4.*"
 
 [tool.stubtest]
 ignore_missing_stub = false
+platforms = ["win32", "linux"]
diff --git a/stubs/colorama/colorama/win32.pyi b/stubs/colorama/colorama/win32.pyi
@@ -25,6 +25,8 @@ if sys.platform == "win32":
     def FillConsoleOutputCharacter(stream_id: int, char: str, length: int, start: COORD) -> int: ...
     def FillConsoleOutputAttribute(stream_id: int, attr: int, length: int, start: COORD) -> wintypes.BOOL: ...
     def SetConsoleTitle(title: str) -> wintypes.BOOL: ...
+    def GetConsoleMode(handle: int) -> int: ...
+    def SetConsoleMode(handle: int, mode: int) -> None: ...
 
 else:
     windll: None

Original file line number	Diff line number	Diff line change
`@@ -2,3 +2,4 @@ version = "0.4.*"`
`2`	`2`
`3`	`3`	`[tool.stubtest]`
`4`	`4`	`ignore_missing_stub = false`
	`5`	`+platforms = ["win32", "linux"]`