Fix unittest (#5)

grpc · Nov 15, 2023 · f6c9039 · f6c9039
1 parent 4cab81e
commit f6c9039
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 11 deletions.
diff --git a/.github/workflows/psm-interop.yaml b/.github/workflows/psm-interop.yaml
@@ -40,9 +40,9 @@ jobs:
           filters:  |
             psm_interop_src:
               - './**'
-              - './proto/empty.proto'
-              - './proto/messages.proto'
-              - './proto/test.proto'
+              - './protos/empty.proto'
+              - './protos/messages.proto'
+              - './protos/test.proto'
 
       - uses: actions/setup-python@v4
         if: ${{ steps.paths_filter.outputs.psm_interop_src == 'true' }}
@@ -65,9 +65,9 @@ jobs:
         run: >
           python -m grpc_tools.protoc --proto_path=.
           --python_out=. --grpc_python_out=.
-          proto/empty.proto
-          proto/messages.proto
-          proto/test.proto
+          protos/empty.proto
+          protos/messages.proto
+          protos/test.proto
 
       - name: "Run unit tests"
         if: ${{ steps.paths_filter.outputs.psm_interop_src == 'true' }}

diff --git a/framework/rpc/grpc_testing.py b/framework/rpc/grpc_testing.py
@@ -23,9 +23,11 @@
 from grpc_health.v1 import health_pb2_grpc
 
 import framework.rpc
-from src.proto.grpc.testing import empty_pb2
-from src.proto.grpc.testing import messages_pb2
-from src.proto.grpc.testing import test_pb2_grpc
+import sys
+print(f">>>>path: {sys.path}")
+from protos import empty_pb2
+from protos import messages_pb2
+from protos import test_pb2_grpc
 
 # Type aliases
 _LoadBalancerStatsRequest = messages_pb2.LoadBalancerStatsRequest

diff --git a/proto/empty.proto → protos/empty.proto b/proto/empty.proto → protos/empty.proto
diff --git a/proto/messages.proto → protos/messages.proto b/proto/messages.proto → protos/messages.proto
diff --git a/proto/test.proto → protos/test.proto b/proto/test.proto → protos/test.proto
@@ -18,8 +18,8 @@
 
 syntax = "proto3";
 
-import "proto/empty.proto";
-import "proto/messages.proto";
+import "protos/empty.proto";
+import "protos/messages.proto";
 
 package grpc.testing;