Added tests

cadence-workflow · Jan 15, 2025 · aee1adf · aee1adf
1 parent 541eca6
commit aee1adf
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/common/rpc/outbounds_test.go b/common/rpc/outbounds_test.go
@@ -180,6 +180,18 @@ func TestDirectOutbound(t *testing.T) {
 	assert.NotNil(t, outbounds["cadence-history"].Unary)
 }
 
+func TestSingleGRPCOutbound(t *testing.T) {
+	grpc := &grpc.Transport{}
+	tchannel := &tchannel.Transport{}
+
+	builder := NewSingleGRPCOutboundBuilder("grpc-only-out", "grpc-service-name", "http://example.com:1234")
+
+	outBound, err := builder.Build(grpc, tchannel)
+	assert.NoError(t, err)
+	assert.Equal(t, "grpc-service-name", outBound.Outbounds["grpc-only-out"].ServiceName)
+	assert.NotNil(t, outBound.Outbounds["grpc-only-out"].Unary)
+}
+
 func TestIsGRPCOutboud(t *testing.T) {
 	assert.True(t, IsGRPCOutbound(&transport.OutboundConfig{Outbounds: transport.Outbounds{Unary: (&grpc.Transport{}).NewSingleOutbound("localhost:1234")}}))
 	assert.False(t, IsGRPCOutbound(&transport.OutboundConfig{Outbounds: transport.Outbounds{Unary: (&tchannel.Transport{}).NewSingleOutbound("localhost:1234")}}))

diff --git a/common/rpc/params_test.go b/common/rpc/params_test.go
@@ -38,8 +38,9 @@ func TestNewParams(t *testing.T) {
 	dc := dynamicconfig.NewNopCollection()
 	makeConfig := func(svc config.Service) *config.Config {
 		return &config.Config{
-			PublicClient: config.PublicClient{HostPort: "localhost:9999"},
-			Services:     map[string]config.Service{"frontend": svc}}
+			PublicClient:           config.PublicClient{HostPort: "localhost:9999"},
+			ShardDistributorClient: config.ShardDistributorClient{HostPort: "localhost:9998"},
+			Services:               map[string]config.Service{"frontend": svc}}
 	}
 	logger := testlogger.New(t)
 	metricsCl := metrics.NewNoopMetricsClient()