Exemple #1
0
        public void PingWithExtensionTest()
        {
            using (RpcServer server = RpcServer.CreateRpc(iid, new SearchService.ServerStub(new AuthenticatedSearch()))
                                      .AddProtocol("ncacn_np", @"\pipe\p1"))
            {
                UnitTestRpcInterop.RegisterAllExtensions(server.ExtensionRegistry);
                server.Ping += delegate(RpcPingRequest r)
                {
                    if (r.HasExtension(UnitTestRpcInterop.CustomPingDataIn))
                    {
                        return(RpcPingResponse.CreateBuilder()
                               .SetExtension(UnitTestRpcInterop.CustomPingDataOut,
                                             r.GetExtension(UnitTestRpcInterop.CustomPingDataIn))
                               .Build());
                    }
                    return(RpcPingResponse.DefaultInstance);
                };
                server.StartListening();

                using (
                    RpcClient client =
                        RpcClient.ConnectRpc(iid, "ncacn_np", null, @"\pipe\p1").Authenticate(
                            RpcAuthenticationType.Anonymous))
                {
                    UnitTestRpcInterop.RegisterAllExtensions(client.ExtensionRegistry);

                    RpcPingRequest r = RpcPingRequest.CreateBuilder()
                                       .SetExtension(UnitTestRpcInterop.CustomPingDataIn, "ping-request-data")
                                       .Build();
                    RpcPingResponse response = client.Ping(r);
                    Assert.IsTrue(response.HasExtension(UnitTestRpcInterop.CustomPingDataOut));
                    Assert.AreEqual("ping-request-data", response.GetExtension(UnitTestRpcInterop.CustomPingDataOut));
                }
            }
        }
Exemple #2
0
        public void RoundTripCallContextExtensionsTest()
        {
            using (RpcServer server = RpcServer.CreateRpc(iid, new SearchService.ServerStub(new AuthenticatedSearch()))
                                      .AddProtocol("ncacn_np", @"\pipe\p1"))
            {
                UnitTestRpcInterop.RegisterAllExtensions(server.ExtensionRegistry);
                server.Ping += delegate
                {
                    String strValue =
                        RpcCallContext.Current.GetExtension(UnitTestRpcInterop.CustomContextString);
                    UInt64 intValue =
                        RpcCallContext.Current.GetExtension(UnitTestRpcInterop.CustomContextNumber);

                    char[] tmp = strValue.ToCharArray();
                    Array.Reverse(tmp);
                    strValue = new string(tmp);
                    intValue = ~intValue;

                    RpcCallContext.Current = RpcCallContext.Current.ToBuilder()
                                             .SetExtension(UnitTestRpcInterop.CustomContextString, strValue)
                                             .SetExtension(UnitTestRpcInterop.CustomContextNumber, intValue)
                                             .Build();

                    return(RpcPingResponse.DefaultInstance);
                };
                server.StartListening();

                using (
                    RpcClient client =
                        RpcClient.ConnectRpc(iid, "ncacn_np", null, @"\pipe\p1").Authenticate(
                            RpcAuthenticationType.Anonymous))
                {
                    UnitTestRpcInterop.RegisterAllExtensions(client.ExtensionRegistry);

                    client.CallContext = client.CallContext.ToBuilder()
                                         .SetExtension(UnitTestRpcInterop.CustomContextString, "abc")
                                         .SetExtension(UnitTestRpcInterop.CustomContextNumber, 0x70000000FFFFFFFFUL)
                                         .Build();
                    client.Ping();

                    Assert.AreEqual("cba", client.CallContext.GetExtension(UnitTestRpcInterop.CustomContextString));
                    Assert.AreEqual(~0x70000000FFFFFFFFUL,
                                    client.CallContext.GetExtension(UnitTestRpcInterop.CustomContextNumber));
                }
            }
        }