aboutsummaryrefslogtreecommitdiffstats
path: root/policy-core/src/test/java/org/onap/policy/drools/core/PolicySessionFeatureApiMock.java
blob: d825706729cd68c94159a21fa416c49586006785 (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
122
123
124
125
126
127
128
129
/*-
 * ============LICENSE_START=======================================================
 * policy-core
 * ================================================================================
 * Copyright (C) 2017-2019 AT&T Intellectual Property. All rights reserved.
 * Modifications Copyright (C) 2024 Nordix Foundation.
 * ================================================================================
 * 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.policy.drools.core;

import java.util.ArrayList;
import lombok.Setter;
import org.kie.api.runtime.KieSession;

/**
 * This class supports 'DroolsContainerTest' by implementing
 * 'PolicySessionFeatureAPI', and providing a means to indicate
 * which hooks have been invoked.
 */
public class PolicySessionFeatureApiMock implements PolicySessionFeatureApi {
    // contains the log entries since the most recent 'getLog()' call
    private static final ArrayList<String> log = new ArrayList<>();

    // if 'true', trigger an exception right after doing the log,
    // to verify that exceptions are handled
    @Setter
    private static boolean exceptionTrigger = false;

    /**
     * Get log.
     * 
     * @return the current contents of the log, and clear the log
     */
    public static ArrayList<String> getLog() {
        synchronized (log) {
            ArrayList<String> rval = new ArrayList<>(log);
            log.clear();
            return rval;
        }
    }

    /**
     * This method adds an entry to the log, and possibly triggers an exception.
     *
     * @param arg value to add to the log
     */
    private static void addLog(String arg) {
        if (exceptionTrigger) {
            // the log entry will include a '-exception' appended to the end
            synchronized (log) {
                log.add(arg + "-exception");
            }
            System.out.println("*** " + arg + "-exception invoked ***");

            // throw an exception -- it is up to the invoking code to catch it
            throw(new IllegalStateException("Triggered from " + arg));
        } else {
            // create a log entry, and display to standard output
            synchronized (log) {
                log.add(arg);
            }
            System.out.println("*** " + arg + " invoked ***");
        }
    }

    /**
     * {@inheritDoc}.
     */
    public int getSequenceNumber() {
        return 1;
    }

    /**
     * {@inheritDoc}.
     */
    public void globalInit(String[] args, String configDir) {
        addLog("globalInit");
    }

    /**
     * {@inheritDoc}.
     */
    public KieSession activatePolicySession(PolicyContainer policyContainer, String name, String kieBaseName) {
        addLog("activatePolicySession");
        return null;
    }

    /**
     * {@inheritDoc}.
     */
    public void newPolicySession(PolicySession policySession) {
        addLog("newPolicySession");
    }

    /**
     * {@inheritDoc}.
     */
    public PolicySession.ThreadModel selectThreadModel(PolicySession session) {
        addLog("selectThreadModel");
        return null;
    }

    /**
     * {@inheritDoc}.
     */
    public void disposeKieSession(PolicySession policySession) {
        addLog("disposeKieSession");
    }

    /**
     * {@inheritDoc}.
     */
    public void destroyKieSession(PolicySession policySession) {
        addLog("destroyKieSession");
    }
}