Add more testing for yarpc options

uber-go · Feb 23, 2017 · c5af058 · c5af058
1 parent ff59b8e
commit c5af058
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/modules/rpc/yarpc_options_test.go b/modules/rpc/yarpc_options_test.go
@@ -36,8 +36,30 @@ func TestWithInboundMiddleware_OK(t *testing.T) {
 	assert.Equal(t, 1, len(inboundMiddlewareFromModuleInfo(mc)))
 }
 
+func TestWithInboundMiddleware_Multiple(t *testing.T) {
+	mc, err := service.NewModuleInfo(
+		service.NopHost(),
+		"hello",
+		WithInboundMiddleware(middleware.NopUnaryInbound),
+		WithInboundMiddleware(middleware.NopUnaryInbound),
+	)
+	require.NoError(t, err)
+	assert.Equal(t, 2, len(inboundMiddlewareFromModuleInfo(mc)))
+}
+
 func TestWithOnewayInboundMiddleware_OK(t *testing.T) {
 	mc, err := service.NewModuleInfo(service.NopHost(), "hello", WithOnewayInboundMiddleware(middleware.NopOnewayInbound))
 	require.NoError(t, err)
 	assert.Equal(t, 1, len(onewayInboundMiddlewareFromModuleInfo(mc)))
 }
+
+func TestWithOnewayInboundMiddleware_Multiple(t *testing.T) {
+	mc, err := service.NewModuleInfo(
+		service.NopHost(),
+		"hello",
+		WithOnewayInboundMiddleware(middleware.NopOnewayInbound),
+		WithOnewayInboundMiddleware(middleware.NopOnewayInbound),
+	)
+	require.NoError(t, err)
+	assert.Equal(t, 2, len(onewayInboundMiddlewareFromModuleInfo(mc)))
+}