Add test.support.skip_if_buildbot().

python · ericsnowcurrently · Oct 28, 2021 · Oct 26, 2021 · Oct 26, 2021 · Oct 26, 2021
commit 4c1810312372515b5a51c87042c4aabd29559b99
diff --git a/Lib/test/support/__init__.py b/Lib/test/support/__init__.py
@@ -372,6 +372,17 @@ def wrapper(*args, **kw):
     return decorator
 
 
+def skip_if_buildbot(reason=None):
+    """Decorator raising SkipTest if running on a buildbot."""
+    if not reason:
+        reason = 'not suitable for buildbots'
+    if sys.platform == 'win32':
+        isbuildbot = os.environ.get('USERNAME') == 'Buildbot'
+    else:
+        isbuildbot = os.environ.get('USER') == 'buildbot'
+    return unittest.skipIf(isbuildbot, reason)
+
+
 def system_must_validate_cert(f):
     """Skip the test on TLS certificate validation failures."""
     @functools.wraps(f)

@@ -1,19 +1,19 @@
 """Sanity-check tests for the "freeze" tool."""
 
-import os
 import sys
 import textwrap
 import unittest
 
+from test import support
+
 from . import imports_under_tool, skip_if_missing
 skip_if_missing('freeze')
 with imports_under_tool('freeze', 'test'):
     import freeze as helper
 
 
 @unittest.skipIf(sys.platform.startswith('win'), 'not supported on Windows')
-@unittest.skipIf(os.environ.get('USER') == 'buildbot',
-                 'not all buildbots have enough space')
+@support.skip_if_buildbot('not all buildbots have enough space')
 class TestFreeze(unittest.TestCase):
 
     def test_freeze_simple_script(self):