parent
63a31d85f1
commit
a2a4629614
2 changed files with 106 additions and 0 deletions
@ -0,0 +1,105 @@ |
|||||||
|
#region Copyright notice and license |
||||||
|
|
||||||
|
// Copyright 2018 The gRPC Authors |
||||||
|
// |
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License"); |
||||||
|
// you may not use this file except in compliance with the License. |
||||||
|
// You may obtain a copy of the License at |
||||||
|
// |
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0 |
||||||
|
// |
||||||
|
// Unless required by applicable law or agreed to in writing, software |
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS, |
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||||
|
// See the License for the specific language governing permissions and |
||||||
|
// limitations under the License. |
||||||
|
|
||||||
|
#endregion |
||||||
|
|
||||||
|
using System; |
||||||
|
using System.Collections.Generic; |
||||||
|
using System.Diagnostics; |
||||||
|
using System.IO; |
||||||
|
using System.Linq; |
||||||
|
using System.Threading; |
||||||
|
using System.Threading.Tasks; |
||||||
|
|
||||||
|
using Grpc.Core; |
||||||
|
using Grpc.Core.Internal; |
||||||
|
using Grpc.Core.Utils; |
||||||
|
using NUnit.Framework; |
||||||
|
|
||||||
|
namespace Grpc.Core.Tests |
||||||
|
{ |
||||||
|
public class MarshallerTest |
||||||
|
{ |
||||||
|
[Test] |
||||||
|
public void ContextualSerializerEmulation() |
||||||
|
{ |
||||||
|
Func<string, byte[]> simpleSerializer = System.Text.Encoding.UTF8.GetBytes; |
||||||
|
Func<byte[], string> simpleDeserializer = System.Text.Encoding.UTF8.GetString; |
||||||
|
var marshaller = new Marshaller<string>(simpleSerializer, |
||||||
|
simpleDeserializer); |
||||||
|
|
||||||
|
Assert.AreSame(simpleSerializer, marshaller.Serializer); |
||||||
|
Assert.AreSame(simpleDeserializer, marshaller.Deserializer); |
||||||
|
|
||||||
|
// test that emulated contextual serializer and deserializer work |
||||||
|
string origMsg = "abc"; |
||||||
|
var serializationContext = new FakeSerializationContext(); |
||||||
|
marshaller.ContextualSerializer(origMsg, serializationContext); |
||||||
|
|
||||||
|
var deserializationContext = new FakeDeserializationContext(serializationContext.Payload); |
||||||
|
Assert.AreEqual(origMsg, marshaller.ContextualDeserializer(deserializationContext)); |
||||||
|
} |
||||||
|
|
||||||
|
[Test] |
||||||
|
public void SimpleSerializerEmulation() |
||||||
|
{ |
||||||
|
Action<string, SerializationContext> contextualSerializer = (str, context) => |
||||||
|
{ |
||||||
|
var bytes = System.Text.Encoding.UTF8.GetBytes(str); |
||||||
|
context.Complete(bytes); |
||||||
|
}; |
||||||
|
Func<DeserializationContext, string> contextualDeserializer = (context) => |
||||||
|
{ |
||||||
|
return System.Text.Encoding.UTF8.GetString(context.PayloadAsNewBuffer()); |
||||||
|
}; |
||||||
|
var marshaller = new Marshaller<string>(contextualSerializer, contextualDeserializer); |
||||||
|
|
||||||
|
Assert.AreSame(contextualSerializer, marshaller.ContextualSerializer); |
||||||
|
Assert.AreSame(contextualDeserializer, marshaller.ContextualDeserializer); |
||||||
|
|
||||||
|
// test that emulated serializer and deserializer work |
||||||
|
var origMsg = "abc"; |
||||||
|
var serialized = marshaller.Serializer(origMsg); |
||||||
|
Assert.AreEqual(origMsg, marshaller.Deserializer(serialized)); |
||||||
|
} |
||||||
|
|
||||||
|
class FakeSerializationContext : SerializationContext |
||||||
|
{ |
||||||
|
public byte[] Payload; |
||||||
|
public override void Complete(byte[] payload) |
||||||
|
{ |
||||||
|
this.Payload = payload; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
class FakeDeserializationContext : DeserializationContext |
||||||
|
{ |
||||||
|
public byte[] payload; |
||||||
|
|
||||||
|
public FakeDeserializationContext(byte[] payload) |
||||||
|
{ |
||||||
|
this.payload = payload; |
||||||
|
} |
||||||
|
|
||||||
|
public override int PayloadLength => payload.Length; |
||||||
|
|
||||||
|
public override byte[] PayloadAsNewBuffer() |
||||||
|
{ |
||||||
|
return payload; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue