summaryrefslogtreecommitdiffstats
path: root/dcae-analytics/dcae-analytics-tca-core/src/test/java/org/onap/dcae/analytics/tca/core/BaseTcaCoreTest.java
blob: 0d3219059b5f813a203c6717bd4affc8149724a1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/*
 * ================================================================================
 * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 *
 */

package org.onap.dcae.analytics.tca.core;

import org.onap.dcae.analytics.model.cef.EventListener;
import org.onap.dcae.analytics.model.util.json.AnalyticsModelJsonConversion;
import org.onap.dcae.analytics.tca.core.domain.TestTcaAaiEnrichmentContext;
import org.onap.dcae.analytics.tca.core.domain.TestTcaAbatementContext;
import org.onap.dcae.analytics.tca.core.service.GenericTcaExecutionContext;
import org.onap.dcae.analytics.tca.core.service.GenericTcaExecutionContext.GenericTcaExecutionContextBuilder;
import org.onap.dcae.analytics.tca.core.service.GenericTcaProcessingContext;
import org.onap.dcae.analytics.tca.core.service.GenericTcaResultContext;
import org.onap.dcae.analytics.tca.core.service.TcaAbatementContext;
import org.onap.dcae.analytics.tca.core.service.TcaExecutionContext;
import org.onap.dcae.analytics.tca.core.service.TcaProcessingContext;
import org.onap.dcae.analytics.tca.core.service.TcaResultContext;
import org.onap.dcae.analytics.tca.model.policy.TcaPolicy;
import org.onap.dcae.analytics.tca.model.util.json.TcaModelJsonConversion;
import org.onap.dcae.analytics.test.BaseAnalyticsUnitTest;

/**
 * @author Rajiv Singla
 */
public abstract class BaseTcaCoreTest extends BaseAnalyticsUnitTest {


    protected static final String TEST_POLICY_JSON_STRING;
    protected static final String TEST_CEF_EVENT_LISTENER_STRING;
    protected static final String TEST_CEF_JSON_MESSAGE_WITH_VIOLATION_STRING;
    protected static final String TEST_CEF_JSON_MESSAGE_WITH_ABATEMENT_STRING;
    protected static final String TEST_CEF_JSON_MESSAGE_WITH_INAPPLICABLE_EVENT_NAME;
    protected static final String TEST_REQUEST_ID = "testRequestId";
    protected static final TcaPolicy TEST_TCA_POLICY;

    static {

        TEST_POLICY_JSON_STRING = fromStream(TestFileLocation.TCA_POLICY_JSON);
        TEST_CEF_EVENT_LISTENER_STRING = fromStream(TestFileLocation.CEF_JSON_MESSAGE);
        TEST_CEF_JSON_MESSAGE_WITH_VIOLATION_STRING = fromStream(TestFileLocation.CEF_JSON_MESSAGE_WITH_VIOLATION);
        TEST_CEF_JSON_MESSAGE_WITH_ABATEMENT_STRING = fromStream(TestFileLocation.CEF_JSON_MESSAGE_WITH_ABATEMENT);
        TEST_CEF_JSON_MESSAGE_WITH_INAPPLICABLE_EVENT_NAME = fromStream(TestFileLocation
                .CEF_JSON_MESSAGE_WITH_INAPPLICABLE_EVENT_NAME);
        TEST_TCA_POLICY = TcaModelJsonConversion.TCA_POLICY_JSON_FUNCTION.apply(TEST_POLICY_JSON_STRING)
                .orElseThrow(() -> new IllegalStateException("Unable to get Test TcaPolicy"));
    }

    /**
     * Provides Event Listener that can be used for testing purposes
     *
     * @return test Event Listener
     */
    protected EventListener getTestEventListener() {
        return AnalyticsModelJsonConversion.EVENT_LISTENER_JSON_FUNCTION
                .apply(TEST_CEF_EVENT_LISTENER_STRING)
                .orElseThrow(() -> new IllegalStateException("Unable to get Test CEF Event Listener"));
    }

    /**
     * Provides Event Listener with violation that can be used for testing purposes
     *
     * @return test Event Listener with violation
     */
    protected EventListener getTestEventListenerWithViolation() {
        return AnalyticsModelJsonConversion.EVENT_LISTENER_JSON_FUNCTION
                .apply(TEST_CEF_JSON_MESSAGE_WITH_VIOLATION_STRING)
                .orElseThrow(() -> new IllegalStateException("Unable to get Test CEF Event Listeners with violation"));
    }


    /**
     * Provides Event Listener with abatement that can be used for testing purposes
     *
     * @return test Event Listener with abatement
     */
    protected EventListener getTestEventListenerWithAbatement() {
        return AnalyticsModelJsonConversion.EVENT_LISTENER_JSON_FUNCTION
                .apply(TEST_CEF_JSON_MESSAGE_WITH_ABATEMENT_STRING)
                .orElseThrow(() -> new IllegalStateException("Unable to get Test CEF Event Listeners with abatement"));
    }


    protected TcaExecutionContext getTestExecutionContext(final String cefMessage) {
        final TestTcaAbatementContext testTcaAbatementContext = new TestTcaAbatementContext();
        return getTestExecutionContextBuilder(cefMessage, TEST_TCA_POLICY, testTcaAbatementContext).build();
    }

    protected GenericTcaExecutionContextBuilder getTestExecutionContextBuilder(
            final String cefMessage, final TcaPolicy tcaPolicy, final TcaAbatementContext tcaAbatementContext) {

        final TcaProcessingContext tcaProcessingContext = new GenericTcaProcessingContext();
        final TcaResultContext tcaResultContext = new GenericTcaResultContext();
        final TestTcaAaiEnrichmentContext testTcaAaiEnrichmentContext = new TestTcaAaiEnrichmentContext();

        return GenericTcaExecutionContext.builder()
                .requestId(TEST_REQUEST_ID)
                .cefMessage(cefMessage)
                .tcaPolicy(tcaPolicy)
                .tcaProcessingContext(tcaProcessingContext)
                .tcaResultContext(tcaResultContext)
                .tcaAbatementContext(tcaAbatementContext)
                .tcaAaiEnrichmentContext(testTcaAaiEnrichmentContext);
    }

}