public override void SharedExecute() { var fakeContext = this.Pull <FakeDBContext>(); var fakeMgr = this.Pull <FakeServicesMgr>(); // Setting fake IHelper var mock = new Mock <IHelper>(); // Stubs here mock.Setup(helper => helper.GetDBContext(It.IsAny <Int32>())).Returns(fakeContext.DBContext); mock.Setup(helper => helper.GetServicesManager()).Returns(fakeMgr.ServicesMgr); this.Helper = mock.Object; // Setting fake IAgentHelper var agentMock = new Mock <IAgentHelper>(); // Stubs here agentMock.Setup(ahelper => ahelper.GetDBContext(It.IsAny <Int32>())).Returns(fakeContext.DBContext); agentMock.Setup(helper => helper.GetServicesManager()).Returns(fakeMgr.ServicesMgr); AgentHelper = agentMock.Object; // Setting fake IEHHelper var ehMock = new Mock <IEHHelper>(); // Stubs here ehMock.Setup(ahelper => ahelper.GetDBContext(It.IsAny <Int32>())).Returns(fakeContext.DBContext); ehMock.Setup(helper => helper.GetServicesManager()).Returns(fakeMgr.ServicesMgr); EHHelper = ehMock.Object; }
public async static Task <int> GetFieldArtifactID(string fieldName, IEHHelper helper, IAPILog logger) { int fieldArtifactId = 0; using (IObjectManager objectManager = helper.GetServicesManager().CreateProxy <IObjectManager>(ExecutionIdentity.System)) { var queryRequest = new QueryRequest() { ObjectType = new ObjectTypeRef() { Name = "Field" }, Condition = $"'Name' == '{fieldName}' AND 'Object Type' == 'Document'" }; var queryResult = await objectManager.QuerySlimAsync(helper.GetActiveCaseID(), queryRequest, 0, 1); if (queryResult.TotalCount > 0) { fieldArtifactId = queryResult.Objects.Select(x => x.ArtifactID).FirstOrDefault(); logger.LogVerbose("Alert field artifactID: {fieldArtifactID}", fieldArtifactId); } } return(fieldArtifactId); }