Exemple #1
0
        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;
        }
Exemple #2
0
        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);
        }