Fix serialization roundtrip testing

OrleansContrib · Jul 24, 2017 · 929bacd · 929bacd
1 parent d554e9f
commit 929bacd
Show file tree

Hide file tree

Showing 7 changed files with 64 additions and 22 deletions.
diff --git a/Source/Demo.App.Tests/ActorFixture.cs b/Source/Demo.App.Tests/ActorFixture.cs
@@ -9,9 +9,6 @@
 
 using Orleankka.TestKit;
 
-using Orleans.Runtime.Configuration;
-using Orleans.Serialization;
-
 namespace Demo
 {
     public abstract class ActorFixture
@@ -21,8 +18,7 @@ public abstract class ActorFixture
         [SetUp]
         public virtual void SetUp()
         {
-            var config = new ClusterConfiguration();
-            Runtime = new ActorRuntimeMock(new SerializationManager(null, config.Globals, config.Defaults));
+            Runtime = new ActorRuntimeMock(new SerializationOptions(roundtrip: true));
         }
 
         protected ActorSystemMock System => Runtime.System;

diff --git a/Source/Orleankka.TestKit/ActorRefMock.cs b/Source/Orleankka.TestKit/ActorRefMock.cs
@@ -4,21 +4,20 @@
 using System.Linq.Expressions;
 using System.Threading.Tasks;
 
-using Orleans.Serialization;
 
 namespace Orleankka.TestKit
 {
     [Serializable]
     public class ActorRefMock : ActorRef
     {
-        [NonSerialized] readonly SerializationManager serialization;
+        [NonSerialized] readonly SerializationOptions serialization;
         [NonSerialized] readonly List<IExpectation> expectations = new List<IExpectation>();
         [NonSerialized] readonly List<RecordedMessage> messages = new List<RecordedMessage>();
 
-        internal ActorRefMock(ActorPath path, SerializationManager serialization = null)
+        internal ActorRefMock(ActorPath path, SerializationOptions serialization = null)
             : base(path)
         {
-            this.serialization = serialization;
+            this.serialization = serialization ?? SerializationOptions.Default;
         }
 
         public TellExpectation<TMessage> ExpectTell<TMessage>(Expression<Func<TMessage, bool>> match = null)

diff --git a/Source/Orleankka.TestKit/ActorRuntimeMock.cs b/Source/Orleankka.TestKit/ActorRuntimeMock.cs
@@ -1,12 +1,10 @@
 using Orleankka.Services;
 
-using Orleans.Serialization;
-
 namespace Orleankka.TestKit
 {
     public class ActorRuntimeMock : IActorRuntime
     {
-        public ActorRuntimeMock(SerializationManager serialization = null)
+        public ActorRuntimeMock(SerializationOptions serialization = null)
         {
             System      = new ActorSystemMock(serialization);
             Timers      = new TimerServiceMock();

diff --git a/Source/Orleankka.TestKit/ActorSystemMock.cs b/Source/Orleankka.TestKit/ActorSystemMock.cs
@@ -2,15 +2,13 @@
 using System.Collections.Generic;
 using System.Threading.Tasks;
 
-using Orleans.Serialization;
-
 namespace Orleankka.TestKit
 {
     using Client;
 
     public class ActorSystemMock : IClientActorSystem
     {
-        readonly SerializationManager serialization;
+        readonly SerializationOptions serialization;
 
         readonly Dictionary<ActorPath, ActorRefMock> actors =
              new Dictionary<ActorPath, ActorRefMock>();
@@ -21,9 +19,9 @@ public class ActorSystemMock : IClientActorSystem
         readonly Queue<ClientObservableMock> observables = 
              new Queue<ClientObservableMock>();
 
-        public ActorSystemMock(SerializationManager serialization = null)
+        public ActorSystemMock(SerializationOptions serialization = null)
         {
-            this.serialization = serialization;
+            this.serialization = serialization ?? SerializationOptions.Default;
         }
 
         public ActorRefMock MockActorOf<TActor>(string id)

diff --git a/Source/Orleankka.TestKit/Orleankka.TestKit.csproj b/Source/Orleankka.TestKit/Orleankka.TestKit.csproj
@@ -53,7 +53,7 @@
       <Link>Properties\Product.cs</Link>
     </Compile>
     <Compile Include="ActorBehaviorExtensions.cs" />
-    <Compile Include="SerializationExtensions.cs" />
+    <Compile Include="SerializationOptions.cs" />
     <Compile Include="StorageServiceMock.cs" />
     <Compile Include="ActivationServiceMock.cs" />
     <Compile Include="StreamSubscriptionMock.cs" />

diff --git a/Source/Orleankka.TestKit/SerializationOptions.cs b/Source/Orleankka.TestKit/SerializationOptions.cs
@@ -0,0 +1,52 @@
+using System;
+using System.Linq;
+using System.Net;
+using System.Reflection;
+
+using Microsoft.Extensions.DependencyInjection;
+
+using Orleans;
+using Orleans.Runtime.Configuration;
+using Orleans.Serialization;
+using Orleans.Runtime;
+
+namespace Orleankka.TestKit
+{
+    using Utility;
+
+    public class SerializationOptions
+    {
+        internal static readonly SerializationOptions Default = new SerializationOptions(false);
+
+        readonly SerializationManager manager;
+
+        public SerializationOptions(bool roundtrip, params Type[] serializers)
+        {
+            Requires.NotNull(serializers, nameof(serializers));
+
+            if (!roundtrip)
+                return;
+
+            var configuration = new ClientConfiguration
+            {
+                GatewayProvider = ClientConfiguration.GatewayProviderType.Config,
+                Gateways = { new IPEndPoint(0, 0) },
+                TraceToConsole = false,
+                DefaultTraceLevel = Severity.Off
+            };
+
+            if (serializers.Length > 0)
+                configuration.SerializationProviders.AddRange(serializers.Select(x => x.GetTypeInfo()));
+
+            var client = new ClientBuilder()
+                .UseConfiguration(configuration)
+                .Build();
+
+            manager = client.ServiceProvider.GetRequiredService<SerializationManager>();
+        }
+
+        internal object Roundtrip(object obj) => manager != null
+            ? manager.RoundTripSerializationForTesting(obj)
+            : obj;
+    }
+}
diff --git a/Source/Orleankka.TestKit/StreamRefMock.cs b/Source/Orleankka.TestKit/StreamRefMock.cs
@@ -4,14 +4,13 @@
 using System.Linq.Expressions;
 using System.Threading.Tasks;
 
-using Orleans.Serialization;
 
 namespace Orleankka.TestKit
 {
     [Serializable]
     public class StreamRefMock : StreamRef
     {
-        [NonSerialized] readonly SerializationManager serialization;
+        [NonSerialized] readonly SerializationOptions serialization;
         [NonSerialized] readonly List<IExpectation> expectations = new List<IExpectation>();
 
         [NonSerialized] readonly List<RecordedItem> items = new List<RecordedItem>();
@@ -24,10 +23,10 @@ public class StreamRefMock : StreamRef
         public Actor Subscribed    { get; private set; }
         public Actor Resumed       { get; private set; }
 
-        internal StreamRefMock(StreamPath path, SerializationManager serialization = null)
+        internal StreamRefMock(StreamPath path, SerializationOptions serialization = null)
             : base(path, null)
         {
-            this.serialization = serialization;
+            this.serialization = serialization ?? SerializationOptions.Default;
         }
 
         public PushExpectation<TItem> Expect<TItem>(Expression<Func<TItem, bool>> match = null)