Add net/grpc test

grafana · Feb 6, 2025 · 95e58e9 · 95e58e9
1 parent 7915a2f
commit 95e58e9
Showing 1 changed file with 34 additions and 1 deletion.
diff --git a/internal/js/modules/k6/grpc/client_test.go b/internal/js/modules/k6/grpc/client_test.go
@@ -1374,7 +1374,6 @@ func TestClientLoadProto(t *testing.T) {
 	ts := newTestState(t)
 
 	tt := testcase{
-		name: "LoadNestedTypesProto",
 		initString: codeBlock{
 			code: `
 			var client = new grpc.Client();
@@ -1402,6 +1401,40 @@ func TestClientLoadProto(t *testing.T) {
 	}
 }
 
+func TestClientLoadProtoAbsoluteRootWithFile(t *testing.T) {
+	t.Parallel()
+
+	ts := newTestState(t)
+	pwd := "file://" + ts.VU.InitEnvField.CWD.JoinPath("../..").String()
+
+	tt := testcase{
+		initString: codeBlock{
+			code: `
+			var client = new grpc.Client();
+			client.load(["` + pwd + `"], "../../lib/testutils/httpmultibin/nested_types/nested_types.proto");`,
+		},
+	}
+
+	val, err := ts.Run(tt.initString.code)
+	assertResponse(t, tt.initString, err, val, ts)
+
+	expectedTypes := []string{
+		"grpc.testdata.nested.types.Outer",
+		"grpc.testdata.nested.types.Outer.MiddleAA",
+		"grpc.testdata.nested.types.Outer.MiddleAA.Inner",
+		"grpc.testdata.nested.types.Outer.MiddleBB",
+		"grpc.testdata.nested.types.Outer.MiddleBB.Inner",
+		"grpc.testdata.nested.types.MeldOuter",
+	}
+
+	for _, expected := range expectedTypes {
+		found, err := protoregistry.GlobalTypes.FindMessageByName(protoreflect.FullName(expected))
+
+		assert.NotNil(t, found, "Expected to find the message type %s, but an error occurred", expected)
+		assert.Nil(t, err, "It was not expected that there would be an error, but it got: %v", err)
+	}
+}
+
 func TestClientConnectionReflectMetadata(t *testing.T) {
 	t.Parallel()