@isTest
public class DataLoaderLeadExportConditionTest {
    public static testMethod void testDataLoaderLeadExportTrue() {
        /* Create a map for the event we’re going to build. */
        /* Insert relative info that will meet criteria of the policy */
        Map<String, String> eventData = new Map<String, String>();
        eventData.put('NumberOfRecords', '2001');
        eventData.put('ExecutionTime', '1001');
        eventData.put('EntityName', 'Lead');

        /* We’re not going to cause a real event in the org.
        Instead, we’re going to create a Transaction Security
        event object and “feed” it to the Policy Engine. */
        TxnSecurity.Event e = new TxnSecurity.Event(
            UserInfo.getOrganizationId(),			/* organizationId */
            UserInfo.getUserId(),					/* userId */
            'Lead',									/* entityName */
            'DataExport',							/* action */
            'Lead',									/* resourceType */
            '01pR00000009D2H',						/* entityId */
            Datetime.newInstance(2018, 8, 16),		/* timeStamp */
            eventData);								/* data - Map with info about this event. */
        DataLoaderLeadExportCondition condition = new DataLoaderLeadExportCondition();
        /* Assert that the condition is triggered by evaluating
        the event e. The Transaction Security PolicyCondition
        interface returns True if the policy is triggered. */
        System.assertEquals(true, condition.evaluate(e));
    }
    public static testMethod void testDataLoaderLeadExportFalse() {
        /* Create a map for the event we’re going to build. */
        /* Insert relative info that will meet criteria of the policy */
        Map<String, String> eventData = new Map<String, String>();
        eventData.put('NumberOfRecords', '2000');
        eventData.put('ExecutionTime', '1000');
        eventData.put('EntityName', 'Lead');

        /* We’re not going to cause a real event in the org.
        Instead, we’re going to create a Transaction Security
        event object and “feed” it to the Policy Engine. */
        TxnSecurity.Event e = new TxnSecurity.Event(
            UserInfo.getOrganizationId(),			/* organizationId */
            UserInfo.getUserId(),					/* userId */
            'Lead',								/* entityName */
            'DataExport',						/* action */
            'Lead',								/* resourceType */
            '01pR00000009D2H',					/* entityId */
            Datetime.newInstance(2018, 8, 16),		/* timeStamp */
            eventData);							/* data - Map with info about this event. */
        DataLoaderLeadExportCondition condition = new DataLoaderLeadExportCondition();
        /* Assert that the condition is triggered by evaluating
        the event e. The Transaction Security PolicyCondition
        interface returns True if the policy is triggered.
        We expect it to return false due to NumberOfRecords and ExecutionTime being below threshold.*/
        System.assertEquals(false, condition.evaluate(e));
    }
}