refacator mocks to mocks

Author: rickeylev

tests/repo_utils/repo_utils_test.bzl

@@ -2,23 +2,20 @@
 
 load("@rules_testing//lib:test_suite.bzl", "test_suite")
 load("//python/private:repo_utils.bzl", "repo_utils")  # buildifier: disable=bzl-visibility
+load("//tests/support:mocks.bzl", "mocks")
 
 _tests = []
 
 def _test_get_platforms_os_name(env):
-    mock_mrctx = struct(
-        os = struct(
-            name = "Mac OS X",
-        ),
-    )
+    mock_mrctx = mocks.rctx(os_name = "Mac OS X")
     got = repo_utils.get_platforms_os_name(mock_mrctx)
     env.expect.that_str(got).equals("osx")
 
 _tests.append(_test_get_platforms_os_name)
 
 def _test_relative_to(env):
-    mock_mrctx_linux = struct(os = struct(name = "linux"))
-    mock_mrctx_win = struct(os = struct(name = "windows"))
+    mock_mrctx_linux = mocks.rctx(os_name = "linux")
+    mock_mrctx_win = mocks.rctx(os_name = "windows")
 
     # Case-sensitive matching (Linux)
     got = repo_utils.relative_to(mock_mrctx_linux, "foo/bar/baz", "foo/bar")

tests/support/mocks.bzl

@@ -0,0 +1,31 @@
+"""Mocks for testing."""
+
+def _rctx(os_name = "linux", os_arch = "x86_64", environ = None, **kwargs):
+    """Creates a mock of repository_ctx or module_ctx.
+
+    Args:
+        os_name: The OS name to mock (e.g., "linux", "Mac OS X", "windows").
+        os_arch: The OS architecture to mock (e.g., "x86_64", "aarch64").
+        environ: A dictionary representing the environment variables.
+        **kwargs: Additional attributes to add to the mock struct.
+
+    Returns:
+        A struct mocking repository_ctx.
+    """
+    if environ == None:
+        environ = {}
+
+    attrs = {
+        "os": struct(
+            name = os_name,
+            arch = os_arch,
+        ),
+        "getenv": environ.get,
+    }
+    attrs.update(kwargs)
+
+    return struct(**attrs)
+
+mocks = struct(
+    rctx = _rctx,
+)