141 lines
4.2 KiB
C#
141 lines
4.2 KiB
C#
/*
|
|
* 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);
|
|
}
|
|
}
|
|
}
|