Restore SEBPatch

SafeExamBrowser.Runtime.UnitTests/Operations/VirtualMachineOperationTests.cs

@@ -0,0 +1,141 @@
+/*
+ * 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 Microsoft.VisualStudio.TestTools.UnitTesting;
+using Moq;
+using SafeExamBrowser.Configuration.Contracts;
+using SafeExamBrowser.Core.Contracts.OperationModel;
+using SafeExamBrowser.Logging.Contracts;
+using SafeExamBrowser.Monitoring.Contracts;
+using SafeExamBrowser.Runtime.Operations;
+using SafeExamBrowser.Settings;
+using SafeExamBrowser.Settings.Security;
+
+namespace SafeExamBrowser.Runtime.UnitTests.Operations
+{
+	[TestClass]
+	public class VirtualMachineOperationTests
+	{
+		private Mock<IVirtualMachineDetector> detector;
+		private Mock<ILogger> logger;
+		private SessionContext context;
+		private VirtualMachineOperation sut;
+
+		[TestInitialize]
+		public void Initialize()
+		{
+			detector = new Mock<IVirtualMachineDetector>();
+			logger = new Mock<ILogger>();
+			context = new SessionContext();
+
+			context.Next = new SessionConfiguration();
+			context.Next.Settings = new AppSettings();
+
+			sut = new VirtualMachineOperation(detector.Object, logger.Object, context);
+		}
+
+		[TestMethod]
+		public void Perform_MustAbortIfVirtualMachineNotAllowed()
+		{
+			context.Next.Settings.Security.VirtualMachinePolicy = VirtualMachinePolicy.Deny;
+			detector.Setup(d => d.IsVirtualMachine()).Returns(true);
+
+			var result = sut.Perform();
+
+			detector.Verify(d => d.IsVirtualMachine(), Times.Once);
+			Assert.AreEqual(OperationResult.Aborted, result);
+		}
+
+		[TestMethod]
+		public void Perform_MustSucceedIfVirtualMachineAllowed()
+		{
+			context.Next.Settings.Security.VirtualMachinePolicy = VirtualMachinePolicy.Allow;
+			detector.Setup(d => d.IsVirtualMachine()).Returns(true);
+
+			var result = sut.Perform();
+
+			detector.Verify(d => d.IsVirtualMachine(), Times.AtMostOnce);
+			Assert.AreEqual(OperationResult.Success, result);
+
+			context.Next.Settings.Security.VirtualMachinePolicy = VirtualMachinePolicy.Allow;
+			detector.Setup(d => d.IsVirtualMachine()).Returns(false);
+
+			result = sut.Perform();
+
+			detector.Verify(d => d.IsVirtualMachine(), Times.AtMost(2));
+			Assert.AreEqual(OperationResult.Success, result);
+		}
+
+		[TestMethod]
+		public void Perform_MustSucceedIfNotAVirtualMachine()
+		{
+			context.Next.Settings.Security.VirtualMachinePolicy = VirtualMachinePolicy.Deny;
+			detector.Setup(d => d.IsVirtualMachine()).Returns(false);
+
+			var result = sut.Perform();
+
+			detector.Verify(d => d.IsVirtualMachine(), Times.Once);
+			Assert.AreEqual(OperationResult.Success, result);
+		}
+
+		[TestMethod]
+		public void Repeat_MustAbortIfVirtualMachineNotAllowed()
+		{
+			context.Next.Settings.Security.VirtualMachinePolicy = VirtualMachinePolicy.Deny;
+			detector.Setup(d => d.IsVirtualMachine()).Returns(true);
+
+			var result = sut.Repeat();
+
+			detector.Verify(d => d.IsVirtualMachine(), Times.Once);
+			Assert.AreEqual(OperationResult.Aborted, result);
+		}
+
+		[TestMethod]
+		public void Repeat_MustSucceedIfVirtualMachineAllowed()
+		{
+			context.Next.Settings.Security.VirtualMachinePolicy = VirtualMachinePolicy.Allow;
+			detector.Setup(d => d.IsVirtualMachine()).Returns(true);
+
+			var result = sut.Repeat();
+
+			detector.Verify(d => d.IsVirtualMachine(), Times.AtMostOnce);
+			Assert.AreEqual(OperationResult.Success, result);
+
+			context.Next.Settings.Security.VirtualMachinePolicy = VirtualMachinePolicy.Allow;
+			detector.Setup(d => d.IsVirtualMachine()).Returns(false);
+
+			result = sut.Repeat();
+
+			detector.Verify(d => d.IsVirtualMachine(), Times.AtMost(2));
+			Assert.AreEqual(OperationResult.Success, result);
+		}
+
+		[TestMethod]
+		public void Repeat_MustSucceedIfNotAVirtualMachine()
+		{
+			context.Next.Settings.Security.VirtualMachinePolicy = VirtualMachinePolicy.Deny;
+			detector.Setup(d => d.IsVirtualMachine()).Returns(false);
+
+			var result = sut.Repeat();
+
+			detector.Verify(d => d.IsVirtualMachine(), Times.Once);
+			Assert.AreEqual(OperationResult.Success, result);
+		}
+
+		[TestMethod]
+		public void Revert_MustDoNothing()
+		{
+			var result = sut.Revert();
+
+			detector.VerifyNoOtherCalls();
+			logger.VerifyNoOtherCalls();
+
+			Assert.AreEqual(OperationResult.Success, result);
+		}
+	}
+}