Restore SEBPatch

SafeExamBrowser.Communication.UnitTests/Hosts/BaseHostStub.cs

@@ -0,0 +1,55 @@
+/*
+ * Copyright (c) 2024 ETH Zürich, IT Services
+ * 
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+using System;
+using System.Linq;
+using SafeExamBrowser.Communication.Hosts;
+using SafeExamBrowser.Communication.Contracts;
+using SafeExamBrowser.Communication.Contracts.Data;
+using SafeExamBrowser.Communication.Contracts.Hosts;
+using SafeExamBrowser.Logging.Contracts;
+
+namespace SafeExamBrowser.Communication.UnitTests.Hosts
+{
+	internal class BaseHostStub : BaseHost
+	{
+		public Func<Guid?, bool> OnConnectStub { get; set; }
+		public Action<Interlocutor> OnDisconnectStub { get; set; }
+		public Func<Message, Response> OnReceiveStub { get; set; }
+		public Func<SimpleMessagePurport, Response> OnReceiveSimpleMessageStub { get; set; }
+
+		public BaseHostStub(string address, IHostObjectFactory factory, ILogger logger, int timeout_ms) : base(address, factory, logger, timeout_ms)
+		{
+		}
+
+		public Guid? GetCommunicationToken()
+		{
+			return CommunicationToken.Any() ? CommunicationToken.First() : default(Guid?);
+		}
+
+		protected override bool OnConnect(Guid? token)
+		{
+			return OnConnectStub?.Invoke(token) == true;
+		}
+
+		protected override void OnDisconnect(Interlocutor interlocutor)
+		{
+			OnDisconnectStub?.Invoke(interlocutor);
+		}
+
+		protected override Response OnReceive(Message message)
+		{
+			return OnReceiveStub?.Invoke(message);
+		}
+
+		protected override Response OnReceive(SimpleMessagePurport message)
+		{
+			return OnReceiveSimpleMessageStub?.Invoke(message);
+		}
+	}
+}