public CollectExecutionManager CreateExecutionManagerForTheSuccessScenario(ProbeResult resultForRegistry, ProbeResult resultForFamily) { MockRepository mocks = new MockRepository(); probeManager = mocks.DynamicMock <IProbeManager>(); probe = mocks.DynamicMock <IProbe>(); familyProbe = mocks.DynamicMock <IProbe>(); dataProvider = mocks.DynamicMock <IDataProvider>(); systemInformationService = mocks.DynamicMock <ISystemInformationService>(); connectionContext = this.CreateConnectionContext(); CollectInfo collectInfo = new CollectInfo() { ObjectTypes = ovalObjects }; Expect.Call(probeManager.GetProbesFor(null, FamilyEnumeration.windows)).IgnoreArguments().Repeat.Any().Return(this.GetSelectedProbes()); Expect.Call(probeManager.GetSystemInformationService(FamilyEnumeration.windows)).IgnoreArguments().Repeat.Any().Return(systemInformationService); Expect.Call(probe.Execute(connectionContext, target, collectInfo)).IgnoreArguments().Repeat.Any().Return(resultForRegistry); Expect.Call(familyProbe.Execute(connectionContext, target, collectInfo)).IgnoreArguments().Repeat.Any().Return(resultForFamily); Expect.Call(dataProvider.GetSession()).Repeat.Any().Return(session).IgnoreArguments(); Expect.Call(dataProvider.GetTransaction(session)).Repeat.Any().Return(new Transaction(session)).IgnoreArguments(); Expect.Call(systemInformationService.GetSystemInformationFrom(target)).IgnoreArguments().Return(this.GetExpectedSystemInformation()); mocks.ReplayAll(); return(new CollectExecutionManager(probeManager) { Target = target, connectionContext = connectionContext }); }
private FileProber GetMockedFileProber(file_item fakeItem) { IList <String> fakeValues = new List <String>(new string[] { "FakeValue" }); CollectedItem fakeCollectedItem = ProbeHelper.CreateFakeCollectedItem(fakeItem); MockRepository mocks = new MockRepository(); IConnectionManager fakeConnection = mocks.DynamicMock <IConnectionManager>(); ISystemInformationService fakeSystemInformation = mocks.DynamicMock <ISystemInformationService>(); FileConnectionProvider fakeProvider = mocks.DynamicMock <FileConnectionProvider>(); WmiDataProvider fakeWmiProvider = mocks.DynamicMock <WmiDataProvider>(); FileObjectCollector fakeDataCollector = mocks.DynamicMock <FileObjectCollector>(); fakeDataCollector.WmiDataProvider = fakeWmiProvider; //Expect.Call(fakeConnection.Connect<FileConnectionProvider>(null, null)).IgnoreArguments().Repeat.Any().Return(fakeProvider); Expect.Call(fakeDataCollector.CollectDataForSystemItem(fakeItem)).IgnoreArguments().Repeat.Any().Return(new List <CollectedItem>() { fakeCollectedItem }); Expect.Call(fakeDataCollector.GetValues(null)).IgnoreArguments().Repeat.Any().Return(fakeValues); Expect.Call(fakeSystemInformation.GetSystemInformationFrom(null)).IgnoreArguments().Return(SystemInformationFactory.GetExpectedSystemInformation()); mocks.ReplayAll(); return(new FileProber() { ConnectionManager = fakeConnection, ObjectCollector = fakeDataCollector }); }
private void GetSystemInformationFromTarget(CollectRequest collectRequest, ISystemInformationService systemInformationService) { SystemInformation systemInformation = systemInformationService.GetSystemInformationFrom(Target); SystemInfo systemInfo = new SystemInfoFactory().CreateSystemInfo(systemInformation); collectRequest.Target.SystemInformation = systemInfo; }
public CollectExecutionManager CreateExecutionManagerWithCannotConnectToHostExceptionScenario( ProbeResult resultForRegistry, ProbeResult resultForFamily, bool exceptionOnSystemInformationGathering = false) { MockRepository mocks = new MockRepository(); probeManager = mocks.DynamicMock <IProbeManager>(); probe = mocks.DynamicMock <IProbe>(); familyProbe = mocks.DynamicMock <IProbe>(); dataProvider = mocks.DynamicMock <IDataProvider>(); var fakeCannotConnectException = new CannotConnectToHostException(target, "Cannot Connection To Host..."); CollectInfo collectInfo = new CollectInfo() { ObjectTypes = ovalObjects }; Expect.Call(probeManager.GetProbesFor(null, FamilyEnumeration.windows)).Repeat.Any().IgnoreArguments().Return(GetSelectedProbes()); Expect.Call(probe.Execute(connectionContext, target, collectInfo)).IgnoreArguments().Throw(fakeCannotConnectException); Expect.Call(familyProbe.Execute(connectionContext, target, collectInfo)).IgnoreArguments().Return(resultForFamily); Expect.Call(dataProvider.GetSession()).Repeat.Any().Return(session); Expect.Call(dataProvider.GetTransaction(session)).Repeat.Any().Return(new Transaction(session)); if (exceptionOnSystemInformationGathering) { systemInformationService = mocks.DynamicMock <ISystemInformationService>(); Expect.Call(probeManager.GetSystemInformationService(FamilyEnumeration.windows)).IgnoreArguments().Return(systemInformationService); Expect.Call(systemInformationService.GetSystemInformationFrom(null)).IgnoreArguments().Throw(fakeCannotConnectException); } mocks.ReplayAll(); return(new CollectExecutionManager(probeManager) { Target = target, connectionContext = connectionContext }); }