1 files changed, 95 insertions, 0 deletions
diff --git a/tools/build_defs/python/tests/py_info_subject.bzl b/tools/build_defs/python/tests/py_info_subject.bzl
new file mode 100644
index 0000000..20185e5
--- /dev/null
+++ b/tools/build_defs/python/tests/py_info_subject.bzl
@@ -0,0 +1,95 @@
+# Copyright 2023 The Bazel Authors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""PyInfo testing subject."""
+
+load("@rules_testing//lib:truth.bzl", "subjects")
+
+def py_info_subject(info, *, meta):
+    """Creates a new `PyInfoSubject` for a PyInfo provider instance.
+
+    Method: PyInfoSubject.new
+
+    Args:
+        info: The PyInfo object
+        meta: ExpectMeta object.
+
+    Returns:
+        A `PyInfoSubject` struct
+    """
+
+    # buildifier: disable=uninitialized
+    public = struct(
+        # go/keep-sorted start
+        has_py2_only_sources = lambda *a, **k: _py_info_subject_has_py2_only_sources(self, *a, **k),
+        has_py3_only_sources = lambda *a, **k: _py_info_subject_has_py3_only_sources(self, *a, **k),
+        imports = lambda *a, **k: _py_info_subject_imports(self, *a, **k),
+        transitive_sources = lambda *a, **k: _py_info_subject_transitive_sources(self, *a, **k),
+        uses_shared_libraries = lambda *a, **k: _py_info_subject_uses_shared_libraries(self, *a, **k),
+        # go/keep-sorted end
+    )
+    self = struct(
+        actual = info,
+        meta = meta,
+    )
+    return public
+
+def _py_info_subject_has_py2_only_sources(self):
+    """Returns a `BoolSubject` for the `has_py2_only_sources` attribute.
+
+    Method: PyInfoSubject.has_py2_only_sources
+    """
+    return subjects.bool(
+        self.actual.has_py2_only_sources,
+        meta = self.meta.derive("has_py2_only_sources()"),
+    )
+
+def _py_info_subject_has_py3_only_sources(self):
+    """Returns a `BoolSubject` for the `has_py3_only_sources` attribute.
+
+    Method: PyInfoSubject.has_py3_only_sources
+    """
+    return subjects.bool(
+        self.actual.has_py3_only_sources,
+        meta = self.meta.derive("has_py3_only_sources()"),
+    )
+
+def _py_info_subject_imports(self):
+    """Returns a `CollectionSubject` for the `imports` attribute.
+
+    Method: PyInfoSubject.imports
+    """
+    return subjects.collection(
+        self.actual.imports,
+        meta = self.meta.derive("imports()"),
+    )
+
+def _py_info_subject_transitive_sources(self):
+    """Returns a `DepsetFileSubject` for the `transitive_sources` attribute.
+
+    Method: PyInfoSubject.transitive_sources
+    """
+    return subjects.depset_file(
+        self.actual.transitive_sources,
+        meta = self.meta.derive("transitive_sources()"),
+    )
+
+def _py_info_subject_uses_shared_libraries(self):
+    """Returns a `BoolSubject` for the `uses_shared_libraries` attribute.
+
+    Method: PyInfoSubject.uses_shared_libraries
+    """
+    return subjects.bool(
+        self.actual.uses_shared_libraries,
+        meta = self.meta.derive("uses_shared_libraries()"),
+    )