plexus-python-protobuf-setup 1.0.2__tar.gz → 1.0.3__tar.gz

{plexus_python_protobuf_setup-1.0.2 → plexus_python_protobuf_setup-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plexus-python-protobuf-setup
-Version: 1.0.2
+Version: 1.0.3
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12

plexus_python_protobuf_setup-1.0.3/resources/unittest/setup/protos/dummy/person.proto

@@ -0,0 +1,9 @@
+syntax = "proto3";
+
+package dummy;
+
+message Person {
+  optional string name = 1;
+  optional int32 age = 2;
+  repeated string hobbies = 3;
+}

plexus_python_protobuf_setup-1.0.3/resources/unittest/setup/protos/dummy/relationship/company.proto

@@ -0,0 +1,12 @@
+syntax = "proto3";
+
+package dummy.relationship;
+
+import "dummy/person.proto";
+
+message Company {
+
+  optional dummy.Person boss = 1;
+  repeated dummy.Person staff = 2;
+
+}

plexus_python_protobuf_setup-1.0.3/resources/unittest/setup/protos/dummy/relationship/family.proto

@@ -0,0 +1,13 @@
+syntax = "proto3";
+
+package dummy.relationship;
+
+import "dummy/person.proto";
+
+message Family {
+
+  optional dummy.Person father = 1;
+  optional dummy.Person mother = 2;
+  repeated dummy.Person children = 3;
+
+}

{plexus_python_protobuf_setup-1.0.2 → plexus_python_protobuf_setup-1.0.3}/src/plexus/protobuf/setup/__init__.py

@@ -24,7 +24,7 @@ def compile_protos(
     :param include_dirs: Directories containing .proto files which are used only in compiling
     :param descriptor_path: Optional path to output the descriptor set file.
     :param with_grpc: If True, also generates gRPC Python code.
-    :raises FileNotFoundError: If the proto directory does not exist.
+    :raises FileNotFoundError: If any of the proto directories and include directories does not exist.
     """
     import glob
     import os

{plexus_python_protobuf_setup-1.0.2 → plexus_python_protobuf_setup-1.0.3}/src/plexus_python_protobuf_setup.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plexus-python-protobuf-setup
-Version: 1.0.2
+Version: 1.0.3
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12

{plexus_python_protobuf_setup-1.0.2 → plexus_python_protobuf_setup-1.0.3}/src/plexus_python_protobuf_setup.egg-info/SOURCES.txt

@@ -7,6 +7,9 @@ pyproject.toml
 setup.py
 .github/workflows/pr.yml
 .github/workflows/push.yml
+resources/unittest/setup/protos/dummy/person.proto
+resources/unittest/setup/protos/dummy/relationship/company.proto
+resources/unittest/setup/protos/dummy/relationship/family.proto
 src/plexus/protobuf/setup/__init__.py
 src/plexus_python_protobuf_setup.egg-info/PKG-INFO
 src/plexus_python_protobuf_setup.egg-info/SOURCES.txt
@@ -15,4 +18,5 @@ src/plexus_python_protobuf_setup.egg-info/not-zip-safe
 src/plexus_python_protobuf_setup.egg-info/requires.txt
 src/plexus_python_protobuf_setup.egg-info/top_level.txt
 test/plexus_test.py
-test/plexus_tests/__init__.py
+test/plexus_tests/__init__.py
+test/plexus_tests/protobuf/setup_test.py

plexus_python_protobuf_setup-1.0.3/test/plexus_tests/protobuf/setup_test.py

@@ -0,0 +1,49 @@
+import os
+import tempfile
+import unittest
+
+import plexus.protobuf.setup
+from plexus_tests import resources_directory
+
+
+class TestSetup(unittest.TestCase):
+
+    def test_compile_protos(self):
+        with tempfile.TemporaryDirectory() as temp_dir:
+            proto_dirs = [os.path.join(resources_directory, "unittest", "setup", "protos")]
+            include_dirs = [os.path.join(resources_directory, "unittest", "setup", "protos")]
+
+            plexus.protobuf.setup.compile_protos(temp_dir,
+                                                 proto_dirs,
+                                                 include_dirs,
+                                                 descriptor_path=os.path.join(temp_dir, "descriptor.proto"),
+                                                 with_grpc=True)
+
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "descriptor.proto")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "person_pb2.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "person_pb2.pyi")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "person_pb2_grpc.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "company_pb2.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "company_pb2.pyi")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "company_pb2_grpc.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "family_pb2.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "family_pb2.pyi")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "family_pb2_grpc.py")))
+
+    def test_compile_protos__without_grpc(self):
+        with tempfile.TemporaryDirectory() as temp_dir:
+            proto_dirs = [os.path.join(resources_directory, "unittest", "setup", "protos")]
+            include_dirs = [os.path.join(resources_directory, "unittest", "setup", "protos")]
+
+            plexus.protobuf.setup.compile_protos(temp_dir, proto_dirs, include_dirs, with_grpc=False)
+
+            self.assertFalse(os.path.exists(os.path.join(temp_dir, "descriptor.proto")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "person_pb2.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "person_pb2.pyi")))
+            self.assertFalse(os.path.exists(os.path.join(temp_dir, "dummy", "person_pb2_grpc.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "company_pb2.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "company_pb2.pyi")))
+            self.assertFalse(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "company_pb2_grpc.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "family_pb2.py")))
+            self.assertTrue(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "family_pb2.pyi")))
+            self.assertFalse(os.path.exists(os.path.join(temp_dir, "dummy", "relationship", "family_pb2_grpc.py")))